diff --git a/test.py b/test.py index 14ef47c..76d8f0c 100644 --- a/test.py +++ b/test.py @@ -15,14 +15,22 @@ class TestHttpRequest(unittest.TestCase): pass def test_response_code(self): + code = urllib2.urlopen('http://www.iana.org/domains/example/').getcode() with vcr.use_cassette(TEST_CASSETTE_FILE): - code = urllib2.urlopen('http://www.iana.org/domains/example/').getcode() + self.assertEqual(code, urllib2.urlopen('http://www.iana.org/domains/example/').getcode()) self.assertEqual(code, urllib2.urlopen('http://www.iana.org/domains/example/').getcode()) def test_response_body(self): + body = urllib2.urlopen('http://www.iana.org/domains/example/').read() with vcr.use_cassette('test/synopsis.yaml'): - body = urllib2.urlopen('http://www.iana.org/domains/example/').read() self.assertEqual(body, urllib2.urlopen('http://www.iana.org/domains/example/').read()) + self.assertEqual(body, urllib2.urlopen('http://www.iana.org/domains/example/').read()) + + def test_response_headers(self): + headers = urllib2.urlopen('http://www.iana.org/domains/example/').info() + with vcr.use_cassette('test/synopsis.yaml'): + self.assertEqual(headers, urllib2.urlopen('http://www.iana.org/domains/example/').info()) + self.assertEqual(headers, urllib2.urlopen('http://www.iana.org/domains/example/').info()) if __name__ == '__main__':