mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
import json
|
|
import os
|
|
from urllib.request import urlopen
|
|
|
|
import pytest
|
|
|
|
import vcr
|
|
from vcr.cassette import Cassette
|
|
|
|
|
|
def test_set_serializer_default_config(tmpdir, httpbin):
|
|
my_vcr = vcr.VCR(serializer="json")
|
|
|
|
with my_vcr.use_cassette(str(tmpdir.join("test.json"))):
|
|
assert my_vcr.serializer == "json"
|
|
urlopen(httpbin.url + "/get")
|
|
|
|
with open(str(tmpdir.join("test.json"))) as f:
|
|
file_content = f.read()
|
|
assert file_content.endswith("\n")
|
|
assert json.loads(file_content)
|
|
|
|
|
|
def test_default_set_cassette_library_dir(tmpdir, httpbin):
|
|
my_vcr = vcr.VCR(cassette_library_dir=str(tmpdir.join("subdir")))
|
|
|
|
with my_vcr.use_cassette("test.json"):
|
|
urlopen(httpbin.url + "/get")
|
|
|
|
assert os.path.exists(str(tmpdir.join("subdir").join("test.json")))
|
|
|
|
|
|
def test_override_set_cassette_library_dir(tmpdir, httpbin):
|
|
my_vcr = vcr.VCR(cassette_library_dir=str(tmpdir.join("subdir")))
|
|
|
|
cld = str(tmpdir.join("subdir2"))
|
|
|
|
with my_vcr.use_cassette("test.json", cassette_library_dir=cld):
|
|
urlopen(httpbin.url + "/get")
|
|
|
|
assert os.path.exists(str(tmpdir.join("subdir2").join("test.json")))
|
|
assert not os.path.exists(str(tmpdir.join("subdir").join("test.json")))
|
|
|
|
|
|
def test_override_match_on(tmpdir, httpbin):
|
|
my_vcr = vcr.VCR(match_on=["method"])
|
|
|
|
with my_vcr.use_cassette(str(tmpdir.join("test.json"))):
|
|
urlopen(httpbin.url)
|
|
|
|
with my_vcr.use_cassette(str(tmpdir.join("test.json"))) as cass:
|
|
urlopen(httpbin.url + "/get")
|
|
|
|
assert len(cass) == 1
|
|
assert cass.play_count == 1
|
|
|
|
|
|
def test_missing_matcher():
|
|
my_vcr = vcr.VCR()
|
|
my_vcr.register_matcher("awesome", object)
|
|
with pytest.raises(KeyError):
|
|
with my_vcr.use_cassette("test.yaml", match_on=["notawesome"]):
|
|
pass
|
|
|
|
|
|
def test_dont_record_on_exception(tmpdir):
|
|
my_vcr = vcr.VCR(record_on_exception=False)
|
|
|
|
@my_vcr.use_cassette(str(tmpdir.join("dontsave.yml")))
|
|
def some_test():
|
|
assert b"Not in content" in urlopen("http://httpbin.org/get")
|
|
|
|
with pytest.raises(AssertionError):
|
|
some_test()
|
|
|
|
assert not os.path.exists(str(tmpdir.join("dontsave.yml")))
|
|
|
|
# Make sure context decorator has the same behavior
|
|
with pytest.raises(AssertionError):
|
|
with my_vcr.use_cassette(str(tmpdir.join("dontsave2.yml"))):
|
|
assert b"Not in content" in urlopen("http://httpbin.org/get").read()
|
|
|
|
assert not os.path.exists(str(tmpdir.join("dontsave2.yml")))
|
|
|
|
|
|
def test_set_drop_unused_requests(tmpdir, httpbin):
|
|
my_vcr = vcr.VCR(drop_unused_requests=True)
|
|
file = str(tmpdir.join("test.yaml"))
|
|
|
|
with my_vcr.use_cassette(file):
|
|
urlopen(httpbin.url)
|
|
urlopen(httpbin.url + "/get")
|
|
|
|
cassette = Cassette.load(path=file)
|
|
assert len(cassette) == 2
|
|
|
|
with my_vcr.use_cassette(file):
|
|
urlopen(httpbin.url)
|
|
|
|
cassette = Cassette.load(path=file)
|
|
assert len(cassette) == 1
|