1 files changed, 52 insertions, 0 deletions
diff --git a/internal/lsp/handlers_helpers_test.go b/internal/lsp/handlers_helpers_test.go
new file mode 100644
index 0000000..84dce77
--- /dev/null
+++ b/internal/lsp/handlers_helpers_test.go
@@ -0,0 +1,52 @@
+package lsp
+
+import (
+    "strings"
+    "testing"
+)
+
+func TestHasDoubleSemicolonTrigger(t *testing.T) {
+    cases := []struct{
+        line string
+        want bool
+    }{
+        {";;todo; remove this", true},
+        {"prefix ;;x; suffix", true},
+        {";; spaced ;", false},
+        {"no markers", false},
+        {";;x ; space before close", false},
+    }
+    for _, tc := range cases {
+        got := hasDoubleSemicolonTrigger(tc.line)
+        if got != tc.want {
+            t.Fatalf("hasDoubleSemicolonTrigger(%q)=%v want %v", tc.line, got, tc.want)
+        }
+    }
+}
+
+func TestCollectSemicolonMarkers(t *testing.T) {
+    line := "keep ;ok; this and ;another; that"
+    edits := collectSemicolonMarkers(line, 7)
+    if len(edits) != 2 {
+        t.Fatalf("expected 2 edits, got %d", len(edits))
+    }
+    // Validate the first edit aligns with ;ok;
+    start := strings.Index(line, ";ok;")
+    if start < 0 { t.Fatalf("test setup: missing ;ok;") }
+    if edits[0].Range.Start.Line != 7 || edits[0].Range.Start.Character != start {
+        t.Fatalf("first edit start got line=%d char=%d want line=7 char=%d", edits[0].Range.Start.Line, edits[0].Range.Start.Character, start)
+    }
+}
+
+func TestPromptRemovalEditsForLine_WholeLine(t *testing.T) {
+    line := ";;todo; remove this whole line"
+    edits := promptRemovalEditsForLine(line, 3)
+    if len(edits) != 1 {
+        t.Fatalf("expected 1 whole-line edit, got %d", len(edits))
+    }
+    e := edits[0]
+    if e.Range.Start.Line != 3 || e.Range.End.Line != 3 || e.Range.Start.Character != 0 || e.Range.End.Character != len(line) {
+        t.Fatalf("unexpected range for whole-line removal: %+v", e.Range)
+    }
+}
+