Stop using mocker in with.

stripe · Oct 7, 2021 · 41a7cd0 · 41a7cd0
1 parent bc7b2e4
commit 41a7cd0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_api_requestor.py b/tests/test_api_requestor.py
@@ -2,6 +2,7 @@
 
 import datetime
 import json
+import platform
 import tempfile
 import uuid
 from collections import OrderedDict
@@ -606,9 +607,9 @@ def test_handles_failed_platform_call(
 
         def fail():
             raise RuntimeError
+        mocker.patch('platform.platform', side_effect=fail)
 
-        with mocker.patch("platform.platform", side_effect=fail):
-            requestor.request("get", self.valid_path, {}, {})
+        requestor.request("get", self.valid_path, {}, {})
 
         check_call("get", headers=APIHeaderMatcher(fail_platform_call=True))