updated tests

aiidateam · Jul 26, 2024 · f866a6f · f866a6f
1 parent 08482c8
commit f866a6f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -47,9 +47,10 @@ def firecrest_computer(myfirecrest, tmpdir: Path):
         token_uri="https://TOKEN_URI",
         client_id="CLIENT_ID",
         client_secret=str(_secret_path),
-        client_machine="MACHINE_NAME",
+        compute_resource="MACHINE_NAME",
         small_file_size_mb=1.0,
         temp_directory=str(_temp_directory),
+        api_version="2",
     )
     computer.store()
     return computer

diff --git a/tests/test_computer.py b/tests/test_computer.py
@@ -58,7 +58,7 @@ def test_validate_temp_directory(myfirecrest, monkeypatch, tmpdir: Path):
         "url": "http://test.com",
         "token_uri": "token_uri",
         "client_id": "client_id",
-        "client_machine": "client_machine",
+        "compute_resource": "compute_resource",
         "client_secret": secret_file.as_posix(),
         "small_file_size_mb": float(10),
     }
@@ -106,9 +106,10 @@ def test_dynamic_info(myfirecrest, monkeypatch, tmpdir: Path):
         "url": "http://test.com",
         "token_uri": "token_uri",
         "client_id": "client_id",
-        "client_machine": "client_machine",
+        "compute_resource": "compute_resource",
         "client_secret": secret_file.as_posix(),
         "small_file_size_mb": float(10),
+        "temp_directory": "temp_directory",
     }
 
     # should catch UTILITIES_MAX_FILE_SIZE if value is not provided