use_cassette -> CassetteContextDecorator

tests/unit/test_cassettes.py

@@ -1,6 +1,9 @@
+from six.moves import http_client as httplib
+
 import pytest
 import yaml
 import mock
+
 from vcr.cassette import Cassette
 from vcr.errors import UnhandledHTTPRequestError
 
@@ -73,8 +76,7 @@ def test_cassette_cant_read_same_request_twice():
 @mock.patch('vcr.cassette.Cassette.can_play_response_for', return_value=True)
 @mock.patch('vcr.stubs.VCRHTTPResponse')
 def test_function_decorated_with_use_cassette_can_be_invoked_multiple_times(*args):
-    from six.moves import http_client as httplib
-    @Cassette.use_cassette('test')
+    @Cassette.use('test')
     def decorated_function():
         conn = httplib.HTTPConnection("www.python.org")
         conn.request("GET", "/index.html")

tests/unit/test_vcr.py

@@ -0,0 +1,16 @@
+import mock
+
+from vcr import VCR
+
+
+
+def test_vcr_use_cassette():
+    filter_headers = mock.Mock()
+    test_vcr = VCR(filter_headers=filter_headers)
+    with mock.patch('vcr.config.Cassette') as mock_cassette_class:
+        @test_vcr.use_cassette('test')
+        def function():
+            pass
+        mock_cassette_class.call_count == 0
+        function()
+    assert mock_cassette_class.use.call_args[1]['filter_headers'] is filter_headers