From 4cdf5f748392f3fb53e842becd86e8d921413013 Mon Sep 17 00:00:00 2001 From: Kevin McCarthy Date: Wed, 7 Aug 2013 21:30:25 -1000 Subject: [PATCH] update cassette unit tests --- tests/unit/test_cassettes.py | 12 ++++-------- vcr/cassette.py | 1 - 2 files changed, 4 insertions(+), 9 deletions(-) diff --git a/tests/unit/test_cassettes.py b/tests/unit/test_cassettes.py index bd42fe2..42d1dfd 100644 --- a/tests/unit/test_cassettes.py +++ b/tests/unit/test_cassettes.py @@ -8,20 +8,17 @@ def test_cassette_load(tmpdir): {'request':'foo', 'response':'bar'} ])) a_cassette = Cassette.load(str(a_file)) - assert a_cassette._requests - assert a_cassette._responses + assert len(a_cassette) == 1 def test_cassette_serialize(): a = Cassette('test') - a._requests = ['foo'] - a._responses = ['bar'] + a.append('foo','bar') assert a.serialize() == [{'request': 'foo', 'response': 'bar'}] def test_cassette_deserialize(): a = Cassette('test') a.deserialize([{'request': 'foo', 'response': 'bar'}]) - assert a._requests == ['foo'] - assert a._responses == ['bar'] + assert a.requests == {'foo':'bar'} def test_cassette_not_played(): a = Cassette('test') @@ -35,8 +32,7 @@ def test_cassette_played(): def test_cassette_append(): a = Cassette('test') a.append('foo', 'bar') - assert a._requests == ['foo'] - assert a._responses == ['bar'] + assert a.requests == {'foo':'bar'} def test_cassette_len(): a = Cassette('test') diff --git a/vcr/cassette.py b/vcr/cassette.py index 5dcd246..7de9148 100644 --- a/vcr/cassette.py +++ b/vcr/cassette.py @@ -10,7 +10,6 @@ except ImportError: # Internal imports from .patch import install, reset from .files import load_cassette, save_cassette -from .request import Request class Cassette(object): '''A container for recorded requests and responses'''