add get_status endpoint as well as test

Author: bisgaard-itis

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/storage/zipping.py

@@ -1,7 +1,7 @@
 from typing import Final
 
 from models_library.api_schemas_long_running_tasks.base import TaskId
-from models_library.api_schemas_long_running_tasks.tasks import TaskGet
+from models_library.api_schemas_long_running_tasks.tasks import TaskGet, TaskStatus
 from models_library.api_schemas_storage import STORAGE_RPC_NAMESPACE
 from models_library.api_schemas_storage.zipping_tasks import (
     ZipTaskAbortOutput,
@@ -41,3 +41,16 @@ async def abort_zipping(
     )
     assert isinstance(result, ZipTaskAbortOutput)
     return result
+
+
+async def get_zipping_status(
+    rabbitmq_rpc_client: RabbitMQRPCClient, *, task_id: TaskId
+) -> TaskStatus:
+    result = await rabbitmq_rpc_client.request(
+        STORAGE_RPC_NAMESPACE,
+        _RPC_METHOD_NAME_ADAPTER.validate_python("get_zipping_status"),
+        task_id=task_id,
+        timeout_s=_DEFAULT_TIMEOUT_S,
+    )
+    assert isinstance(result, TaskStatus)
+    return result

services/storage/src/simcore_service_storage/api/rabbitmq_rpc/_zipping.py

@@ -1,12 +1,19 @@
+from datetime import datetime
 from uuid import uuid4
 
 from fastapi import FastAPI
-from models_library.api_schemas_long_running_tasks.tasks import TaskGet, TaskId
+from models_library.api_schemas_long_running_tasks.base import ProgressPercent
+from models_library.api_schemas_long_running_tasks.tasks import (
+    TaskGet,
+    TaskId,
+    TaskStatus,
+)
 from models_library.api_schemas_storage.zipping_tasks import (
     ZipTaskAbortOutput,
     ZipTaskStartInput,
 )
 from servicelib.rabbitmq import RPCRouter
+from simcore_service_storage.simcore_s3_dsm import TaskProgress
 
 router = RPCRouter()
 
@@ -26,3 +33,13 @@ async def start_zipping(app: FastAPI, paths: ZipTaskStartInput) -> TaskGet:
 @router.expose()
 async def abort_zipping(app: FastAPI, task_id: TaskId) -> ZipTaskAbortOutput:
     return ZipTaskAbortOutput(result=True, task_id=task_id)
+
+
+@router.expose()
+async def get_zipping_status(app: FastAPI, task_id: TaskId) -> TaskStatus:
+    progress = TaskProgress(
+        task_id=task_id,
+        message="Here's a status for you. You are welcome",
+        percent=ProgressPercent(0.5),
+    )
+    return TaskStatus(task_progress=progress, done=False, started=datetime.now())

services/storage/tests/unit/test_zipping.py

@@ -4,7 +4,7 @@
 import pytest
 from faker import Faker
 from fastapi import FastAPI
-from models_library.api_schemas_long_running_tasks.tasks import TaskGet
+from models_library.api_schemas_long_running_tasks.tasks import TaskGet, TaskStatus
 from models_library.api_schemas_storage.zipping_tasks import (
     ZipTaskAbortOutput,
     ZipTaskStartInput,
@@ -72,3 +72,10 @@ async def test_abort_zipping(rpc_client: RabbitMQRPCClient, faker: Faker):
     result = await zipping.abort_zipping(rpc_client, task_id=_task_id)
     assert isinstance(result, ZipTaskAbortOutput)
     assert result.task_id == _task_id
+
+
+async def test_get_zipping_status(rpc_client: RabbitMQRPCClient, faker: Faker):
+    _task_id = TaskId(f"{faker.uuid4()}")
+    result = await zipping.get_zipping_status(rpc_client, task_id=_task_id)
+    assert isinstance(result, TaskStatus)
+    assert result.task_progress.task_id == _task_id