Skip to content

Commit 7cb1564

Browse files
committed
refactor: update RPC client usage in license checkout tests
1 parent 3402c26 commit 7cb1564

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

services/web/server/tests/unit/with_dbs/04/licenses/test_licenses_rpc.py

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,7 @@
1818
from pytest_simcore.helpers.typing_env import EnvVarsDict
1919
from pytest_simcore.helpers.webserver_users import UserInfoDict
2020
from servicelib.rabbitmq import RabbitMQRPCClient
21-
from servicelib.rabbitmq.rpc_interfaces.webserver.licenses.licensed_items import (
22-
checkout_licensed_item_for_wallet,
23-
get_available_licensed_items_for_wallet,
24-
get_licensed_items,
25-
release_licensed_item_for_wallet,
26-
)
21+
from servicelib.rabbitmq.rpc_interfaces.webserver.v1 import WebServerRpcClient
2722
from settings_library.rabbit import RabbitSettings
2823
from simcore_postgres_database.models.licensed_item_to_resource import (
2924
licensed_item_to_resource,
@@ -42,15 +37,23 @@
4237
]
4338

4439

40+
@pytest.fixture(scope="session")
41+
def service_name() -> str:
42+
# Overrides service_name fixture needed in docker_compose_service_environment_dict fixture
43+
return "wb-api-server"
44+
45+
4546
@pytest.fixture
4647
def app_environment(
4748
rabbit_service: RabbitSettings,
4849
app_environment: EnvVarsDict,
50+
docker_compose_service_environment_dict: EnvVarsDict,
4951
monkeypatch: pytest.MonkeyPatch,
5052
):
5153
new_envs = setenvs_from_dict(
5254
monkeypatch,
5355
{
56+
**docker_compose_service_environment_dict,
5457
**app_environment,
5558
"RABBIT_HOST": rabbit_service.RABBIT_HOST,
5659
"RABBIT_PORT": f"{rabbit_service.RABBIT_PORT}",
@@ -124,7 +127,7 @@ def mock_release_licensed_item(mocker: MockerFixture) -> tuple:
124127
)
125128
async def test_license_checkout_workflow(
126129
client: TestClient,
127-
rpc_client: RabbitMQRPCClient,
130+
webserver_rpc_client: WebServerRpcClient,
128131
osparc_product_name: ProductName,
129132
logged_user: UserInfoDict,
130133
pricing_plan_id: int,
@@ -135,8 +138,8 @@ async def test_license_checkout_workflow(
135138
):
136139
assert client.app
137140

138-
result = await get_licensed_items(
139-
rpc_client, product_name=osparc_product_name, offset=0, limit=20
141+
result = await webserver_rpc_client.license.get_licensed_items(
142+
product_name=osparc_product_name, offset=0, limit=20
140143
)
141144
assert len(result.items) == 0
142145
assert result.total == 0
@@ -176,23 +179,21 @@ async def test_license_checkout_workflow(
176179
)
177180
)
178181

179-
result = await get_licensed_items(
180-
rpc_client, product_name=osparc_product_name, offset=0, limit=20
182+
result = await webserver_rpc_client.license.get_licensed_items(
183+
product_name=osparc_product_name, offset=0, limit=20
181184
)
182185
assert len(result.items) == 1
183186
assert result.total == 1
184187
assert isinstance(result, LicensedItemRpcGetPage)
185188

186189
with pytest.raises(NotImplementedError):
187-
await get_available_licensed_items_for_wallet(
188-
rpc_client,
190+
await webserver_rpc_client.license.get_available_licensed_items_for_wallet(
189191
user_id=logged_user["id"],
190192
product_name=osparc_product_name,
191193
wallet_id=1,
192194
)
193195

194-
checkout = await checkout_licensed_item_for_wallet(
195-
rpc_client,
196+
checkout = await webserver_rpc_client.license.checkout_licensed_item_for_wallet(
196197
product_name=osparc_product_name,
197198
user_id=logged_user["id"],
198199
wallet_id=1,
@@ -201,8 +202,7 @@ async def test_license_checkout_workflow(
201202
service_run_id="run_1",
202203
)
203204

204-
await release_licensed_item_for_wallet(
205-
rpc_client,
205+
await webserver_rpc_client.license.release_licensed_item_for_wallet(
206206
product_name=osparc_product_name,
207207
user_id=logged_user["id"],
208208
licensed_item_checkout_id=checkout.licensed_item_checkout_id,

0 commit comments

Comments
 (0)