Skip to content

Commit 5026d01

Browse files
author
Johan Broberg
committed
Refactor test_send_chat_history.py to remove workaround for namespace package resolution and clean up imports
1 parent c08e729 commit 5026d01

2 files changed

Lines changed: 13 additions & 30 deletions

File tree

‎libraries/microsoft-agents-a365-tooling/microsoft_agents_a365/tooling/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,3 +27,6 @@
2727
"get_mcp_base_url",
2828
"build_mcp_server_url",
2929
]
30+
31+
# Enable namespace package extension for tooling-extensions-* packages
32+
__path__ = __import__("pkgutil").extend_path(__path__, __name__)

‎tests/tooling/extensions/agentframework/services/test_send_chat_history.py‎

Lines changed: 10 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -3,37 +3,17 @@
33

44
"""Unit tests for send_chat_history_from_store methods in McpToolRegistrationService."""
55

6-
# Direct import from service file to work around namespace package resolution issues
7-
# with editable installs in the uv workspace
8-
import importlib.util
9-
import sys
10-
from pathlib import Path
11-
12-
_service_path = (
13-
Path(__file__).parent.parent.parent.parent.parent.parent
14-
/ "libraries"
15-
/ "microsoft-agents-a365-tooling-extensions-agentframework"
16-
/ "microsoft_agents_a365"
17-
/ "tooling"
18-
/ "extensions"
19-
/ "agentframework"
20-
/ "services"
21-
/ "mcp_tool_registration_service.py"
6+
import uuid
7+
from datetime import UTC, datetime
8+
from unittest.mock import AsyncMock, Mock, patch
9+
10+
import pytest
11+
from microsoft_agents.hosting.core import TurnContext
12+
from microsoft_agents_a365.runtime import OperationError, OperationResult
13+
from microsoft_agents_a365.tooling.extensions.agentframework.services import (
14+
McpToolRegistrationService,
2215
)
23-
_spec = importlib.util.spec_from_file_location("mcp_tool_registration_service", _service_path)
24-
_module = importlib.util.module_from_spec(_spec)
25-
sys.modules["mcp_tool_registration_service"] = _module
26-
_spec.loader.exec_module(_module)
27-
McpToolRegistrationService = _module.McpToolRegistrationService
28-
29-
import uuid # noqa: E402
30-
from datetime import UTC, datetime # noqa: E402
31-
from unittest.mock import AsyncMock, Mock, patch # noqa: E402
32-
33-
import pytest # noqa: E402
34-
from microsoft_agents.hosting.core import TurnContext # noqa: E402
35-
from microsoft_agents_a365.runtime import OperationError, OperationResult # noqa: E402
36-
from microsoft_agents_a365.tooling.models import ToolOptions # noqa: E402
16+
from microsoft_agents_a365.tooling.models import ToolOptions
3717

3818

3919
class TestSendChatHistoryAsync:

0 commit comments

Comments
 (0)