Skip to content

Commit b0d4397

Browse files
jopemachineclaude
andcommitted
fix(BA-5650): update test fixtures for mypy compliance
Cherry-pick test updates from slice I so that this branch passes typecheck independently. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 6f78186 commit b0d4397

18 files changed

Lines changed: 134 additions & 154 deletions

tests/unit/manager/dependencies/agents/test_registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ async def test_provide_agent_registry(
3636
hook_plugin_ctx=MagicMock(),
3737
network_plugin_ctx=MagicMock(),
3838
scheduling_controller=MagicMock(),
39+
user_repository=MagicMock(),
3940
debug=False,
4041
manager_public_key=MagicMock(),
4142
manager_secret_key=MagicMock(),

tests/unit/manager/repositories/session/test_session_repository.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,6 @@ async def session_with_kernel(self, db_with_cleanup: ExtendedAsyncSAEngine) -> S
222222
domain_name=domain_name,
223223
group_id=group_id,
224224
user_uuid=user_id,
225-
access_key=access_key,
226225
tag=None,
227226
status=SessionStatus.RUNNING,
228227
status_info=None,
@@ -254,7 +253,6 @@ async def session_with_kernel(self, db_with_cleanup: ExtendedAsyncSAEngine) -> S
254253
domain_name=domain_name,
255254
group_id=group_id,
256255
user_uuid=user_id,
257-
access_key=access_key,
258256
cluster_mode=ClusterMode.SINGLE_NODE.value,
259257
cluster_size=1,
260258
cluster_role="main",
@@ -375,8 +373,7 @@ async def test_search_sessions(
375373
assert session_data.name == "test-session"
376374
assert session_data.domain_name == session_with_kernel.domain_name
377375
assert session_data.group_id == session_with_kernel.group_id
378-
assert session_data.user_uuid == session_with_kernel.user_id
379-
assert session_data.access_key == session_with_kernel.access_key
376+
assert session_data.owner_id == session_with_kernel.user_id
380377

381378
async def test_search_sessions_empty_result(
382379
self,
@@ -548,7 +545,6 @@ async def session_with_allocations(
548545
domain_name=domain_name,
549546
group_id=group_id,
550547
user_uuid=user_id,
551-
access_key=access_key,
552548
tag=None,
553549
status=SessionStatus.RUNNING,
554550
status_info=None,
@@ -578,7 +574,6 @@ async def session_with_allocations(
578574
domain_name=domain_name,
579575
group_id=group_id,
580576
user_uuid=user_id,
581-
access_key=access_key,
582577
cluster_mode=ClusterMode.SINGLE_NODE.value,
583578
cluster_size=1,
584579
cluster_role="main",

0 commit comments

Comments
 (0)