[Internal]: Replace Instance.termination_reason values with codes #3182

Author: peterschmidt85

src/dstack/_internal/server/services/jobs/__init__.py

@@ -749,6 +749,8 @@ def _get_job_status_message(job_model: JobModel) -> str:
             return "stopped"
         elif job_model.termination_reason == JobTerminationReason.ABORTED_BY_USER:
             return "aborted"
+        elif job_model.termination_reason == JobTerminationReason.NO_BALANCE:
+            return "no balance"
     return job_model.status.value
 
 

src/tests/_internal/core/models/test_runs.py

@@ -43,6 +43,7 @@ def test_get_error_returns_expected_messages():
         JobTerminationReason.ABORTED_BY_USER,
         JobTerminationReason.TERMINATED_BY_SERVER,
         JobTerminationReason.CONTAINER_EXITED_WITH_ERROR,
+        JobTerminationReason.NO_BALANCE,
     ]
 
     for reason in JobTerminationReason:

src/tests/_internal/server/background/tasks/test_process_running_jobs.py

@@ -528,7 +528,7 @@ async def test_pulling_shim_failed(self, test_db, session: AsyncSession):
             assert SSHTunnelMock.call_count == 3
         await session.refresh(job)
         assert job.status == JobStatus.TERMINATING
-        assert job.termination_reason == JobTerminationReason.INTERRUPTED_BY_NO_CAPACITY
+        assert job.termination_reason == JobTerminationReason.INSTANCE_UNREACHABLE
         assert job.remove_at is None
 
     @pytest.mark.asyncio

src/tests/_internal/server/routers/test_fleets.py

@@ -402,6 +402,7 @@ async def test_creates_fleet(self, test_db, session: AsyncSession, client: Async
                     "unreachable": False,
                     "health_status": "healthy",
                     "termination_reason": None,
+                    "termination_reason_message": None,
                     "created": "2023-01-02T03:04:00+00:00",
                     "backend": None,
                     "region": None,
@@ -537,6 +538,7 @@ async def test_creates_ssh_fleet(self, test_db, session: AsyncSession, client: A
                     "unreachable": False,
                     "health_status": "healthy",
                     "termination_reason": None,
+                    "termination_reason_message": None,
                     "created": "2023-01-02T03:04:00+00:00",
                     "region": "remote",
                     "availability_zone": None,
@@ -708,6 +710,7 @@ async def test_updates_ssh_fleet(self, test_db, session: AsyncSession, client: A
                     "unreachable": False,
                     "health_status": "healthy",
                     "termination_reason": None,
+                    "termination_reason_message": None,
                     "created": "2023-01-02T03:04:00+00:00",
                     "region": "remote",
                     "availability_zone": None,
@@ -741,6 +744,7 @@ async def test_updates_ssh_fleet(self, test_db, session: AsyncSession, client: A
                     "unreachable": False,
                     "health_status": "healthy",
                     "termination_reason": None,
+                    "termination_reason_message": None,
                     "created": "2023-01-02T03:04:00+00:00",
                     "region": "remote",
                     "availability_zone": None,