3737from simcore_postgres_database .models .resource_tracker_pricing_units import (
3838 resource_tracker_pricing_units ,
3939)
40+ from simcore_postgres_database .models .services import services_meta_data
4041
4142pytest_simcore_core_services_selection = ["postgres" , "rabbit" ]
4243pytest_simcore_ops_services_selection = [
4344 "adminer" ,
4445]
4546
4647
48+ _SERVICE_KEY = "simcore/services/comp/itis/sleeper"
49+ _SERVICE_VERSION_1 = "2.0.2"
50+ _SERVICE_VERSION_2 = "3.0.0"
51+
52+ _SERVICE_KEY_3 = "simcore/services/comp/itis/different-service"
53+ _SERVICE_VERSION_3 = "1.0.1"
54+
55+
4756@pytest .fixture ()
4857def resource_tracker_setup_db (
4958 postgres_db : sa .engine .Engine ,
5059) -> Iterator [None ]:
5160 with postgres_db .connect () as con :
5261
62+ con .execute (
63+ services_meta_data .insert ().values (
64+ key = _SERVICE_KEY ,
65+ version = _SERVICE_VERSION_1 ,
66+ name = "name" ,
67+ description = "description" ,
68+ )
69+ )
70+ con .execute (
71+ services_meta_data .insert ().values (
72+ key = _SERVICE_KEY ,
73+ version = _SERVICE_VERSION_2 ,
74+ name = "name" ,
75+ description = "description" ,
76+ )
77+ )
78+ con .execute (
79+ services_meta_data .insert ().values (
80+ key = _SERVICE_KEY_3 ,
81+ version = _SERVICE_VERSION_3 ,
82+ name = "name" ,
83+ description = "description" ,
84+ )
85+ )
86+
5387 yield
5488
5589 con .execute (resource_tracker_pricing_unit_costs .delete ())
5690 con .execute (resource_tracker_pricing_units .delete ())
5791 con .execute (resource_tracker_pricing_plan_to_service .delete ())
5892 con .execute (resource_tracker_pricing_plans .delete ())
93+ con .execute (services_meta_data .delete ())
5994
6095
6196async def test_rpc_pricing_plans_workflow (
@@ -295,12 +330,12 @@ async def test_rpc_pricing_plans_to_service_workflow(
295330 assert isinstance (result , list )
296331 assert result == []
297332
298- _first_service_version = ServiceVersion ("2.0.2" )
333+ _first_service_version = ServiceVersion (_SERVICE_VERSION_1 )
299334 result = await pricing_plans .connect_service_to_pricing_plan (
300335 rpc_client ,
301336 product_name = "osparc" ,
302337 pricing_plan_id = _pricing_plan_id ,
303- service_key = ServiceKey ("simcore/services/comp/itis/sleeper" ),
338+ service_key = ServiceKey (_SERVICE_KEY ),
304339 service_version = _first_service_version ,
305340 )
306341 assert isinstance (result , PricingPlanToServiceGet )
@@ -318,12 +353,12 @@ async def test_rpc_pricing_plans_to_service_workflow(
318353 assert len (result ) == 1
319354
320355 # Connect different version
321- _second_service_version = ServiceVersion ("3.0.0" )
356+ _second_service_version = ServiceVersion (_SERVICE_VERSION_2 )
322357 result = await pricing_plans .connect_service_to_pricing_plan (
323358 rpc_client ,
324359 product_name = "osparc" ,
325360 pricing_plan_id = _pricing_plan_id ,
326- service_key = ServiceKey ("simcore/services/comp/itis/sleeper" ),
361+ service_key = ServiceKey (_SERVICE_KEY ),
327362 service_version = _second_service_version ,
328363 )
329364 assert isinstance (result , PricingPlanToServiceGet )
@@ -341,13 +376,13 @@ async def test_rpc_pricing_plans_to_service_workflow(
341376 assert len (result ) == 2
342377
343378 # Connect different service
344- _different_service_key = ServiceKey ("simcore/services/comp/itis/different-service" )
379+ _different_service_key = ServiceKey (_SERVICE_KEY_3 )
345380 result = await pricing_plans .connect_service_to_pricing_plan (
346381 rpc_client ,
347382 product_name = "osparc" ,
348383 pricing_plan_id = _pricing_plan_id ,
349384 service_key = _different_service_key ,
350- service_version = ServiceVersion ("1.0.0" ),
385+ service_version = ServiceVersion (_SERVICE_VERSION_3 ),
351386 )
352387 assert isinstance (result , PricingPlanToServiceGet )
353388 assert result .pricing_plan_id == _pricing_plan_id
0 commit comments