mirrors/claude-code-sdk-python
1
0
Fork 0
mirror of https://github.com/anthropics/claude-code-sdk-python.git synced 2025-07-07 14:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix tests to match updated ResultMessage type

Browse source 
- Remove cost_usd field from test expectations
- Update to use total_cost_usd instead of total_cost
- All 30 tests now pass

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Lina Tawfik 2025-06-17 16:51:28 -07:00
parent 6cc5b34d54
commit b3448f4baa
No known key found for this signature in database
3 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_client.py
View file

@ -88,13 +88,12 @@ class TestQueryFunction:
yield {
"type": "result",
"subtype": "success",
"cost_usd": 0.001,
"duration_ms": 1000,
"duration_api_ms": 800,
"is_error": False,
"num_turns": 1,
"session_id": "test-session",
"total_cost": 0.001,
"total_cost_usd": 0.001,
}
mock_transport.receive_messages = mock_receive

8
tests/test_integration.py
View file

@ -43,13 +43,12 @@ class TestIntegration:
yield {
"type": "result",
"subtype": "success",
"cost_usd": 0.001,
"duration_ms": 1000,
"duration_api_ms": 800,
"is_error": False,
"num_turns": 1,
"session_id": "test-session",
"total_cost": 0.001,
"total_cost_usd": 0.001,
}
mock_transport.receive_messages = mock_receive
@ -71,7 +70,7 @@ class TestIntegration:
# Check result message
assert isinstance(messages[1], ResultMessage)
assert messages[1].cost_usd == 0.001
assert messages[1].total_cost_usd == 0.001
assert messages[1].session_id == "test-session"
anyio.run(_test)
@ -109,13 +108,12 @@ class TestIntegration:
yield {
"type": "result",
"subtype": "success",
"cost_usd": 0.002,
"duration_ms": 1500,
"duration_api_ms": 1200,
"is_error": False,
"num_turns": 1,
"session_id": "test-session-2",
"total_cost": 0.002,
"total_cost_usd": 0.002,
}
mock_transport.receive_messages = mock_receive

3
tests/test_types.py
View file

@ -45,7 +45,6 @@ class TestMessageTypes:
"""Test creating a ResultMessage."""
msg = ResultMessage(
subtype="success",
cost_usd=0.01,
duration_ms=1500,
duration_api_ms=1200,
is_error=False,
@ -54,7 +53,7 @@ class TestMessageTypes:
total_cost_usd=0.01,
)
assert msg.subtype == "success"
assert msg.cost_usd == 0.01
assert msg.total_cost_usd == 0.01
assert msg.session_id == "session-123"