mirror of
https://github.com/anthropics/claude-agent-sdk-python.git
synced 2025-10-06 01:00:03 +03:00
b3448f4baa
- 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>
108 lines
4.0 KiB
Python
108 lines
4.0 KiB
Python
"""Tests for Claude SDK type definitions."""
|
|
|
|
from claude_code_sdk import (
|
|
AssistantMessage,
|
|
ClaudeCodeOptions,
|
|
ResultMessage,
|
|
)
|
|
from claude_code_sdk.types import TextBlock, ToolResultBlock, ToolUseBlock, UserMessage
|
|
|
|
|
|
class TestMessageTypes:
|
|
"""Test message type creation and validation."""
|
|
|
|
def test_user_message_creation(self):
|
|
"""Test creating a UserMessage."""
|
|
msg = UserMessage(content="Hello, Claude!")
|
|
assert msg.content == "Hello, Claude!"
|
|
|
|
def test_assistant_message_with_text(self):
|
|
"""Test creating an AssistantMessage with text content."""
|
|
text_block = TextBlock(text="Hello, human!")
|
|
msg = AssistantMessage(content=[text_block])
|
|
assert len(msg.content) == 1
|
|
assert msg.content[0].text == "Hello, human!"
|
|
|
|
def test_tool_use_block(self):
|
|
"""Test creating a ToolUseBlock."""
|
|
block = ToolUseBlock(
|
|
id="tool-123", name="Read", input={"file_path": "/test.txt"}
|
|
)
|
|
assert block.id == "tool-123"
|
|
assert block.name == "Read"
|
|
assert block.input["file_path"] == "/test.txt"
|
|
|
|
def test_tool_result_block(self):
|
|
"""Test creating a ToolResultBlock."""
|
|
block = ToolResultBlock(
|
|
tool_use_id="tool-123", content="File contents here", is_error=False
|
|
)
|
|
assert block.tool_use_id == "tool-123"
|
|
assert block.content == "File contents here"
|
|
assert block.is_error is False
|
|
|
|
def test_result_message(self):
|
|
"""Test creating a ResultMessage."""
|
|
msg = ResultMessage(
|
|
subtype="success",
|
|
duration_ms=1500,
|
|
duration_api_ms=1200,
|
|
is_error=False,
|
|
num_turns=1,
|
|
session_id="session-123",
|
|
total_cost_usd=0.01,
|
|
)
|
|
assert msg.subtype == "success"
|
|
assert msg.total_cost_usd == 0.01
|
|
assert msg.session_id == "session-123"
|
|
|
|
|
|
class TestOptions:
|
|
"""Test Options configuration."""
|
|
|
|
def test_default_options(self):
|
|
"""Test Options with default values."""
|
|
options = ClaudeCodeOptions()
|
|
assert options.allowed_tools == []
|
|
assert options.max_thinking_tokens == 8000
|
|
assert options.system_prompt is None
|
|
assert options.permission_mode is None
|
|
assert options.continue_conversation is False
|
|
assert options.disallowed_tools == []
|
|
|
|
def test_claude_code_options_with_tools(self):
|
|
"""Test Options with built-in tools."""
|
|
options = ClaudeCodeOptions(
|
|
allowed_tools=["Read", "Write", "Edit"], disallowed_tools=["Bash"]
|
|
)
|
|
assert options.allowed_tools == ["Read", "Write", "Edit"]
|
|
assert options.disallowed_tools == ["Bash"]
|
|
|
|
def test_claude_code_options_with_permission_mode(self):
|
|
"""Test Options with permission mode."""
|
|
options = ClaudeCodeOptions(permission_mode="bypassPermissions")
|
|
assert options.permission_mode == "bypassPermissions"
|
|
|
|
def test_claude_code_options_with_system_prompt(self):
|
|
"""Test Options with system prompt."""
|
|
options = ClaudeCodeOptions(
|
|
system_prompt="You are a helpful assistant.",
|
|
append_system_prompt="Be concise.",
|
|
)
|
|
assert options.system_prompt == "You are a helpful assistant."
|
|
assert options.append_system_prompt == "Be concise."
|
|
|
|
def test_claude_code_options_with_session_continuation(self):
|
|
"""Test Options with session continuation."""
|
|
options = ClaudeCodeOptions(continue_conversation=True, resume="session-123")
|
|
assert options.continue_conversation is True
|
|
assert options.resume == "session-123"
|
|
|
|
def test_claude_code_options_with_model_specification(self):
|
|
"""Test Options with model specification."""
|
|
options = ClaudeCodeOptions(
|
|
model="claude-3-5-sonnet-20241022", permission_prompt_tool_name="CustomTool"
|
|
)
|
|
assert options.model == "claude-3-5-sonnet-20241022"
|
|
assert options.permission_prompt_tool_name == "CustomTool"
|