mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 17:15:35 +00:00
a66f462dcd
This commit not only changes the default method of matching requests (just match on method and URI instead of the entire request + headers) but also allows the user to add custom matchers.
72 lines
1.6 KiB
Python
72 lines
1.6 KiB
Python
import pytest
|
|
import yaml
|
|
import mock
|
|
from vcr.cassette import Cassette
|
|
|
|
|
|
def test_cassette_load(tmpdir):
|
|
a_file = tmpdir.join('test_cassette.yml')
|
|
a_file.write(yaml.dump([
|
|
{'request': 'foo', 'response': 'bar'}
|
|
]))
|
|
a_cassette = Cassette.load(str(a_file))
|
|
assert len(a_cassette) == 1
|
|
|
|
|
|
def test_cassette_not_played():
|
|
a = Cassette('test')
|
|
assert not a.play_count
|
|
|
|
|
|
def test_cassette_played():
|
|
a = Cassette('test')
|
|
a.mark_played('foo')
|
|
a.mark_played('foo')
|
|
assert a.play_count == 2
|
|
|
|
|
|
def test_cassette_play_counter():
|
|
a = Cassette('test')
|
|
a.mark_played('foo')
|
|
a.mark_played('bar')
|
|
assert a.play_counts['foo'] == 1
|
|
assert a.play_counts['bar'] == 1
|
|
|
|
|
|
def test_cassette_append():
|
|
a = Cassette('test')
|
|
a.append('foo', 'bar')
|
|
assert a.requests == ['foo']
|
|
assert a.responses == ['bar']
|
|
|
|
|
|
def test_cassette_len():
|
|
a = Cassette('test')
|
|
a.append('foo', 'bar')
|
|
a.append('foo2', 'bar2')
|
|
assert len(a) == 2
|
|
|
|
|
|
def _mock_requests_match(request1, request2, matchers):
|
|
return request1 == request2
|
|
|
|
@mock.patch('vcr.cassette.requests_match', _mock_requests_match)
|
|
def test_cassette_contains():
|
|
a = Cassette('test')
|
|
a.append('foo', 'bar')
|
|
assert 'foo' in a
|
|
|
|
|
|
@mock.patch('vcr.cassette.requests_match', _mock_requests_match)
|
|
def test_cassette_response_of():
|
|
a = Cassette('test')
|
|
a.append('foo', 'bar')
|
|
assert a.response_of('foo') == 'bar'
|
|
|
|
|
|
@mock.patch('vcr.cassette.requests_match', _mock_requests_match)
|
|
def test_cassette_get_missing_response():
|
|
a = Cassette('test')
|
|
with pytest.raises(KeyError):
|
|
a.response_of('foo')
|