tests: Use dummy model in integration tests

Signed-off-by: Phoevos Kalemkeris <[email protected]>

Author: phoevos

tests/integration/test_api.py

@@ -134,7 +134,7 @@ def test_unsupported_task(client: TestClient, test_model_service_ip: str):
 def test_process(client: TestClient, config: Config, test_model_service_ip: str):
     response = client.post(
         f"/models/{test_model_service_ip}/process",
-        data="Spinal stenosis",
+        data="Kidney failure",
         headers={"Content-Type": "text/plain"},
     )
     assert response.status_code == 200
@@ -153,7 +153,7 @@ def test_process(client: TestClient, config: Config, test_model_service_ip: str)
     task_payload_key = f"{task_uuid}_payload.txt"
     tom: ObjectStoreManager = config.task_object_store_manager
     payload = tom.get_object(task_payload_key)
-    assert payload == b"Spinal stenosis"
+    assert payload == b"Kidney failure"
 
     # Verify that the queue is empty after the task is processed
     qm: QueueManager = config.queue_manager
@@ -171,7 +171,7 @@ def test_process(client: TestClient, config: Config, test_model_service_ip: str)
     except json.JSONDecodeError as e:
         pytest.fail(f"Failed to parse the result as JSON: {result}, {e}")
 
-    assert result_json["text"] == "Spinal stenosis"
+    assert result_json["text"] == "Kidney failure"
     assert len(result_json["annotations"]) == 1
 
     annotation = result_json["annotations"][0]
@@ -188,7 +188,7 @@ def test_process(client: TestClient, config: Config, test_model_service_ip: str)
             "athena_ids",
         ]
     )
-    assert annotation["label_name"] == "Spinal Stenosis"
+    assert annotation["label_name"] == "Loss Of Kidney Function"
 
     # Verify that the above match the information exposed through the user-facing API
     get_response = client.get(f"/tasks/{task_uuid}", params={"detail": True, "download_url": True})

tests/integration/utils.py

@@ -6,7 +6,6 @@
 
 import pytest
 from docker.models.containers import Container
-from dotenv import load_dotenv
 from git import Repo
 from testcontainers.compose import DockerCompose
 from testcontainers.core.container import DockerClient, DockerContainer
@@ -22,6 +21,7 @@
 
 TEST_ASSETS = Path("tests/integration/assets")
 TEST_CMS_ENV_FILE = TEST_ASSETS / "cms.env"
+TEST_CMS_MODEL_PACK = TEST_ASSETS / "simple_model4test-3.9-1.12.0_edeb88f7986cb05c.zip"
 
 COGSTACK_MODEL_SERVE_REPO = "https://github.com/CogStack/CogStack-ModelServe.git"
 COGSTACK_MODEL_SERVE_COMMIT = "a55be7b10a83e3bdbdbd1a9e13248e1557fdb0db"
@@ -172,11 +172,6 @@ def remove_cogstack_model_serve(ignore_errors: bool = False):
 
 def start_cogstack_model_serve(model_services: list[str]) -> list[DockerCompose]:
     log.info("Deploying CogStack Model Serve (this might take a few minutes)...")
-    load_dotenv()
-    model_package = os.getenv("MODEL_PACKAGE_FULL_PATH")
-    if not model_package:
-        raise ValueError("MODEL_PACKAGE_FULL_PATH environment variable is not set")
-
     with open(TEST_CMS_ENV_FILE) as f:
         const_envvars = f.read()
 
@@ -185,7 +180,7 @@ def start_cogstack_model_serve(model_services: list[str]) -> list[DockerCompose]
         env_file.write(const_envvars)
         env_file.write(f"CMS_UID={os.getuid()}\n")
         env_file.write(f"CMS_GID={os.getgid()}\n")
-        env_file.write(f"MODEL_PACKAGE_FULL_PATH={model_package}\n")
+        env_file.write(f"MODEL_PACKAGE_FULL_PATH={TEST_CMS_MODEL_PACK.absolute()}\n")
 
     log.debug(f"CogStack Model Serve environment file: {env_file_path}")