feat: Add session history retriever test

Author: seedspirit

changes/4697.feature.md

@@ -0,0 +1 @@
+Add session status history retrieval test validating history not empty and contains valid statuses

src/ai/backend/test/testcases/session/session_status_history_retriever.py

@@ -0,0 +1,39 @@
+from typing import override
+
+from ai.backend.test.contexts.client_session import ClientSessionContext
+from ai.backend.test.contexts.session import CreatedSessionMetaContext
+from ai.backend.test.templates.template import TestCode
+
+
+class SessionStatusHistoryRetriever(TestCode):
+    @override
+    async def test(self) -> None:
+        session = ClientSessionContext.current()
+        session_meta = CreatedSessionMetaContext.current()
+        session_name = session_meta.name
+
+        expected_status = set([
+            "SCHEDULED",
+            "PREPARING",
+            "PULLING",
+            "PREPARED",
+            "PENDING",
+            "CREATING",
+            "RUNNING",
+        ])
+
+        unexpected_status = set([
+            "TERMINATING",
+            "TERMINATED",
+            "EXPIRED",
+        ])
+
+        result = await session.ComputeSession(name=session_name).get_status_history()
+        assert result["result"] != "", "Status history should not be empty"
+        for status_key, status_value in result["result"].items():
+            assert status_key in expected_status, (
+                f"Status {status_key} is not a valid session status"
+            )
+            assert status_key not in unexpected_status, (
+                f"Status {status_key} should not be in the history"
+            )

src/ai/backend/test/testcases/session/testspecs.py

@@ -11,6 +11,7 @@
     InteractiveSessionFromTemplateTemplate,
     SessionTemplateTemplate,
 )
+from ai.backend.test.templates.template import BasicTestTemplate, NopTestCode
 from ai.backend.test.testcases.session.container_log_retriever import TestContainerLogRetriever
 from ai.backend.test.testcases.session.creation_failure_low_resources import (
     SessionCreationFailureLowResources,
@@ -26,11 +27,12 @@
     InteractiveSessionExecuteCodeSuccess,
 )
 from ai.backend.test.testcases.session.graph_dependency_retriever import DependencyGraphRetriever
+from ai.backend.test.testcases.session.session_status_history_retriever import (
+    SessionStatusHistoryRetriever,
+)
 from ai.backend.test.testcases.spec_manager import TestSpec, TestTag
 from ai.backend.test.tester.dependency import ClusterDep, CodeExecutionDep
 
-from ...templates.template import BasicTestTemplate, NopTestCode
-
 BATCH_SESSION_TEST_SPECS = {
     "creation_batch_session_success": TestSpec(
         name="creation_batch_session_success",
@@ -296,6 +298,17 @@
         template=BasicTestTemplate(testcode=DependencyGraphRetriever()).with_wrappers(
             KeypairAuthTemplate, BatchSessionTemplate, DependentSessionTemplate
         ),
+    ),
+    "session_status_history": TestSpec(
+        name="session_status_history",
+        description=textwrap.dedent("""
+            Tests retrieval of session status history
+            Validate that the status history is not empty and contains valid statuses
+        """),
+        tags={TestTag.SESSION, TestTag.MANAGER},
+        template=BasicTestTemplate(
+            testcode=SessionStatusHistoryRetriever(),
+        ).with_wrappers(KeypairAuthTemplate, InteractiveSessionTemplate),
         parametrizes={
             ContextName.CLUSTER_CONFIG: [
                 ClusterDep(
@@ -315,7 +328,6 @@
     ),
 }
 
-
 SESSION_TEST_SPECS = {
     **BATCH_SESSION_TEST_SPECS,
     **INTERACTIVE_SESSION_TEST_SPECS,