fix linting

crim-ca · Dec 5, 2023 · a39e375 · a39e375
1 parent 05b6643
commit a39e375
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -289,8 +289,8 @@ def mock_upload(_href, *_, **__):
 
 @pytest.mark.cli
 def test_file_inputs_array_uploaded_to_vault():
-    fake_href1 = f"https://some-host.com/file1.json"
-    fake_href2 = f"https://some-host.com/file2.zip"
+    fake_href1 = "https://some-host.com/file1.json"
+    fake_href2 = "https://some-host.com/file2.zip"
     fake_id = "fake_id"
     fake_token = "fake_token"
 
@@ -322,12 +322,12 @@ def mock_upload(_href, *_, **__):
         return mock_result
 
     with contextlib.ExitStack() as stack:
-        input_file1 = tempfile.NamedTemporaryFile(mode="w", suffix=os.path.splitext(fake_href1)[-1])
-        input_file2 = tempfile.NamedTemporaryFile(mode="w", suffix=os.path.splitext(fake_href2)[-1])
+        input1 = stack.enter_context(tempfile.NamedTemporaryFile(mode="w", suffix=os.path.splitext(fake_href1)[-1]))
+        input2 = stack.enter_context(tempfile.NamedTemporaryFile(mode="w", suffix=os.path.splitext(fake_href2)[-1]))
         inputs = {
             "file": [
-                {"href": input_file1.name},
-                {"href": input_file2.name}
+                {"href": input1.name},
+                {"href": input2.name}
             ]
         }
         with mock.patch("weaver.cli.WeaverClient.upload", side_effect=mock_upload):