Skip to content

Commit 57f013f

Browse files
fix tests
1 parent 1a2d1ca commit 57f013f

5 files changed

+45
-6
lines changed

services/resource-usage-tracker/tests/unit/with_dbs/test_api_resource_tracker_pricing_plans.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,7 @@ def resource_tracker_pricing_tables_db(postgres_db: sa.engine.Engine) -> Iterato
226226
con.execute(resource_tracker_pricing_units.delete())
227227
con.execute(resource_tracker_pricing_plans.delete())
228228
con.execute(resource_tracker_pricing_unit_costs.delete())
229+
con.execute(services_meta_data.delete())
229230

230231

231232
async def test_get_default_pricing_plan_for_service(

services/resource-usage-tracker/tests/unit/with_dbs/test_api_resource_tracker_pricing_plans_rpc.py

Lines changed: 41 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,25 +37,60 @@
3737
from 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

4142
pytest_simcore_core_services_selection = ["postgres", "rabbit"]
4243
pytest_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()
4857
def 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

6196
async 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

services/resource-usage-tracker/tests/unit/with_dbs/test_process_rabbitmq_message_triggered_by_listening_with_billing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ def resource_tracker_pricing_tables_db(postgres_db: sa.engine.Engine) -> Iterato
153153
con.execute(resource_tracker_pricing_plans.delete())
154154
con.execute(resource_tracker_pricing_unit_costs.delete())
155155
con.execute(resource_tracker_credit_transactions.delete())
156+
con.execute(services_meta_data.delete())
156157

157158

158159
@pytest.mark.flaky(max_runs=3)

services/resource-usage-tracker/tests/unit/with_dbs/test_process_rabbitmq_message_with_billing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,7 @@ def resource_tracker_pricing_tables_db(postgres_db: sa.engine.Engine) -> Iterato
167167
con.execute(resource_tracker_pricing_plans.delete())
168168
con.execute(resource_tracker_pricing_unit_costs.delete())
169169
con.execute(resource_tracker_credit_transactions.delete())
170+
con.execute(services_meta_data.delete())
170171

171172

172173
@pytest.fixture

services/resource-usage-tracker/tests/unit/with_dbs/test_process_rabbitmq_message_with_billing_cost_0.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ def resource_tracker_pricing_tables_db(postgres_db: sa.engine.Engine) -> Iterato
113113
con.execute(resource_tracker_pricing_plans.delete())
114114
con.execute(resource_tracker_pricing_unit_costs.delete())
115115
con.execute(resource_tracker_credit_transactions.delete())
116+
con.execute(services_meta_data.delete())
116117

117118

118119
@pytest.fixture

0 commit comments

Comments
 (0)