Skip to content

Commit a083ea7

Browse files
committed
fix in tests
1 parent 98e2a85 commit a083ea7

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

services/api-server/tests/unit/service/conftest.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@
2222
from simcore_service_api_server._service_studies import StudyService
2323
from simcore_service_api_server.services_http.webserver import AuthSession
2424
from simcore_service_api_server.services_rpc.catalog import CatalogService
25+
from simcore_service_api_server.services_rpc.director_v2 import DirectorV2Service
26+
from simcore_service_api_server.services_rpc.storage import StorageService
2527
from simcore_service_api_server.services_rpc.wb_api_server import WbApiRpcClient
2628

2729

@@ -71,6 +73,24 @@ def wb_api_rpc_client(
7173
return WbApiRpcClient(_client=mocked_rpc_client)
7274

7375

76+
@pytest.fixture
77+
def director_v2_rpc_client(
78+
mocked_rpc_client: MockType,
79+
) -> DirectorV2Service:
80+
return DirectorV2Service(_rpc_client=mocked_rpc_client)
81+
82+
83+
@pytest.fixture
84+
def storage_rpc_client(
85+
mocked_rpc_client: MockType,
86+
user_id: UserID,
87+
product_name: ProductName,
88+
) -> StorageService:
89+
return StorageService(
90+
_rpc_client=mocked_rpc_client, _user_id=user_id, _product_name=product_name
91+
)
92+
93+
7494
@pytest.fixture
7595
def auth_session(
7696
mocker: MockerFixture,
@@ -91,13 +111,17 @@ async def _create_project(project: ProjectCreateNew, **kwargs):
91111
@pytest.fixture
92112
def job_service(
93113
auth_session: AuthSession,
114+
director_v2_rpc_client: DirectorV2Service,
115+
storage_rpc_client: StorageService,
94116
wb_api_rpc_client: WbApiRpcClient,
95117
product_name: ProductName,
96118
user_id: UserID,
97119
) -> JobService:
98120
return JobService(
99121
_web_rest_client=auth_session,
100122
_web_rpc_client=wb_api_rpc_client,
123+
_storage_rpc_client=storage_rpc_client,
124+
_directorv2_rpc_client=director_v2_rpc_client,
101125
user_id=user_id,
102126
product_name=product_name,
103127
)

0 commit comments

Comments
 (0)