|
50 | 50 | from simcore_service_api_server.api.dependencies.authentication import Identity |
51 | 51 | from simcore_service_api_server.celery.worker_tasks import functions_tasks |
52 | 52 | from simcore_service_api_server.models.api_resources import JobLinks |
| 53 | +from simcore_service_api_server.models.domain.functions import ( |
| 54 | + PreRegisteredFunctionJobData, |
| 55 | +) |
53 | 56 | from simcore_service_api_server.models.schemas.jobs import JobInputs |
54 | 57 | from simcore_service_api_server.services_rpc.wb_api_server import WbApiRpcClient |
55 | 58 |
|
@@ -473,13 +476,21 @@ def _default_side_effect( |
473 | 476 | read_functions=True, |
474 | 477 | ), |
475 | 478 | ) |
| 479 | + mock_handler_in_functions_rpc_interface( |
| 480 | + "patch_registered_function_job", mock_registered_project_function_job |
| 481 | + ) |
| 482 | + |
| 483 | + pre_registered_function_job_data = PreRegisteredFunctionJobData( |
| 484 | + job_inputs=JobInputs(values={}), |
| 485 | + function_job_id=mock_registered_project_function.uid, |
| 486 | + ) |
476 | 487 |
|
477 | 488 | job = await functions_tasks.run_function( |
478 | 489 | task=MagicMock(spec=Task), |
479 | 490 | task_id=TaskID(_faker.uuid4()), |
480 | 491 | user_identity=user_identity, |
481 | 492 | function=mock_registered_project_function, |
482 | | - job_inputs=JobInputs(values={}), |
| 493 | + pre_registered_function_job_data=pre_registered_function_job_data, |
483 | 494 | pricing_spec=None, |
484 | 495 | job_links=job_links, |
485 | 496 | x_simcore_parent_project_uuid=None, |
|
0 commit comments