fix test

Author: matusdrobuliak66

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

@@ -33,6 +33,7 @@
     RutPricingPlanGet,
     RutPricingUnitGet,
 )
+from models_library.computations import CollectionRunID
 from models_library.projects import ProjectAtDB
 from models_library.projects_nodes import NodeID, NodeState
 from models_library.projects_pipeline import PipelineDetails
@@ -380,6 +381,7 @@ async def test_computation_create_validators(
     product_name: str,
     product_api_base_url: AnyHttpUrl,
     faker: Faker,
+    faker_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     proj = await project(user, workbench=fake_workbench_without_outputs)
@@ -389,13 +391,15 @@ async def test_computation_create_validators(
         product_name=product_name,
         product_api_base_url=product_api_base_url,
         use_on_demand_clusters=True,
+        collection_run_id=faker_collection_run_id,
     )
     ComputationCreate(
         user_id=user["id"],
         project_id=proj.uuid,
         product_name=product_name,
         product_api_base_url=product_api_base_url,
         use_on_demand_clusters=False,
+        collection_run_id=faker_collection_run_id,
     )
 
 
@@ -409,6 +413,7 @@ async def test_create_computation(
     create_registered_user: Callable[..., dict[str, Any]],
     project: Callable[..., Awaitable[ProjectAtDB]],
     async_client: httpx.AsyncClient,
+    faker_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     proj = await project(user, workbench=fake_workbench_without_outputs)
@@ -421,6 +426,7 @@ async def test_create_computation(
                 project_id=proj.uuid,
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
+                collection_run_id=faker_collection_run_id,
             )
         ),
     )
@@ -516,6 +522,7 @@ async def test_create_computation_with_wallet(
     sqlalchemy_async_engine: AsyncEngine,
     fake_ec2_cpus: PositiveInt,
     fake_ec2_ram: ByteSize,
+    faker_collection_run_id: CollectionRunID,
 ):
     # In billable product a wallet is passed, with a selected pricing plan
     # the pricing plan contains information about the hardware that should be used
@@ -537,6 +544,7 @@ async def test_create_computation_with_wallet(
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
                 wallet_info=wallet_info,
+                collection_run_id=faker_collection_run_id,
             )
         ),
     )
@@ -623,6 +631,7 @@ async def test_create_computation_with_wallet_with_invalid_pricing_unit_name_rai
     project: Callable[..., Awaitable[ProjectAtDB]],
     async_client: httpx.AsyncClient,
     wallet_info: WalletInfo,
+    faker_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     proj = await project(
@@ -639,6 +648,7 @@ async def test_create_computation_with_wallet_with_invalid_pricing_unit_name_rai
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
                 wallet_info=wallet_info,
+                collection_run_id=faker_collection_run_id,
             )
         ),
     )
@@ -666,6 +676,7 @@ async def test_create_computation_with_wallet_with_no_clusters_keeper_raises_503
     project: Callable[..., Awaitable[ProjectAtDB]],
     async_client: httpx.AsyncClient,
     wallet_info: WalletInfo,
+    faker_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     proj = await project(user, workbench=fake_workbench_without_outputs)
@@ -679,6 +690,7 @@ async def test_create_computation_with_wallet_with_no_clusters_keeper_raises_503
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
                 wallet_info=wallet_info,
+                collection_run_id=faker_collection_run_id,
             )
         ),
     )
@@ -719,6 +731,7 @@ async def test_start_computation(
     create_registered_user: Callable[..., dict[str, Any]],
     project: Callable[..., Awaitable[ProjectAtDB]],
     async_client: httpx.AsyncClient,
+    faker_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     proj = await project(user, workbench=fake_workbench_without_outputs)
@@ -732,6 +745,7 @@ async def test_start_computation(
                 start_pipeline=True,
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
+                collection_run_id=faker_collection_run_id,
             )
         ),
     )
@@ -753,6 +767,7 @@ async def test_start_computation_with_project_node_resources_defined(
     create_registered_user: Callable[..., dict[str, Any]],
     project: Callable[..., Awaitable[ProjectAtDB]],
     async_client: httpx.AsyncClient,
+    fake_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     assert "json_schema_extra" in ServiceResourcesDictHelpers.model_config
@@ -781,6 +796,7 @@ async def test_start_computation_with_project_node_resources_defined(
                 start_pipeline=True,
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
+                collection_run_id=fake_collection_run_id,
             )
         ),
     )
@@ -801,6 +817,7 @@ async def test_start_computation_with_deprecated_services_raises_406(
     create_registered_user: Callable[..., dict[str, Any]],
     project: Callable[..., Awaitable[ProjectAtDB]],
     async_client: httpx.AsyncClient,
+    faker_collection_run_id: CollectionRunID,
 ):
     user = create_registered_user()
     proj = await project(user, workbench=fake_workbench_without_outputs)
@@ -814,6 +831,7 @@ async def test_start_computation_with_deprecated_services_raises_406(
                 start_pipeline=True,
                 product_name=product_name,
                 product_api_base_url=product_api_base_url,
+                collection_run_id=faker_collection_run_id,
             )
         ),
     )