add rpc method for getting jobs associated with user

Author: bisgaard-itis

packages/models-library/src/models_library/api_schemas_rpc_async_jobs/async_jobs.py

@@ -35,7 +35,7 @@ class AsyncJobResult(BaseModel):
 
 class AsyncJobGet(BaseModel):
     job_id: AsyncJobId
-    task_name: str
+    job_name: str
 
 
 class AsyncJobAbort(BaseModel):

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

@@ -6,6 +6,7 @@
     DataExportTaskStartInput,
 )
 from models_library.rabbitmq_basic_types import RPCMethodName
+from models_library.users import UserID
 from pydantic import NonNegativeInt, TypeAdapter
 
 from ... import RabbitMQRPCClient
@@ -26,3 +27,15 @@ async def start_data_export(
     )
     assert isinstance(result, AsyncJobGet)
     return result
+
+
+async def get_user_jobs(
+    rabbitmq_rpc_client: RabbitMQRPCClient, *, user_id: UserID
+) -> list[AsyncJobGet]:
+    result: list[AsyncJobGet] = await rabbitmq_rpc_client.request(
+        STORAGE_RPC_NAMESPACE,
+        _RPC_METHOD_NAME_ADAPTER.validate_python("get_user_jobs"),
+        user_id=user_id,
+        timeout_s=_DEFAULT_TIMEOUT_S,
+    )
+    return result

services/storage/src/simcore_service_storage/api/rpc/_data_export.py

@@ -8,6 +8,7 @@
     DataExportTaskStartInput,
     InvalidFileIdentifierError,
 )
+from models_library.users import UserID
 from servicelib.rabbitmq import RPCRouter
 
 router = RPCRouter()
@@ -26,5 +27,12 @@ async def start_data_export(
     assert app  # nosec
     return AsyncJobGet(
         job_id=uuid4(),
-        task_name=", ".join(str(p) for p in paths.paths),
+        job_name=", ".join(str(p) for p in paths.paths),
     )
+
+
+@router.expose()
+async def get_user_jobs(app: FastAPI, user_id: UserID) -> list[AsyncJobGet]:
+    assert app  # nosec
+    assert user_id  # nosec
+    return [AsyncJobGet(job_id=uuid4(), job_name="myjob")]

services/storage/tests/unit/test_data_export.py

@@ -109,3 +109,11 @@ async def test_get_data_export_result(rpc_client: RabbitMQRPCClient, faker: Fake
         rpc_client, rpc_namespace=STORAGE_RPC_NAMESPACE, job_id=_job_id
     )
     assert isinstance(result, AsyncJobResult)
+
+
+async def test_get_user_jobs(rpc_client: RabbitMQRPCClient, faker: Faker):
+    result = await data_export.get_user_jobs(
+        rpc_client, user_id=faker.pyint(min_value=1)
+    )
+    assert isinstance(result, list)
+    assert all(isinstance(elm, AsyncJobGet) for elm in result)

services/web/server/tests/unit/with_dbs/03/test_storage_rpc.py

@@ -61,7 +61,7 @@ def side_effect(*args, **kwargs):
 @pytest.mark.parametrize(
     "backend_result_or_exception",
     [
-        AsyncJobGet(job_id=AsyncJobId(_faker.uuid4()), task_name=_faker.text()),
+        AsyncJobGet(job_id=AsyncJobId(_faker.uuid4()), job_name=_faker.text()),
         InvalidFileIdentifierError(file_id=Path("/my/file")),
         AccessRightError(user_id=_faker.pyint(min_value=0), file_id=Path("/my/file")),
         DataExportError(job_id=_faker.pyint(min_value=0)),