diff --git a/packages/server/src/repowise/server/mcp_server/tool_refactoring.py b/packages/server/src/repowise/server/mcp_server/tool_refactoring.py index e8290a7c7..7979fa6ee 100644 --- a/packages/server/src/repowise/server/mcp_server/tool_refactoring.py +++ b/packages/server/src/repowise/server/mcp_server/tool_refactoring.py @@ -96,7 +96,7 @@ async def generate_refactoring_code(suggestion_id: str, repo: str | None = None) } async with get_session(ctx.session_factory) as session: - repository = await _get_repo(session, repo) + repository = await _get_repo(session) row = await get_refactoring_suggestion(session, repository.id, suggestion_id) if row is None: return { diff --git a/tests/unit/server/test_mcp_workspace.py b/tests/unit/server/test_mcp_workspace.py index a084c3f68..ee61cbc10 100644 --- a/tests/unit/server/test_mcp_workspace.py +++ b/tests/unit/server/test_mcp_workspace.py @@ -301,6 +301,15 @@ async def test_get_health_resolves_alias_different_from_repo_name(workspace_mcp) assert result["kpis"]["file_count"] == 0 +@pytest.mark.asyncio +async def test_generate_refactoring_code_resolves_alias_different_from_repo_name(workspace_mcp): + from repowise.server.mcp_server import generate_refactoring_code + + result = await generate_refactoring_code(suggestion_id="does-not-exist", repo="frontend") + + assert result["error"] is not None + + # --------------------------------------------------------------------------- # get_overview — workspace footer # ---------------------------------------------------------------------------