mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
03c22d79dd
This support will let you select one of four different behaviors for VCR's cassettes. Closes #23
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
'''Basic tests about save behavior'''
|
|
# coding=utf-8
|
|
|
|
# External imports
|
|
import os
|
|
import urllib2
|
|
import time
|
|
|
|
# Internal imports
|
|
import vcr
|
|
|
|
|
|
def test_disk_saver_nowrite(tmpdir):
|
|
'''
|
|
Ensure that when you close a cassette without changing it it doesn't
|
|
rewrite the file
|
|
'''
|
|
fname = str(tmpdir.join('synopsis.yaml'))
|
|
with vcr.use_cassette(fname) as cass:
|
|
urllib2.urlopen('http://www.iana.org/domains/reserved').read()
|
|
assert cass.play_count == 0
|
|
last_mod = os.path.getmtime(fname)
|
|
|
|
with vcr.use_cassette(fname) as cass:
|
|
urllib2.urlopen('http://www.iana.org/domains/reserved').read()
|
|
assert cass.play_count == 1
|
|
assert cass.dirty is False
|
|
last_mod2 = os.path.getmtime(fname)
|
|
|
|
assert last_mod == last_mod2
|
|
|
|
|
|
def test_disk_saver_write(tmpdir):
|
|
'''
|
|
Ensure that when you close a cassette after changing it it does
|
|
rewrite the file
|
|
'''
|
|
fname = str(tmpdir.join('synopsis.yaml'))
|
|
with vcr.use_cassette(fname) as cass:
|
|
urllib2.urlopen('http://www.iana.org/domains/reserved').read()
|
|
assert cass.play_count == 0
|
|
last_mod = os.path.getmtime(fname)
|
|
|
|
# Make sure at least 1 second passes, otherwise sometimes
|
|
# the mtime doesn't change
|
|
time.sleep(1)
|
|
|
|
with vcr.use_cassette(fname, record_mode='any') as cass:
|
|
urllib2.urlopen('http://www.iana.org/domains/reserved').read()
|
|
urllib2.urlopen('http://httpbin.org/').read()
|
|
assert cass.play_count == 1
|
|
assert cass.dirty
|
|
last_mod2 = os.path.getmtime(fname)
|
|
|
|
assert last_mod != last_mod2
|