mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
'''Basic tests about cassettes'''
|
|
# coding=utf-8
|
|
|
|
# Internal imports
|
|
import vcr
|
|
from .common import TestVCR
|
|
|
|
# External imports
|
|
import os
|
|
import urllib2
|
|
|
|
|
|
class TestCassette(TestVCR):
|
|
'''We should be able to save a cassette'''
|
|
fixtures = os.path.join('tests', 'fixtures', 'basic')
|
|
|
|
def test_nonexistent_directory(self):
|
|
'''If we load a cassette in a nonexistent directory, it can save ok'''
|
|
self.assertFalse(os.path.exists(self.fixture('nonexistent')))
|
|
with vcr.use_cassette(self.fixture('nonexistent', 'cass.yaml')):
|
|
urllib2.urlopen('http://httpbin.org/').read()
|
|
# This should have made the file and the directory
|
|
self.assertTrue(
|
|
os.path.exists(self.fixture('nonexistent', 'cass.yaml')))
|
|
|
|
def test_unpatch(self):
|
|
'''Ensure that our cassette gets unpatched when we're done'''
|
|
with vcr.use_cassette(self.fixture('unpatch.yaml')) as cass:
|
|
urllib2.urlopen('http://httpbin.org/').read()
|
|
|
|
# Make the same requests, and assert that we haven't served any more
|
|
# requests out of cache
|
|
urllib2.urlopen('http://httpbin.org/').read()
|
|
self.assertEqual(len(cass.cached()), 0)
|