add test

Author: vickyliin

tests/integration/test_disksaver.py

@@ -20,7 +20,7 @@ def test_disk_saver_nowrite(tmpdir, mockbin_request_url):
     fname = str(tmpdir.join("synopsis.yaml"))
     with vcr.use_cassette(fname) as cass:
         urlopen(mockbin_request_url).read()
-        assert cass.play_count == 0
+        assert cass.play_count == 1
     last_mod = os.path.getmtime(fname)
 
     with vcr.use_cassette(fname) as cass:
@@ -41,7 +41,7 @@ def test_disk_saver_write(tmpdir, mockbin_request_url):
     fname = str(tmpdir.join("synopsis.yaml"))
     with vcr.use_cassette(fname) as cass:
         urlopen(mockbin_request_url).read()
-        assert cass.play_count == 0
+        assert cass.play_count == 1
     last_mod = os.path.getmtime(fname)
 
     # Make sure at least 1 second passes, otherwise sometimes
@@ -51,7 +51,7 @@ def test_disk_saver_write(tmpdir, mockbin_request_url):
     with vcr.use_cassette(fname, record_mode=vcr.mode.ANY) as cass:
         urlopen(mockbin_request_url).read()
         urlopen(mockbin_request_url + "/get").read()
-        assert cass.play_count == 1
+        assert cass.play_count == 2
         assert cass.dirty
     last_mod2 = os.path.getmtime(fname)
 

tests/unit/test_cassettes.py

@@ -55,6 +55,7 @@ def test_cassette_append():
     a.append("foo", "bar")
     assert a.requests == ["foo"]
     assert a.responses == ["bar"]
+    assert a.play_count == 1
 
 
 def test_cassette_len():
@@ -72,6 +73,7 @@ def _mock_requests_match(request1, request2, matchers):
 def test_cassette_contains():
     a = Cassette("test")
     a.append("foo", "bar")
+    a.rewind()
     assert "foo" in a
 
 
@@ -93,6 +95,7 @@ def test_cassette_get_missing_response():
 def test_cassette_cant_read_same_request_twice():
     a = Cassette("test")
     a.append("foo", "bar")
+    a.rewind()
     a.play_response("foo")
     with pytest.raises(UnhandledHTTPRequestError):
         a.play_response("foo")
@@ -143,13 +146,15 @@ def function():
 def test_cassette_not_all_played():
     a = Cassette("test")
     a.append("foo", "bar")
+    a.rewind()
     assert not a.all_played
 
 
 @mock.patch("vcr.cassette.requests_match", _mock_requests_match)
 def test_cassette_all_played():
     a = Cassette("test")
     a.append("foo", "bar")
+    a.rewind()
     a.play_response("foo")
     assert a.all_played
 
@@ -159,6 +164,7 @@ def test_cassette_allow_playback_repeats():
     a = Cassette("test", allow_playback_repeats=True)
     a.append("foo", "bar")
     a.append("other", "resp")
+    a.rewind()
     for _ in range(10):
         assert a.play_response("foo") == "bar"
     assert a.play_count == 10
@@ -183,6 +189,7 @@ def test_cassette_allow_playback_repeats():
 def test_cassette_rewound():
     a = Cassette("test")
     a.append("foo", "bar")
+    a.rewind()
     a.play_response("foo")
     assert a.all_played
 

tests/unit/test_unittest.py

@@ -180,7 +180,7 @@ def _get_cassette_library_dir(self):
     test = run_testcase(MyTest)[0][0]
     assert b"illustrative examples" in test.response
     assert len(test.cassette.requests) == 1
-    assert test.cassette.play_count == 0
+    assert test.cassette.play_count == 1
 
     # Second test reads from cassette.