@@ -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