adding unit tests

Author: matusdrobuliak66

services/director-v2/tests/unit/with_dbs/comp_scheduler/test_api_rpc_computations.py

@@ -7,7 +7,7 @@
 # pylint: disable=too-many-positional-arguments
 
 from collections.abc import Awaitable, Callable
-from datetime import datetime, timezone
+from datetime import UTC, datetime, timedelta
 from typing import Any
 
 from models_library.api_schemas_directorv2.comp_runs import (
@@ -66,7 +66,7 @@ async def test_rpc_list_computation_runs_and_tasks(
         user=user,
         project=proj,
         result=RunningState.PENDING,
-        started=datetime.now(tz=timezone.utc),
+        started=datetime.now(tz=UTC),
         iteration=2,
     )
     output = await rpc_computations.list_computations_latest_iteration_page(
@@ -82,8 +82,8 @@ async def test_rpc_list_computation_runs_and_tasks(
         user=user,
         project=proj,
         result=RunningState.SUCCESS,
-        started=datetime.now(tz=timezone.utc),
-        ended=datetime.now(tz=timezone.utc),
+        started=datetime.now(tz=UTC),
+        ended=datetime.now(tz=UTC),
         iteration=3,
     )
     output = await rpc_computations.list_computations_latest_iteration_page(
@@ -103,3 +103,105 @@ async def test_rpc_list_computation_runs_and_tasks(
     assert output.total == 4
     assert isinstance(output, ComputationTaskRpcGetPage)
     assert len(output.items) == 4
+
+
+async def test_rpc_list_computation_runs_with_filtering(
+    fake_workbench_without_outputs: dict[str, Any],
+    fake_workbench_adjacency: dict[str, Any],
+    registered_user: Callable[..., dict[str, Any]],
+    project: Callable[..., Awaitable[ProjectAtDB]],
+    create_pipeline: Callable[..., Awaitable[CompPipelineAtDB]],
+    create_tasks: Callable[..., Awaitable[list[CompTaskAtDB]]],
+    create_comp_run: Callable[..., Awaitable[CompRunsAtDB]],
+    rpc_client: RabbitMQRPCClient,
+):
+    user = registered_user()
+
+    proj_1 = await project(user, workbench=fake_workbench_without_outputs)
+    await create_pipeline(
+        project_id=f"{proj_1.uuid}",
+        dag_adjacency_list=fake_workbench_adjacency,
+    )
+    comp_tasks = await create_tasks(
+        user=user, project=proj_1, state=StateType.PUBLISHED, progress=None
+    )
+    comp_runs = await create_comp_run(
+        user=user, project=proj_1, result=RunningState.PUBLISHED
+    )
+
+    proj_2 = await project(user, workbench=fake_workbench_without_outputs)
+    await create_pipeline(
+        project_id=f"{proj_2.uuid}",
+        dag_adjacency_list=fake_workbench_adjacency,
+    )
+    comp_tasks = await create_tasks(
+        user=user, project=proj_2, state=StateType.SUCCESS, progress=None
+    )
+    comp_runs = await create_comp_run(
+        user=user, project=proj_2, result=RunningState.SUCCESS
+    )
+
+    # Test default behaviour `filter_only_running=False`
+    output = await rpc_computations.list_computations_latest_iteration_page(
+        rpc_client, product_name="osparc", user_id=user["id"]
+    )
+    assert output.total == 2
+
+    # Test filtering
+    output = await rpc_computations.list_computations_latest_iteration_page(
+        rpc_client, product_name="osparc", user_id=user["id"], filter_only_running=True
+    )
+    assert output.total == 1
+    assert output.items[0].project_uuid == proj_1.uuid
+
+
+async def test_rpc_list_computation_runs_history(
+    fake_workbench_without_outputs: dict[str, Any],
+    fake_workbench_adjacency: dict[str, Any],
+    registered_user: Callable[..., dict[str, Any]],
+    project: Callable[..., Awaitable[ProjectAtDB]],
+    create_pipeline: Callable[..., Awaitable[CompPipelineAtDB]],
+    create_tasks: Callable[..., Awaitable[list[CompTaskAtDB]]],
+    create_comp_run: Callable[..., Awaitable[CompRunsAtDB]],
+    rpc_client: RabbitMQRPCClient,
+):
+    user = registered_user()
+
+    proj = await project(user, workbench=fake_workbench_without_outputs)
+    await create_pipeline(
+        project_id=f"{proj.uuid}",
+        dag_adjacency_list=fake_workbench_adjacency,
+    )
+    comp_tasks = await create_tasks(
+        user=user, project=proj, state=StateType.PUBLISHED, progress=None
+    )
+    comp_runs_1 = await create_comp_run(
+        user=user,
+        project=proj,
+        result=RunningState.SUCCESS,
+        started=datetime.now(tz=UTC) - timedelta(minutes=120),
+        ended=datetime.now(tz=UTC) - timedelta(minutes=100),
+        iteration=1,
+    )
+    comp_runs_2 = await create_comp_run(
+        user=user,
+        project=proj,
+        result=RunningState.SUCCESS,
+        started=datetime.now(tz=UTC) - timedelta(minutes=90),
+        ended=datetime.now(tz=UTC) - timedelta(minutes=60),
+        iteration=2,
+    )
+    comp_runs_3 = await create_comp_run(
+        user=user,
+        project=proj,
+        result=RunningState.FAILED,
+        started=datetime.now(tz=UTC) - timedelta(minutes=50),
+        ended=datetime.now(tz=UTC),
+        iteration=3,
+    )
+
+    output = await rpc_computations.list_computations_iterations_page(
+        rpc_client, product_name="osparc", user_id=user["id"], project_id=proj.uuid
+    )
+    assert output.total == 3
+    assert isinstance(output, ComputationRunRpcGetPage)