dstackai · un-def · Dec 17, 2024 · Dec 17, 2024
diff --git a/pytest.ini b/pytest.ini
@@ -0,0 +1,7 @@
+[pytest]
+testpaths = src/tests
+addopts =
+    --disable-socket
+    --allow-hosts=127.0.0.1,localhost
+    ; unix socket for Docker/testcontainers
+    --allow-unix-socket
diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -4,6 +4,7 @@ httpx>=0.23
 pytest~=7.2
 pytest-asyncio>=0.21
 pytest-httpbin==2.1.0
+pytest-socket>=0.7.0
 openai>=1.53.0,<2.0.0
 freezegun>=1.2.0
 ruff==0.5.3  # Should match .pre-commit-config.yaml

diff --git a/src/tests/_internal/server/background/tasks/test_process_instances.py b/src/tests/_internal/server/background/tasks/test_process_instances.py
@@ -32,6 +32,8 @@
 )
 from dstack._internal.utils.common import get_current_datetime
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 class TestCheckShim:
     @pytest.mark.asyncio

diff --git a/src/tests/_internal/server/background/tasks/test_process_metrics.py b/src/tests/_internal/server/background/tasks/test_process_metrics.py
@@ -26,6 +26,8 @@
     get_job_provisioning_data,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 class TestCollectMetrics:
     @pytest.mark.asyncio

diff --git a/src/tests/_internal/server/background/tasks/test_process_running_jobs.py b/src/tests/_internal/server/background/tasks/test_process_running_jobs.py
@@ -38,6 +38,8 @@
 )
 from dstack._internal.utils.common import get_current_datetime
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 def get_job_provisioning_data(dockerized: bool) -> JobProvisioningData:
     return JobProvisioningData(

diff --git a/src/tests/_internal/server/background/tasks/test_process_runs.py b/src/tests/_internal/server/background/tasks/test_process_runs.py
@@ -29,6 +29,8 @@
     get_run_spec,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 async def make_run(
     session: AsyncSession, status: RunStatus = RunStatus.SUBMITTED, replicas: Union[str, int] = 1

diff --git a/src/tests/_internal/server/background/tasks/test_process_submitted_jobs.py b/src/tests/_internal/server/background/tasks/test_process_submitted_jobs.py
@@ -42,6 +42,8 @@
     get_volume_provisioning_data,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 class TestProcessSubmittedJobs:
     @pytest.mark.asyncio

diff --git a/src/tests/_internal/server/conftest.py b/src/tests/_internal/server/conftest.py
@@ -1,11 +1,13 @@
 from pathlib import Path
+from unittest.mock import Mock
 
 import httpx
 import pytest
 
 from dstack._internal.server.main import app
 from dstack._internal.server.services import encryption as encryption  # import for side-effect
 from dstack._internal.server.services import logs as logs_services
+from dstack._internal.server.services.docker import ImageConfig, ImageConfigObject
 from dstack._internal.server.testing.conf import postgres_container, session, test_db  # noqa: F401
 
 
@@ -24,3 +26,17 @@ def test_log_storage(
     storage = logs_services.FileLogStorage(root)
     monkeypatch.setattr(logs_services, "_default_log_storage", storage)
     return storage
+
+
+@pytest.fixture
+def image_config_mock(monkeypatch: pytest.MonkeyPatch) -> ImageConfig:
+    image_config = ImageConfig.parse_obj({"User": None, "Entrypoint": None, "Cmd": ["/bin/bash"]})
+    monkeypatch.setattr(
+        "dstack._internal.server.services.jobs.configurators.base._get_image_config",
+        Mock(return_value=image_config),
+    )
+    monkeypatch.setattr(
+        "dstack._internal.server.services.docker.get_image_config",
+        Mock(return_value=ImageConfigObject(config=image_config)),
+    )
+    return image_config
diff --git a/src/tests/_internal/server/routers/test_fleets.py b/src/tests/_internal/server/routers/test_fleets.py
@@ -39,6 +39,8 @@
     get_private_key_string,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 class TestListFleets:
     @pytest.mark.asyncio

diff --git a/src/tests/_internal/server/routers/test_metrics.py b/src/tests/_internal/server/routers/test_metrics.py
@@ -16,6 +16,8 @@
     get_auth_headers,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 class TestGetJobMetrics:
     @pytest.mark.asyncio

diff --git a/src/tests/_internal/server/routers/test_runs.py b/src/tests/_internal/server/routers/test_runs.py
@@ -57,6 +57,8 @@
     get_run_spec,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 client = TestClient(app)
 
 

diff --git a/src/tests/_internal/server/services/test_runs.py b/src/tests/_internal/server/services/test_runs.py
@@ -24,6 +24,8 @@
     get_volume,
 )
 
+pytestmark = pytest.mark.usefixtures("image_config_mock")
+
 
 async def make_run(
     session: AsyncSession,
-Original file line number
+Diff line change
@@ Expand Up / @@ -57,6 +57,8 @@ @@
         get_run_spec,
     )
+    pytestmark = pytest.mark.usefixtures("image_config_mock")
     client = TestClient(app)
@@ Expand Down @@