1
0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 09:13:23 +00:00

Fix lint errors

This commit is contained in:
Terseus
2022-10-30 20:18:36 +01:00
committed by Jair Henrique
parent 7599f4d50a
commit 2980bfccde
2 changed files with 10 additions and 10 deletions

View File

@@ -65,18 +65,18 @@ def test_missing_matcher():
def test_dont_record_on_exception(tmpdir):
my_vcr = vcr.VCR(record_on_exception=False)
@my_vcr.use_cassette(str(tmpdir.join('dontsave.yml')))
@my_vcr.use_cassette(str(tmpdir.join("dontsave.yml")))
def some_test():
assert b'Not in content' in urlopen('http://httpbin.org/get')
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')))
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()
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')))
assert not os.path.exists(str(tmpdir.join("dontsave2.yml")))

View File

@@ -46,7 +46,9 @@ class CassetteContextDecorator:
"""
_non_cassette_arguments = (
"path_transformer", "func_path_generator", "record_on_exception",
"path_transformer",
"func_path_generator",
"record_on_exception",
)
@classmethod
@@ -91,9 +93,7 @@ class CassetteContextDecorator:
def __exit__(self, *exc_info):
exception_was_raised = any(exc_info)
record_on_exception = self._args_getter().get(
'record_on_exception', True
)
record_on_exception = self._args_getter().get("record_on_exception", True)
if record_on_exception or not exception_was_raised:
next(self.__finish, None)
self.__finish = None