Fix tests to match updated ResultMessage type

- 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

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