Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00
Code Releases Activity

convert tests to pytest

Browse Source
This commit is contained in:
Kevin McCarthy
2013-08-10 16:45:35 -10:00
parent 1a73334b50
commit c5294a09f7
1 changed files with 109 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

229
tests/integration/test_urllib2.py
View File

@@ -1,137 +1,126 @@
'''Integration tests with urllib2''' '''Integration tests with urllib2'''
# coding=utf-8 # coding=utf-8
# Internal imports # External imports
import vcr
from .common import TestVCR
# Testing urllib2 fetching
import os import os
import urllib2 import urllib2
from urllib import urlencode from urllib import urlencode
import pytest
# Internal imports
import vcr
class TestUrllib2(TestVCR): @pytest.fixture(params=["https","http"])
'''Base class for tests for urllib2''' def scheme(request):
fixtures = os.path.join('tests', 'fixtures', 'urllib2') """
Fixture that returns both http and https
"""
return request.param
def test_response_code(scheme, tmpdir):
'''Ensure we can read a response code from a fetch'''
url = scheme + '://httpbin.org/'
with vcr.use_cassette(str(tmpdir.join('atts.yaml'))) as cass:
# Ensure that this is empty to begin with
assert len(cass) == 0
assert cass.play_count == 0
assert urllib2.urlopen(url).getcode() == urllib2.urlopen(url).getcode()
# Ensure that we've now cached a single response
assert len(cass) == 1
assert cass.play_count == 1
class TestUrllib2Http(TestUrllib2): def test_random_body(scheme, tmpdir):
'''Tests for integration with urllib2''' '''Ensure we can read the content, and that it's served from cache'''
scheme = 'http' url = scheme + '://httpbin.org/bytes/1024'
with vcr.use_cassette(str(tmpdir.join('body.yaml'))) as cass:
# Ensure that this is empty to begin with
assert len(cass) == 0
assert cass.play_count == 0
assert urllib2.urlopen(url).read() == urllib2.urlopen(url).read()
# Ensure that we've now cached a single response
assert len(cass) == 1
assert cass.play_count == 1
def test_response_code(self): def test_response_headers(scheme, tmpdir):
'''Ensure we can read a response code from a fetch''' '''Ensure we can get information from the response'''
url = self.scheme + '://httpbin.org/' url = scheme + '://httpbin.org/'
with vcr.use_cassette(self.fixture('atts.yaml')) as cass: with vcr.use_cassette(str(tmpdir.join('headers.yaml'))) as cass:
# Ensure that this is empty to begin with # Ensure that this is empty to begin with
assert len(cass) == 0 assert len(cass) == 0
assert cass.play_count == 0 assert cass.play_count == 0
assert urllib2.urlopen(url).getcode() == urllib2.urlopen(url).getcode() assert urllib2.urlopen(url).info().items() == urllib2.urlopen(url).info().items()
# Ensure that we've now cached a single response # Ensure that we've now cached a single response
assert len(cass) == 1 assert len(cass) == 1
assert cass.play_count == 1
def test_random_body(self): def test_multiple_requests(scheme, tmpdir):
'''Ensure we can read the content, and that it's served from cache''' '''Ensure that we can cache multiple requests'''
url = self.scheme + '://httpbin.org/bytes/1024' urls = [
with vcr.use_cassette(self.fixture('body.yaml')) as cass: scheme + '://httpbin.org/',
# Ensure that this is empty to begin with scheme + '://httpbin.org/get',
self.assertEqual(len(cass), 0) scheme + '://httpbin.org/bytes/1024'
self.assertEqual(cass.play_count, 0) ]
with vcr.use_cassette(str(tmpdir.join('multiple.yaml'))) as cass:
for index in range(len(urls)):
url = urls[index]
assert len(cass) == index
assert cass.play_count == index
assert urllib2.urlopen(url).read() == urllib2.urlopen(url).read() assert urllib2.urlopen(url).read() == urllib2.urlopen(url).read()
# Ensure that we've now cached a single response assert len(cass) == index + 1
self.assertEqual(len(cass), 1) assert cass.play_count == index + 1
self.assertEqual(cass.play_count, 1)
def test_response_headers(self): def test_get_data(scheme, tmpdir):
'''Ensure we can get information from the response''' '''Ensure that it works with query data'''
url = self.scheme + '://httpbin.org/' data = urlencode({'some': 1, 'data': 'here'})
with vcr.use_cassette(self.fixture('headers.yaml')) as cass: url = scheme + '://httpbin.org/get?' + data
# Ensure that this is empty to begin with with vcr.use_cassette(str(tmpdir.join('get_data.yaml'))) as cass:
self.assertEqual(len(cass), 0) # Ensure that this is empty to begin with
self.assertEqual(cass.play_count, 0) assert len(cass) == 0
self.assertEqual( assert cass.play_count == 0
urllib2.urlopen(url).info().items(), res1 = urllib2.urlopen(url).read()
urllib2.urlopen(url).info().items()) res2 = urllib2.urlopen(url).read()
# Ensure that we've now cached a single response assert res1 == res2
self.assertEqual(len(cass), 1) # Ensure that we've now cached a single response
assert len(cass) == 1
assert cass.play_count == 1
def test_multiple_requests(self): def test_post_data(scheme, tmpdir):
'''Ensure that we can cache multiple requests''' '''Ensure that it works when posting data'''
urls = [ data = urlencode({'some': 1, 'data': 'here'})
self.scheme + '://httpbin.org/', url = scheme + '://httpbin.org/post'
self.scheme + '://httpbin.org/get', with vcr.use_cassette(str(tmpdir.join('post_data.yaml'))) as cass:
self.scheme + '://httpbin.org/bytes/1024' # Ensure that this is empty to begin with
] assert len(cass) == 0
with vcr.use_cassette(self.fixture('multiple.yaml')) as cass: assert cass.play_count == 0
for index in range(len(urls)): res1 = urllib2.urlopen(url, data).read()
url = urls[index] res2 = urllib2.urlopen(url, data).read()
self.assertEqual(len(cass), index) assert res1 == res2
self.assertEqual(cass.play_count, index) # Ensure that we've now cached a single response
self.assertEqual( assert len(cass) == 1
urllib2.urlopen(url).read(), assert cass.play_count == 1
urllib2.urlopen(url).read())
self.assertEqual(len(cass), index + 1)
self.assertEqual(cass.play_count, index + 1)
def test_get_data(self): def test_post_unicode_data(scheme, tmpdir):
'''Ensure that it works with query data''' '''Ensure that it works when posting unicode data'''
data = urlencode({'some': 1, 'data': 'here'}) data = urlencode({'snowman': u'☃'.encode('utf-8')})
url = self.scheme + '://httpbin.org/get?' + data url = scheme + '://httpbin.org/post'
with vcr.use_cassette(self.fixture('get_data.yaml')) as cass: with vcr.use_cassette(str(tmpdir.join('post_data.yaml'))) as cass:
# Ensure that this is empty to begin with # Ensure that this is empty to begin with
assert len(cass) == 0 assert len(cass) == 0
assert cass.play_count == 0 assert cass.play_count == 0
res1 = urllib2.urlopen(url).read() res1 = urllib2.urlopen(url, data).read()
res2 = urllib2.urlopen(url).read() res2 = urllib2.urlopen(url, data).read()
assert res1 == res2 assert res1 == res2
# Ensure that we've now cached a single response # Ensure that we've now cached a single response
assert len(cass) == 1 assert len(cass) == 1
assert cass.play_count == 1 assert cass.play_count == 1
def test_post_data(self): def test_cross_scheme(tmpdir):
'''Ensure that it works when posting data''' '''Ensure that requests between schemes are treated separately'''
data = urlencode({'some': 1, 'data': 'here'}) # First fetch a url under https, and then again under https and then
url = self.scheme + '://httpbin.org/post' # ensure that we haven't served anything out of cache, and we have two
with vcr.use_cassette(self.fixture('post_data.yaml')) as cass: # requests / response pairs in the cassette
# Ensure that this is empty to begin with with vcr.use_cassette(str(tmpdir.join('cross_scheme.yaml'))) as cass:
self.assertEqual(len(cass), 0) urllib2.urlopen('https://httpbin.org/')
self.assertEqual(cass.play_count, 0) urllib2.urlopen('http://httpbin.org/')
self.assertEqual( assert len(cass) == 2
urllib2.urlopen(url, data).read(), assert cass.play_count == 0
urllib2.urlopen(url, data).read())
# Ensure that we've now cached a single response
self.assertEqual(len(cass), 1)
self.assertEqual(cass.play_count, 1)
def test_post_unicode_data(self):
'''Ensure that it works when posting unicode data'''
data = urlencode({'snowman': u'☃'.encode('utf-8')})
url = self.scheme + '://httpbin.org/post'
with vcr.use_cassette(self.fixture('post_data.yaml')) as cass:
# Ensure that this is empty to begin with
self.assertEqual(len(cass), 0)
self.assertEqual(cass.play_count, 0)
self.assertEqual(
urllib2.urlopen(url, data).read(),
urllib2.urlopen(url, data).read())
# Ensure that we've now cached a single response
self.assertEqual(len(cass), 1)
self.assertEqual(cass.play_count, 1)
class TestUrllib2Https(TestUrllib2Http):
'''Similar tests but for http status codes'''
scheme = 'https'
def test_cross_scheme(self):
'''Ensure that requests between schemes are treated separately'''
# First fetch a url under https, and then again under https and then
# ensure that we haven't served anything out of cache, and we have two
# requests / response pairs in the cassette
with vcr.use_cassette(self.fixture('cross_scheme.yaml')) as cass:
urllib2.urlopen('https://httpbin.org/')
urllib2.urlopen('http://httpbin.org/')
self.assertEqual(len(cass), 2)
self.assertEqual(cass.play_count, 0)