2222from simcore_service_api_server ._service_studies import StudyService
2323from simcore_service_api_server .services_http .webserver import AuthSession
2424from 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
2527from 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
7595def auth_session (
7696 mocker : MockerFixture ,
@@ -91,13 +111,17 @@ async def _create_project(project: ProjectCreateNew, **kwargs):
91111@pytest .fixture
92112def 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