1818from pytest_simcore .helpers .typing_env import EnvVarsDict
1919from pytest_simcore .helpers .webserver_users import UserInfoDict
2020from 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
2722from settings_library .rabbit import RabbitSettings
2823from simcore_postgres_database .models .licensed_item_to_resource import (
2924 licensed_item_to_resource ,
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
4647def 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)
125128async 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