verifying close mock

distributed-system-analysis · Jun 24, 2023 · b05aca5 · b05aca5
1 parent 1c9ee2d
commit b05aca5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lib/pbench/test/unit/server/test_datasets_inventory.py b/lib/pbench/test/unit/server/test_datasets_inventory.py
@@ -108,8 +108,12 @@ def test_not_a_file(self, query_get_as, monkeypatch):
         }
 
     def test_dataset_in_given_path(self, query_get_as, monkeypatch):
+        mock_close = False
+
         class MockBytesIO(io.BytesIO):
             def close(self):
+                nonlocal mock_close
+                mock_close = True
                 super().close()
 
         mock_args: Optional[tuple[Path, dict[str, Any]]] = None
@@ -143,3 +147,4 @@ def mock_send_file(path_or_file, *args, **kwargs):
         assert file_content == exp_stream
         assert args["as_attachment"] is False
         assert args["download_name"] == "f1.json"
+        assert mock_close