Tests for MCPServerConnection initialization.
| 86 | |
| 87 | |
| 88 | class TestMCPServerConnectionInit: |
| 89 | """Tests for MCPServerConnection initialization.""" |
| 90 | |
| 91 | def test_stdio_connection_init(self): |
| 92 | """Test STDIO connection initialization.""" |
| 93 | conn = MCPServerConnection( |
| 94 | name="test-stdio", |
| 95 | connection_type="stdio", |
| 96 | command="npx", |
| 97 | args=["-y", "test-server"], |
| 98 | env={"API_KEY": "test"}, |
| 99 | ) |
| 100 | assert conn.name == "test-stdio" |
| 101 | assert conn.connection_type == "stdio" |
| 102 | assert conn.command == "npx" |
| 103 | assert conn.args == ["-y", "test-server"] |
| 104 | assert conn.env == {"API_KEY": "test"} |
| 105 | assert conn.url is None |
| 106 | |
| 107 | def test_url_connection_init(self): |
| 108 | """Test URL-based connection initialization.""" |
| 109 | conn = MCPServerConnection( |
| 110 | name="test-url", |
| 111 | connection_type="streamable_http", |
| 112 | url="https://mcp.example.com/mcp", |
| 113 | headers={"Authorization": "Bearer token"}, |
| 114 | ) |
| 115 | assert conn.name == "test-url" |
| 116 | assert conn.connection_type == "streamable_http" |
| 117 | assert conn.url == "https://mcp.example.com/mcp" |
| 118 | assert conn.headers == {"Authorization": "Bearer token"} |
| 119 | assert conn.command is None |
| 120 | |
| 121 | def test_sse_connection_init(self): |
| 122 | """Test SSE connection initialization.""" |
| 123 | conn = MCPServerConnection( |
| 124 | name="test-sse", |
| 125 | connection_type="sse", |
| 126 | url="https://mcp.example.com/sse", |
| 127 | ) |
| 128 | assert conn.name == "test-sse" |
| 129 | assert conn.connection_type == "sse" |
| 130 | assert conn.url == "https://mcp.example.com/sse" |
| 131 | |
| 132 | def test_default_values(self): |
| 133 | """Test default values for optional parameters.""" |
| 134 | conn = MCPServerConnection(name="test-default") |
| 135 | assert conn.connection_type == "stdio" |
| 136 | assert conn.args == [] |
| 137 | assert conn.env == {} |
| 138 | assert conn.headers == {} |
| 139 | |
| 140 | def test_timeout_overrides(self): |
| 141 | """Test per-server timeout override initialization.""" |
| 142 | conn = MCPServerConnection( |
| 143 | name="test-timeout", |
| 144 | connection_type="sse", |
| 145 | url="https://mcp.example.com/sse", |
nothing calls this directly
no outgoing calls
no test coverage detected