fix assert used
This commit is contained in:
@@ -542,3 +542,42 @@ async def test_dispatch_uses_coordinator_tool_calling(mock_coord_class: MagicMoc
|
||||
)
|
||||
|
||||
|
||||
async def test_dispatcher_raises_type_error_for_invalid_task_info() -> None:
|
||||
mock_client: MagicMock = MagicMock(spec=GiteaClient)
|
||||
mock_tools: MagicMock = MagicMock(spec=GiteaTools)
|
||||
|
||||
# Mock return values for methods called prior to the isinstance check
|
||||
mock_client.list_repo_pull_requests.return_value = []
|
||||
mock_client.get_issue_comments.return_value = []
|
||||
mock_client.get_authenticated_user.return_value = UserModel(login="meeks-ai")
|
||||
|
||||
dispatcher = AgentDispatcher(client=mock_client, tools=mock_tools)
|
||||
|
||||
# 1. Test dispatch raises TypeError if task_info is not IssueModel for an issue task
|
||||
work_item_invalid_issue = WorkItem(
|
||||
repo_full_name="meeks/repo1",
|
||||
task_type="issue",
|
||||
task_number=42,
|
||||
task_info=PullRequestModel(number=42), # Invalid model type
|
||||
priority=0
|
||||
)
|
||||
with pytest.raises(TypeError, match="Expected task_info to be an IssueModel"):
|
||||
await dispatcher.dispatch("meeks/repo1", [work_item_invalid_issue])
|
||||
|
||||
# 2. Test _build_pr_mission raises TypeError if task_info is not PullRequestModel
|
||||
work_item_invalid_pr = WorkItem(
|
||||
repo_full_name="meeks/repo1",
|
||||
task_type="pr",
|
||||
task_number=42,
|
||||
task_info=IssueModel(number=42), # Invalid model type
|
||||
priority=0
|
||||
)
|
||||
with pytest.raises(TypeError, match="Expected task_info to be a PullRequestModel"):
|
||||
dispatcher._build_pr_mission(work_item_invalid_pr)
|
||||
|
||||
# 3. Test _build_issue_mission raises TypeError if task_info is not IssueModel
|
||||
with pytest.raises(TypeError, match="Expected task_info to be an IssueModel"):
|
||||
dispatcher._build_issue_mission(work_item_invalid_issue)
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user