1package main
2
3import (
4 "encoding/json"
5 "os"
6 "path/filepath"
7 "testing"
8)
9
10func TestHandleSetupAddGrepGlobal(t *testing.T) {
11 tmp := t.TempDir()
12 t.Setenv("ZOT_HOME", tmp)
13
14 out, err := handleSetup([]string{"add", "grep"}, tmp)
15 if err != nil {
16 t.Fatalf("handleSetup add grep: %v", err)
17 }
18 if out == "" {
19 t.Fatal("expected setup output")
20 }
21
22 data, err := os.ReadFile(filepath.Join(tmp, "mcp.json"))
23 if err != nil {
24 t.Fatalf("read mcp.json: %v", err)
25 }
26 var cfg Config
27 if err := json.Unmarshal(data, &cfg); err != nil {
28 t.Fatalf("parse mcp.json: %v", err)
29 }
30 grep := cfg.MCPServers["grep"]
31 if grep.Transport != "streamable-http" || grep.URL != "https://mcp.grep.app/" {
32 t.Fatalf("unexpected grep config: %+v", grep)
33 }
34 if len(grep.Headers) != 0 {
35 t.Fatalf("protocol headers must not be written to mcp.json: %+v", grep.Headers)
36 }
37}
38
39func TestHandleSetupAddFilesystemProject(t *testing.T) {
40 cwd := t.TempDir()
41
42 _, err := handleSetup([]string{"add", "filesystem", "--project"}, cwd)
43 if err != nil {
44 t.Fatalf("handleSetup add filesystem: %v", err)
45 }
46
47 data, err := os.ReadFile(filepath.Join(cwd, ".zot", "mcp.json"))
48 if err != nil {
49 t.Fatalf("read project mcp.json: %v", err)
50 }
51 var cfg Config
52 if err := json.Unmarshal(data, &cfg); err != nil {
53 t.Fatalf("parse project mcp.json: %v", err)
54 }
55 fs := cfg.MCPServers["filesystem"]
56 if fs.Command != "npx" || len(fs.Args) != 3 || fs.Args[2] != cwd {
57 t.Fatalf("unexpected filesystem config: %+v", fs)
58 }
59}
60
61func TestHandleSetupAddYouGlobal(t *testing.T) {
62 tmp := t.TempDir()
63 t.Setenv("ZOT_HOME", tmp)
64
65 out, err := handleSetup([]string{"add", "you"}, tmp)
66 if err != nil {
67 t.Fatalf("handleSetup add you: %v", err)
68 }
69 if out == "" {
70 t.Fatal("expected setup output")
71 }
72
73 data, err := os.ReadFile(filepath.Join(tmp, "mcp.json"))
74 if err != nil {
75 t.Fatalf("read mcp.json: %v", err)
76 }
77 var cfg Config
78 if err := json.Unmarshal(data, &cfg); err != nil {
79 t.Fatalf("parse mcp.json: %v", err)
80 }
81 you := cfg.MCPServers["you"]
82 if you.Transport != "streamable-http" || you.URL != "https://api.you.com/mcp?profile=free" {
83 t.Fatalf("unexpected you config: %+v", you)
84 }
85 if len(you.Headers) != 0 {
86 t.Fatalf("keyless template must not write auth headers: %+v", you.Headers)
87 }
88}
89
90func TestHandleSetupAddSerplyGlobal(t *testing.T) {
91 tmp := t.TempDir()
92 t.Setenv("ZOT_HOME", tmp)
93
94 out, err := handleSetup([]string{"add", "serply"}, tmp)
95 if err != nil {
96 t.Fatalf("handleSetup add serply: %v", err)
97 }
98 if out == "" {
99 t.Fatal("expected setup output")
100 }
101
102 data, err := os.ReadFile(filepath.Join(tmp, "mcp.json"))
103 if err != nil {
104 t.Fatalf("read mcp.json: %v", err)
105 }
106 var cfg Config
107 if err := json.Unmarshal(data, &cfg); err != nil {
108 t.Fatalf("parse mcp.json: %v", err)
109 }
110 serply := cfg.MCPServers["serply"]
111 if serply.Transport != "streamable-http" || serply.URL != "https://api.serply.io/mcp" {
112 t.Fatalf("unexpected serply config: %+v", serply)
113 }
114 // The keyed template must reference the key by variable so the written
115 // config carries no secret. expandServerEnv resolves it at load time, and
116 // drops the server entirely when SERPLY_API_KEY is unset.
117 if got := serply.Headers["X-Api-Key"]; got != "${SERPLY_API_KEY}" {
118 t.Fatalf("api key must stay an unexpanded variable reference, got %q", got)
119 }
120}
121
122func TestHandleSetupDuplicate(t *testing.T) {
123 tmp := t.TempDir()
124 t.Setenv("ZOT_HOME", tmp)
125
126 if _, err := handleSetup([]string{"add", "grep"}, tmp); err != nil {
127 t.Fatalf("first add failed: %v", err)
128 }
129 if _, err := handleSetup([]string{"add", "grep"}, tmp); err == nil {
130 t.Fatal("expected duplicate server error")
131 }
132}