Skip to content

Commit 083841c

Browse files
committed
test(agent): cover repository copy name conflicts
1 parent 40b3a64 commit 083841c

1 file changed

Lines changed: 24 additions & 6 deletions

File tree

test/backend/services/test_agent_service.py

Lines changed: 24 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9128,13 +9128,20 @@ async def test_import_agent_by_agent_id_resolves_name_and_display_name_conflicts
91289128
mock_create_agent,
91299129
mock_create_tool,
91309130
):
9131-
mock_query_agents.return_value = [{
9132-
"agent_id": 1,
9133-
"name": "duplicate_name",
9134-
"display_name": "Duplicate Display",
9135-
}]
9131+
mock_query_agents.side_effect = [
9132+
[{
9133+
"agent_id": 1,
9134+
"name": "duplicate_name",
9135+
"display_name": "Duplicate Display",
9136+
}],
9137+
[{
9138+
"agent_id": 2,
9139+
"name": "existing_agent",
9140+
"display_name": "Existing Agent",
9141+
}],
9142+
]
91369143
mock_query_all_tools.return_value = []
9137-
mock_resolve_model.side_effect = [[1], [2]]
9144+
mock_resolve_model.side_effect = [[1], [2], [1], [2]]
91389145
mock_create_agent.return_value = {"agent_id": 456}
91399146

91409147
agent_info = ExportAndImportAgentInfo(
@@ -9168,6 +9175,17 @@ async def test_import_agent_by_agent_id_resolves_name_and_display_name_conflicts
91689175
assert created_agent["name"] == "duplicate_name_1"
91699176
assert created_agent["display_name"] == "Duplicate Display_1"
91709177

9178+
await import_agent_by_agent_id(
9179+
import_agent_info=agent_info,
9180+
tenant_id="test_tenant",
9181+
user_id="test_user",
9182+
resolve_name_conflicts=True,
9183+
)
9184+
9185+
created_agent = mock_create_agent.call_args.kwargs["agent_info"]
9186+
assert created_agent["name"] == "duplicate_name"
9187+
assert created_agent["display_name"] == "Duplicate Display"
9188+
91719189

91729190
@pytest.mark.asyncio
91739191
@patch('backend.services.agent_service.create_or_update_tool_by_tool_info')

0 commit comments

Comments
 (0)