test: Test for task create/get

Author: Alan Christie

tests/api_adapter.py

@@ -109,18 +109,17 @@ def create_instance(self, *, running_workflow_step_id: str) -> Dict[str, Any]:
             "running_workflow_step": running_workflow_step_id,
         }
         self._instances[instance_id] = record
-        return {"id": running_workflow_step_id}
+        return {"id": instance_id}
 
     def get_instance(self, *, instance_id: str) -> Dict[str, Any]:
         if instance_id not in self._instances:
             return {}
-        return {self._instances[instance_id]}
+        return self._instances[instance_id]
 
     def create_task(self, *, instance_id: str) -> Dict[str, Any]:
         next_id: int = len(self._instances) + 1
         task_id: str = _INSTANCE_ID_FORMAT.format(id=next_id)
         record = {
-            "instance_id": instance_id,
             "done": False,
             "exit_code": 0,
         }
@@ -130,7 +129,7 @@ def create_task(self, *, instance_id: str) -> Dict[str, Any]:
     def get_task(self, *, task_id: str) -> Dict[str, Any]:
         if task_id not in self._tasks:
             return {}
-        return {self._tasks[task_id]}
+        return self._tasks[task_id]
 
     def get_job(
         self, *, collection: str, job: str, version: str

tests/test_test_api_adapter.py

@@ -158,3 +158,52 @@ def test_get_running_workflow_steps():
     assert rwfs["running_workflow_step"]["step"] == "step-1"
     assert not rwfs["running_workflow_step"]["done"]
     assert rwfs["running_workflow_step"]["running_workflow"] == rwfid
+
+
+def test_create_instance():
+    # Arrange
+    utaa = UnitTestAPIAdapter()
+
+    # Act
+    response = utaa.create_instance(running_workflow_step_id="r-workflow-step-000")
+
+    # Assert
+    assert "id" in response
+
+
+def test_create_and_get_instance():
+    # Arrange
+    utaa = UnitTestAPIAdapter()
+    response = utaa.create_instance(running_workflow_step_id="r-workflow-step-000")
+    instance_id = response["id"]
+
+    # Act
+    response = utaa.get_instance(instance_id=instance_id)
+
+    # Assert
+    assert response["running_workflow_step"] == "r-workflow-step-000"
+
+
+def test_create_task():
+    # Arrange
+    utaa = UnitTestAPIAdapter()
+
+    # Act
+    response = utaa.create_task(instance_id="instance-000")
+
+    # Assert
+    assert "id" in response
+
+
+def test_create_and_get_task():
+    # Arrange
+    utaa = UnitTestAPIAdapter()
+    response = utaa.create_task(instance_id="instance-000")
+    task_id = response["id"]
+
+    # Act
+    response = utaa.get_task(task_id=task_id)
+
+    # Assert
+    assert not response["done"]
+    assert response["exit_code"] == 0