mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 09:13:23 +00:00
41da9ddbab
Added support and tests for the Requests library. At least basic support seems to be working. The tests should run fine if you don't have requests installed; it just won't verify the requests- specific tests.
110 lines
4.0 KiB
Python
110 lines
4.0 KiB
Python
# coding=utf-8
|
|
import os
|
|
import unittest
|
|
import vcr
|
|
import requests
|
|
|
|
TEST_CASSETTE_FILE = 'cassettes/test_req.yaml'
|
|
|
|
|
|
class TestRequestsGet(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.unmolested_response = requests.get('http://httpbin.org/')
|
|
with vcr.use_cassette(TEST_CASSETTE_FILE):
|
|
self.initial_response = requests.get('http://httpbin.org/')
|
|
self.cached_response = requests.get('http://httpbin.org/')
|
|
|
|
def tearDown(self):
|
|
try:
|
|
os.remove(TEST_CASSETTE_FILE)
|
|
except OSError:
|
|
pass
|
|
|
|
def test_initial_response_code(self):
|
|
self.assertEqual(self.unmolested_response.status_code, self.initial_response.status_code)
|
|
|
|
def test_cached_response_code(self):
|
|
self.assertEqual(self.unmolested_response.status_code, self.cached_response.status_code)
|
|
|
|
def test_initial_response_headers(self):
|
|
self.assertEqual(self.unmolested_response.headers['content-type'], self.initial_response.headers['content-type'])
|
|
|
|
def test_cached_response_headers(self):
|
|
self.assertEqual(self.unmolested_response.headers['content-type'], self.cached_response.headers['content-type'])
|
|
|
|
def test_initial_response_text(self):
|
|
self.assertEqual(self.unmolested_response.text, self.initial_response.text)
|
|
|
|
def test_cached_response_text(self):
|
|
self.assertEqual(self.unmolested_response.text, self.cached_response.text)
|
|
|
|
|
|
class TestRequestsAuth(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.unmolested_response = requests.get('https://httpbin.org/basic-auth/user/passwd', auth=('user', 'passwd'))
|
|
with vcr.use_cassette(TEST_CASSETTE_FILE):
|
|
self.initial_response = requests.get('https://httpbin.org/basic-auth/user/passwd', auth=('user', 'passwd'))
|
|
self.cached_response = requests.get('https://httpbin.org/basic-auth/user/passwd', auth=('user', 'passwd'))
|
|
|
|
def tearDown(self):
|
|
try:
|
|
os.remove(TEST_CASSETTE_FILE)
|
|
except OSError:
|
|
pass
|
|
|
|
def test_initial_response_code(self):
|
|
self.assertEqual(self.unmolested_response.status_code, self.initial_response.status_code)
|
|
|
|
def test_cached_response_code(self):
|
|
self.assertEqual(self.unmolested_response.status_code, self.cached_response.status_code)
|
|
|
|
def test_cached_response_auth_can_fail(self):
|
|
auth_fail_cached = requests.get('https://httpbin.org/basic-auth/user/passwd', auth=('user', 'passwdzzz'))
|
|
self.assertNotEqual(self.unmolested_response.status_code, auth_fail_cached.status_code)
|
|
|
|
|
|
class TestRequestsPost(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
payload = {'key1': 'value1', 'key2': 'value2'}
|
|
self.unmolested_response = requests.post('http://httpbin.org/post', payload)
|
|
with vcr.use_cassette(TEST_CASSETTE_FILE):
|
|
self.initial_response = requests.post('http://httpbin.org/post', payload)
|
|
self.cached_response = requests.post('http://httpbin.org/post', payload)
|
|
|
|
def tearDown(self):
|
|
try:
|
|
os.remove(TEST_CASSETTE_FILE)
|
|
except OSError:
|
|
pass
|
|
|
|
def test_initial_post_response_text(self):
|
|
self.assertEqual(self.unmolested_response.text, self.initial_response.text)
|
|
|
|
def test_cached_post_response_text(self):
|
|
self.assertEqual(self.unmolested_response.text, self.cached_response.text)
|
|
|
|
|
|
class TestRequestsHTTPS(unittest.TestCase):
|
|
maxDiff = None
|
|
|
|
def setUp(self):
|
|
self.unmolested_response = requests.get('https://httpbin.org/get')
|
|
with vcr.use_cassette(TEST_CASSETTE_FILE):
|
|
self.initial_response = requests.get('https://httpbin.org/get')
|
|
self.cached_response = requests.get('https://httpbin.org/get')
|
|
|
|
def tearDown(self):
|
|
try:
|
|
os.remove(TEST_CASSETTE_FILE)
|
|
except OSError:
|
|
pass
|
|
|
|
def test_initial_https_response_text(self):
|
|
self.assertEqual(self.unmolested_response.text, self.initial_response.text)
|
|
|
|
def test_cached_https_response_text(self):
|
|
self.assertEqual(self.unmolested_response.text, self.cached_response.text)
|