Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions tests/test_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -235,7 +235,7 @@ class TestWherobotsClientHeader:
@patch("wherobots.db.driver.requests.post")
def test_appends_dbapi_hop_to_inbound_chain(self, mock_post, mock_get):
"""An inbound X-Wherobots-Client chain gets the dbapi hop appended on the right."""
inbound = "client=claude_web, client=mcp;ver=0.9"
inbound = "client=claude-web, client=mcp;ver=0.9"
post_kwargs = _run_connect(
mock_post,
mock_get,
Expand All @@ -259,7 +259,7 @@ def test_sets_dbapi_hop_when_no_inbound_chain(self, mock_post, mock_get):
@patch("wherobots.db.driver.requests.post")
def test_appends_case_insensitively(self, mock_post, mock_get):
"""A differently-cased inbound header is matched and collapsed to one canonical header."""
inbound = "client=claude_web"
inbound = "client=claude-web"
post_kwargs = _run_connect(
mock_post,
mock_get,
Expand Down Expand Up @@ -292,7 +292,7 @@ def test_extra_headers_do_not_override_auth(self, mock_post, mock_get):
post_kwargs = _run_connect(
mock_post,
mock_get,
extra_headers={"X-Wherobots-Client": "client=claude_web"},
extra_headers={"X-Wherobots-Client": "client=claude-web"},
)
# api_key auth is set by _run_connect; extra_headers must not remove it.
assert post_kwargs["headers"]["X-API-Key"] == "test-key"
Expand Down Expand Up @@ -347,7 +347,7 @@ def test_extra_headers_cannot_spoof_bearer_token(self, mock_post, mock_get):
@patch("wherobots.db.driver.requests.post")
def test_ws_upgrade_carries_appended_header(self, mock_post, mock_get):
"""The appended X-Wherobots-Client also flows to the WebSocket upgrade path."""
inbound = "client=claude_web, client=mcp;ver=0.9"
inbound = "client=claude-web, client=mcp;ver=0.9"
_post_kwargs, cd_kwargs = _run_connect_full(
mock_post,
mock_get,
Expand Down
Loading