/tests/test_snippet_v3.py

import unittest

from lib import schema, snippet


def make_item(**overrides):
    payload = {
        "item_id": "i1",
        "source": "grounding",
        "title": "OpenClaw comparison guide",
        "body": "",
        "url": "https://example.com",
        "snippet": "",
    }
    payload.update(overrides)
    return schema.SourceItem(**payload)


class SnippetV3Tests(unittest.TestCase):
    def test_truncate_words_preserves_short_text_and_truncates_long_text(self):
        self.assertEqual("short text", snippet._truncate_words("short text", 5))
        self.assertEqual("one two three...", snippet._truncate_words("one two three four", 3))

    def test_windows_handles_empty_short_and_overlapping_inputs(self):
        self.assertEqual([], snippet._windows([], size=5, overlap=2))
        self.assertEqual(["one two"], snippet._windows(["one", "two"], size=5, overlap=2))
        self.assertEqual(
            ["one two three", "two three four", "three four five", "four five", "five"],
            snippet._windows(["one", "two", "three", "four", "five"], size=3, overlap=2),
        )

    def test_extract_best_snippet_prefers_existing_snippet(self):
        item = make_item(snippet="existing evidence window " * 20)
        result = snippet.extract_best_snippet(item, "ignored", max_words=5)
        self.assertEqual("existing evidence window existing evidence...", result)

    def test_extract_best_snippet_falls_back_to_title_when_body_missing(self):
        item = make_item(title="OpenClaw vs NanoClaw", body="")
        self.assertEqual("OpenClaw vs NanoClaw", snippet.extract_best_snippet(item, "openclaw"))

    def test_extract_best_snippet_selects_best_matching_body_window(self):
        body = " ".join(
            [
                "generic filler words" for _ in range(40)
            ]
            + [
                "openclaw nanoclaw ironclaw comparison details" for _ in range(15)
            ]
            + [
                "more generic filler words" for _ in range(40)
            ]
        )
        item = make_item(body=body)
        result = snippet.extract_best_snippet(item, "openclaw nanoclaw ironclaw", max_words=20)
        self.assertIn("openclaw nanoclaw ironclaw", result)

if __name__ == "__main__":
    unittest.main()