import talon

if hasattr(talon, "test_mode"):
    # Only include this when we're running tests

    pass

    from .draft_ui import calculate_text_anchors

    def test_finds_anchors():
        examples = [
            ("one-word", [("a", 0, 8, 8)]),
            ("two words", [("a", 0, 3, 4), ("b", 4, 9, 9)]),
            ("two\nwords", [("a", 0, 3, 4), ("b", 4, 9, 9)]),
        ]
        anchor_labels = ["a", "b"]
        for text, expected in examples:
            # Given an example

            # When we calculate the result and turn it into a list
            result = list(calculate_text_anchors(text, 0, anchor_labels=anchor_labels))

            # Then it matches what we expect
            assert result == expected, text

    def test_positions_anchors_around_cursor():
        # In these examples the cursor is at the asterisk which is stripped by the test
        # code. Indicies after the asterisk have to take this into account.
        examples = [
            ("one*-word", [("a", 0, 8, 8)]),
            ("one-word*", [("a", 0, 8, 8)]),
            ("the three words*", [("a", 0, 3, 4), ("b", 4, 9, 10), ("c", 10, 15, 15)]),
            ("*the three words", [("a", 0, 3, 4), ("b", 4, 9, 10), ("c", 10, 15, 15)]),
            (
                "too many* words for the number of anchors",
                [("a", 0, 3, 4), ("b", 4, 8, 9), ("c", 9, 14, 15)],
            ),
            (
                "too many words fo*r the number of anchors",
                [("a", 9, 14, 15), ("b", 15, 18, 19), ("c", 19, 22, 23)],
            ),
        ]
        anchor_labels = ["a", "b", "c"]

        for text_with_cursor, expected in examples:
            # Given an example
            cursor_pos = text_with_cursor.index("*")
            text = text_with_cursor.replace("*", "")

            # When we calculate the result and turn it into a list
            result = list(
                calculate_text_anchors(text, cursor_pos, anchor_labels=anchor_labels)
            )

            # Then it matches what we expect
            assert result == expected, text