more tests

1 files changed, 23 insertions, 0 deletions

diff --git a/cmd/hexai-lsp/main_test.go b/cmd/hexai-lsp/main_test.go
new file mode 100644
index 0000000..987bfb4
--- /dev/null
+++ b/cmd/hexai-lsp/main_test.go
@@ -0,0 +1,23 @@
+package main
+
+import (
+    "bytes"
+    "log"
+    "os"
+    "testing"
+)
+
+func TestMain_Version(t *testing.T) {
+    oldArgs := os.Args
+    defer func() { os.Args = oldArgs }()
+    os.Args = []string{"hexai-lsp", "-version"}
+    var buf bytes.Buffer
+    old := log.Writer()
+    log.SetOutput(&buf)
+    defer log.SetOutput(old)
+    main()
+    if buf.Len() == 0 {
+        t.Fatalf("expected version log")
+    }
+}
+