Skip to content

Commit e7e9666

Browse files
committed
fixing mocks
1 parent 52b2604 commit e7e9666

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

services/api-server/tests/unit/api_solvers/test_api_routers_solvers_jobs_read.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,12 @@ async def test_list_all_solvers_jobs(
124124
assert job.runner_url is not None
125125
assert job.outputs_url is not None
126126

127-
assert mocked_backend.webserver_rpc["list_projects_marked_as_jobs"].called
127+
assert mocked_backend.webserver_rpc[
128+
"mocked_rabbit_rpc_client"
129+
].request.call_args.args == (
130+
"webserver",
131+
"list_projects_marked_as_jobs",
132+
)
128133

129134

130135
async def test_list_all_solvers_jobs_with_metadata_filter(
@@ -164,11 +169,15 @@ async def test_list_all_solvers_jobs_with_metadata_filter(
164169
assert jobs_page.offset == 0
165170

166171
# Check that the backend was called with the correct filter parameters
167-
assert mocked_backend.webserver_rpc["list_projects_marked_as_jobs"].called
172+
call_args = mocked_backend.webserver_rpc[
173+
"mocked_rabbit_rpc_client"
174+
].request.call_args
175+
assert call_args.args == (
176+
"webserver",
177+
"list_projects_marked_as_jobs",
178+
)
168179

169180
# Get the call args to verify filter parameters were passed correctly
170-
call_args = mocked_backend.webserver_rpc["list_projects_marked_as_jobs"].call_args
171-
172181
# The filter_any_custom_metadata parameter should contain our filters
173182
# The exact structure will depend on how your mocked function is called
174183
assert call_args is not None

0 commit comments

Comments
 (0)