Skip to content

Commit 294d0d8

Browse files
test(mcp-augment): register test fixtures and update tests
- canonical_configs.json: add augment entry (stdio config with type field) - host_registry.py: AUGMENT_FIELDS in FIELD_SETS, AugmentAdapter in get_adapter(), AUGMENT_FIELDS added to all_possible_fields union for filter test generation - test_adapter_registry.py: add "augment" to expected default hosts set Auto-generates 19+ new test cases: 1 host-config roundtrip, 19 cross-host sync pairs (10x10 - 9x9), 1 transport mutual exclusion, ~28 field filtering. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 367b736 commit 294d0d8

File tree

3 files changed

+14
-0
lines changed

3 files changed

+14
-0
lines changed

tests/test_data/mcp_adapters/canonical_configs.json

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,14 @@
7575
"enabled_tools": ["tool1", "tool2"],
7676
"disabled_tools": ["tool3"]
7777
},
78+
"augment": {
79+
"command": "python",
80+
"args": ["-m", "mcp_server"],
81+
"env": {"API_KEY": "test_key"},
82+
"url": null,
83+
"headers": null,
84+
"type": "stdio"
85+
},
7886
"opencode": {
7987
"command": "npx",
8088
"args": ["-y", "@modelcontextprotocol/server-filesystem", "/tmp"],

tests/test_data/mcp_adapters/host_registry.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from pathlib import Path
2020
from typing import Any, Dict, FrozenSet, List, Optional, Set, Tuple
2121

22+
from hatch.mcp_host_config.adapters.augment import AugmentAdapter
2223
from hatch.mcp_host_config.adapters.base import BaseAdapter
2324
from hatch.mcp_host_config.adapters.claude import ClaudeAdapter
2425
from hatch.mcp_host_config.adapters.codex import CodexAdapter
@@ -29,6 +30,7 @@
2930
from hatch.mcp_host_config.adapters.opencode import OpenCodeAdapter
3031
from hatch.mcp_host_config.adapters.vscode import VSCodeAdapter
3132
from hatch.mcp_host_config.fields import (
33+
AUGMENT_FIELDS,
3234
CLAUDE_FIELDS,
3335
CODEX_FIELD_MAPPINGS,
3436
CODEX_FIELDS,
@@ -58,6 +60,7 @@
5860
"kiro": KIRO_FIELDS,
5961
"codex": CODEX_FIELDS,
6062
"opencode": OPENCODE_FIELDS,
63+
"augment": AUGMENT_FIELDS,
6164
}
6265

6366
# Reverse mappings for Codex (host-native name → universal name)
@@ -97,6 +100,7 @@ def get_adapter(self) -> BaseAdapter:
97100
"kiro": KiroAdapter,
98101
"codex": CodexAdapter,
99102
"opencode": OpenCodeAdapter,
103+
"augment": AugmentAdapter,
100104
}
101105
factory = adapter_map[self.host_name]
102106
return factory()
@@ -355,6 +359,7 @@ def generate_unsupported_field_test_cases(
355359
| KIRO_FIELDS
356360
| CODEX_FIELDS
357361
| OPENCODE_FIELDS
362+
| AUGMENT_FIELDS
358363
)
359364

360365
cases: List[FilterTestCase] = []

tests/unit/mcp/test_adapter_registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ def setUp(self):
3131
def test_AR01_registry_has_all_default_hosts(self):
3232
"""AR-01: Registry initializes with all default host adapters."""
3333
expected_hosts = {
34+
"augment",
3435
"claude-desktop",
3536
"claude-code",
3637
"codex",

0 commit comments

Comments
 (0)