File size: 6,096 Bytes
bba8514 0d0f0f9 bba8514 133255a b12d65f 7b98b95 b12d65f bba8514 7b3c6de bba8514 b12d65f bba8514 b12d65f bba8514 b12d65f bba8514 b12d65f bba8514 0d0f0f9 bba8514 7b98b95 bba8514 b12d65f bba8514 133255a bba8514 133255a bba8514 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 | """Edge case tests for api/detection.py."""
from unittest.mock import patch
from free_claude_code.api.detection import (
is_filepath_extraction_request,
is_prefix_detection_request,
is_quota_check_request,
is_safety_classifier_request,
is_title_generation_request,
)
from free_claude_code.core.anthropic.models import Message, MessagesRequest
def _make_request(
content: str, *, inline_system: str | None = None, **kwargs
) -> MessagesRequest:
messages = []
if inline_system is not None:
messages.append(Message(role="system", content=inline_system))
messages.append(Message(role="user", content=content))
return MessagesRequest(
model="claude-3-sonnet",
max_tokens=kwargs.pop("max_tokens", 100),
messages=messages,
**kwargs,
)
def test_quota_detection_ignores_inline_system_context() -> None:
request = _make_request(
"Check my quota", inline_system="Current request context", max_tokens=1
)
assert is_quota_check_request(request) is True
def test_title_detection_reads_inline_system_context() -> None:
request = _make_request(
"Summarize this session",
inline_system=(
"Generate a concise, sentence-case title for this coding session. "
'Return JSON with a single "title" field.'
),
)
assert is_title_generation_request(request) is True
class TestIsPrefixDetectionRequest:
def test_inline_system_context_does_not_hide_single_user_turn(self):
req = _make_request(
"<policy_spec> Command: git status",
inline_system="Current request context",
)
assert is_prefix_detection_request(req) == (True, "git status")
def test_output_marker_handling(self):
"""Content with Command: but Output: after cmd_start; output has < or \\n\\n."""
content = "<policy_spec> Command:\nls -la\nOutput:\na.txt\nb.txt\n\nmore"
req = _make_request(content)
is_req, cmd = is_prefix_detection_request(req)
assert is_req is True
assert "ls -la" in cmd
def test_prefix_detection_with_empty_command_section(self):
"""Command: at end with no content returns empty command."""
req = _make_request("<policy_spec> Command: ")
is_req, cmd = is_prefix_detection_request(req)
assert is_req is True
assert cmd == ""
def test_exception_in_try_returns_false(self):
"""Exception in try block (e.g. content slice) returns False, ''."""
req = _make_request("<policy_spec> Command: x")
# Return object that raises when sliced - triggers except in is_prefix_detection_request
class BadStr(str):
def __getitem__(self, key):
raise TypeError("bad slice")
with patch(
"free_claude_code.api.detection.extract_text_from_content",
return_value=BadStr("<policy_spec> Command: x"),
):
is_req, cmd = is_prefix_detection_request(req)
assert is_req is False
assert cmd == ""
class TestIsSafetyClassifierRequest:
_SYSTEM = (
"You are a security monitor. Respond with <block>yes</block> "
"or <block>no</block>."
)
_USER = (
"<transcript>\nUser: review the repo\n"
"WebFetch https://example.com: fetch\n</transcript>\n<block> immediately."
)
def test_classifier_request_detected(self):
req = _make_request(self._USER, system=self._SYSTEM)
assert is_safety_classifier_request(req) is True
def test_markers_split_across_system_and_user(self):
req = _make_request(
"<transcript>\nWebFetch x\n</transcript>", system=self._SYSTEM
)
assert is_safety_classifier_request(req) is True
def test_request_with_tools_is_not_classifier(self):
req = _make_request(self._USER, system=self._SYSTEM, tools=[{"name": "search"}])
assert is_safety_classifier_request(req) is False
def test_missing_transcript_marker(self):
req = _make_request("<block> immediately", system=self._SYSTEM)
assert is_safety_classifier_request(req) is False
def test_missing_verdict_instruction(self):
req = _make_request(
"<transcript>\nWebFetch x\n</transcript>", system="just chatting"
)
assert is_safety_classifier_request(req) is False
def test_xml_content_without_verdict_instruction(self):
req = _make_request(
"Explain this format: <transcript> ... </transcript> and a <block> tag."
)
assert is_safety_classifier_request(req) is False
class TestIsFilepathExtractionRequest:
def test_inline_system_context_preserves_filepath_detection(self):
req = _make_request(
"Command:\nls\nOutput:\na.txt",
inline_system="Extract any file paths that this command reads or modifies.",
)
assert is_filepath_extraction_request(req) == (True, "ls", "a.txt")
def test_output_marker_minus_one_returns_false(self):
"""Output: not found after Command: returns False."""
content = "Command:\nls\nfilepaths"
req = _make_request(content)
is_fp, cmd, out = is_filepath_extraction_request(req)
assert is_fp is False
assert cmd == ""
assert out == ""
def test_output_has_angle_bracket_splits(self):
"""Output containing < is split and first part used."""
content = "Command:\nls\nOutput:\na.txt b.txt <extra>\nfilepaths"
req = _make_request(content)
is_fp, _cmd, out = is_filepath_extraction_request(req)
assert is_fp is True
assert "<" not in out
assert out == "a.txt b.txt"
def test_output_has_double_newline_splits(self):
"""Output containing \\n\\n is split and first part used."""
content = "Command:\nls\nOutput:\na.txt\nb.txt\n\nmore text\nfilepaths"
req = _make_request(content)
is_fp, _cmd, out = is_filepath_extraction_request(req)
assert is_fp is True
assert "more" not in out
|