1// SPDX-License-Identifier: AGPL-3.0-or-later
2
3package app
4
5import (
6 "bytes"
7 "context"
8 "errors"
9 "net/http"
10 "net/http/httptest"
11 "path/filepath"
12 "strings"
13 "sync"
14 "testing"
15 "time"
16
17 "connectrpc.com/connect"
18 "github.com/stretchr/testify/require"
19 "google.golang.org/protobuf/types/known/timestamppb"
20
21 "bnema.dev/ermite-api/gen/ermite"
22 "bnema.dev/ermite-api/gen/ermite/ermiteconnect"
23)
24
25// fakeAPI is a Connect server with stub services: each field, when set,
26// answers the matching RPC; the others answer unimplemented. It records the
27// requests it receives.
28type fakeAPI struct {
29 ermiteconnect.UnimplementedRepoServiceHandler
30 ermiteconnect.UnimplementedTokenServiceHandler
31 ermiteconnect.UnimplementedServerServiceHandler
32 ermiteconnect.UnimplementedEventServiceHandler
33 ermiteconnect.UnimplementedMergeRequestServiceHandler
34 ermiteconnect.UnimplementedIssueServiceHandler
35 ermiteconnect.UnimplementedSessionServiceHandler
36 ermiteconnect.UnimplementedRunServiceHandler
37 ermiteconnect.UnimplementedTrustedKeyServiceHandler
38
39 runners []*ermite.Runner // ListRunners answer; nil means one default runner
40
41 mu sync.Mutex
42 calls []string // procedure names, in order
43 reqs []any // request messages, in order
44 header []http.Header // request headers, in order
45
46 listRepos func(*ermite.ListReposRequest) (*ermite.ListReposResponse, error)
47 getRepo func(*ermite.GetRepoRequest) (*ermite.GetRepoResponse, error)
48 createRepo func(*ermite.CreateRepoRequest) (*ermite.CreateRepoResponse, error)
49 updateRepo func(*ermite.UpdateRepoRequest) (*ermite.UpdateRepoResponse, error)
50 deleteRepo func(*ermite.DeleteRepoRequest) (*ermite.DeleteRepoResponse, error)
51 createToken func(*ermite.CreateTokenRequest) (*ermite.CreateTokenResponse, error)
52 listTokens func(*ermite.ListTokensRequest) (*ermite.ListTokensResponse, error)
53 revokeToken func(*ermite.RevokeTokenRequest) (*ermite.RevokeTokenResponse, error)
54 getVersion func(*ermite.GetVersionRequest) (*ermite.GetVersionResponse, error)
55 watch func(context.Context, *ermite.WatchRequest, *connect.ServerStream[ermite.WatchResponse]) error
56
57 createMR func(*ermite.CreateMergeRequestRequest) (*ermite.CreateMergeRequestResponse, error)
58 listMR func(*ermite.ListMergeRequestsRequest) (*ermite.ListMergeRequestsResponse, error)
59 getMR func(*ermite.GetMergeRequestRequest) (*ermite.GetMergeRequestResponse, error)
60 addComment func(*ermite.AddCommentRequest) (*ermite.AddCommentResponse, error)
61 resolveCmt func(*ermite.ResolveCommentRequest) (*ermite.ResolveCommentResponse, error)
62 closeMR func(*ermite.CloseMergeRequestRequest) (*ermite.CloseMergeRequestResponse, error)
63 mergeMR func(*ermite.MergeMergeRequestRequest) (*ermite.MergeMergeRequestResponse, error)
64 markRead func(*ermite.MarkReadRequest) (*ermite.MarkReadResponse, error)
65 enableAuto func(*ermite.EnableAutoMergeRequest) (*ermite.EnableAutoMergeResponse, error)
66 cancelAuto func(*ermite.CancelAutoMergeRequest) (*ermite.CancelAutoMergeResponse, error)
67 createIssue func(*ermite.CreateIssueRequest) (*ermite.CreateIssueResponse, error)
68 listIssues func(*ermite.ListIssuesRequest) (*ermite.ListIssuesResponse, error)
69 getIssue func(*ermite.GetIssueRequest) (*ermite.GetIssueResponse, error)
70 addIssueCmt func(*ermite.AddIssueCommentRequest) (*ermite.AddIssueCommentResponse, error)
71 resolveIssue func(*ermite.ResolveIssueRequest) (*ermite.ResolveIssueResponse, error)
72 closeIssue func(*ermite.CloseIssueRequest) (*ermite.CloseIssueResponse, error)
73 reopenIssue func(*ermite.ReopenIssueRequest) (*ermite.ReopenIssueResponse, error)
74 getSigningKey func(*ermite.GetSigningKeyRequest) (*ermite.GetSigningKeyResponse, error)
75
76 listPasskeys func(*ermite.ListPasskeysRequest) (*ermite.ListPasskeysResponse, error)
77 removePasskey func(*ermite.RemovePasskeyRequest) (*ermite.RemovePasskeyResponse, error)
78
79 listTrustedKeys func(*ermite.ListTrustedKeysRequest) (*ermite.ListTrustedKeysResponse, error)
80 addTrustedKey func(*ermite.AddTrustedKeyRequest) (*ermite.AddTrustedKeyResponse, error)
81 removeTrustedKey func(*ermite.RemoveTrustedKeyRequest) (*ermite.RemoveTrustedKeyResponse, error)
82}
83
84func (f *fakeAPI) record(name string, h http.Header, req any) {
85 f.mu.Lock()
86 defer f.mu.Unlock()
87 f.calls = append(f.calls, name)
88 f.reqs = append(f.reqs, req)
89 f.header = append(f.header, h.Clone())
90}
91
92func (f *fakeAPI) snapshot() (calls []string, reqs []any) {
93 f.mu.Lock()
94 defer f.mu.Unlock()
95 return append([]string(nil), f.calls...), append([]any(nil), f.reqs...)
96}
97
98func respond[Req, Res any](f *fakeAPI, name string, req *connect.Request[Req], fn func(*Req) (*Res, error)) (*connect.Response[Res], error) {
99 f.record(name, req.Header(), req.Msg)
100 if fn == nil {
101 return nil, connect.NewError(connect.CodeUnimplemented, nil)
102 }
103 res, err := fn(req.Msg)
104 if err != nil {
105 return nil, err
106 }
107 return connect.NewResponse(res), nil
108}
109
110func (f *fakeAPI) ListRepos(_ context.Context, r *connect.Request[ermite.ListReposRequest]) (*connect.Response[ermite.ListReposResponse], error) {
111 return respond(f, "ListRepos", r, f.listRepos)
112}
113func (f *fakeAPI) GetRepo(_ context.Context, r *connect.Request[ermite.GetRepoRequest]) (*connect.Response[ermite.GetRepoResponse], error) {
114 return respond(f, "GetRepo", r, f.getRepo)
115}
116func (f *fakeAPI) CreateRepo(_ context.Context, r *connect.Request[ermite.CreateRepoRequest]) (*connect.Response[ermite.CreateRepoResponse], error) {
117 return respond(f, "CreateRepo", r, f.createRepo)
118}
119func (f *fakeAPI) UpdateRepo(_ context.Context, r *connect.Request[ermite.UpdateRepoRequest]) (*connect.Response[ermite.UpdateRepoResponse], error) {
120 return respond(f, "UpdateRepo", r, f.updateRepo)
121}
122func (f *fakeAPI) DeleteRepo(_ context.Context, r *connect.Request[ermite.DeleteRepoRequest]) (*connect.Response[ermite.DeleteRepoResponse], error) {
123 return respond(f, "DeleteRepo", r, f.deleteRepo)
124}
125func (f *fakeAPI) CreateToken(_ context.Context, r *connect.Request[ermite.CreateTokenRequest]) (*connect.Response[ermite.CreateTokenResponse], error) {
126 return respond(f, "CreateToken", r, f.createToken)
127}
128func (f *fakeAPI) ListTokens(_ context.Context, r *connect.Request[ermite.ListTokensRequest]) (*connect.Response[ermite.ListTokensResponse], error) {
129 return respond(f, "ListTokens", r, f.listTokens)
130}
131func (f *fakeAPI) RevokeToken(_ context.Context, r *connect.Request[ermite.RevokeTokenRequest]) (*connect.Response[ermite.RevokeTokenResponse], error) {
132 return respond(f, "RevokeToken", r, f.revokeToken)
133}
134func (f *fakeAPI) GetVersion(_ context.Context, r *connect.Request[ermite.GetVersionRequest]) (*connect.Response[ermite.GetVersionResponse], error) {
135 return respond(f, "GetVersion", r, f.getVersion)
136}
137func (f *fakeAPI) CreateMergeRequest(_ context.Context, r *connect.Request[ermite.CreateMergeRequestRequest]) (*connect.Response[ermite.CreateMergeRequestResponse], error) {
138 return respond(f, "CreateMergeRequest", r, f.createMR)
139}
140func (f *fakeAPI) ListMergeRequests(_ context.Context, r *connect.Request[ermite.ListMergeRequestsRequest]) (*connect.Response[ermite.ListMergeRequestsResponse], error) {
141 return respond(f, "ListMergeRequests", r, f.listMR)
142}
143func (f *fakeAPI) GetMergeRequest(_ context.Context, r *connect.Request[ermite.GetMergeRequestRequest]) (*connect.Response[ermite.GetMergeRequestResponse], error) {
144 return respond(f, "GetMergeRequest", r, f.getMR)
145}
146func (f *fakeAPI) AddComment(_ context.Context, r *connect.Request[ermite.AddCommentRequest]) (*connect.Response[ermite.AddCommentResponse], error) {
147 return respond(f, "AddComment", r, f.addComment)
148}
149func (f *fakeAPI) ResolveComment(_ context.Context, r *connect.Request[ermite.ResolveCommentRequest]) (*connect.Response[ermite.ResolveCommentResponse], error) {
150 return respond(f, "ResolveComment", r, f.resolveCmt)
151}
152func (f *fakeAPI) CloseMergeRequest(_ context.Context, r *connect.Request[ermite.CloseMergeRequestRequest]) (*connect.Response[ermite.CloseMergeRequestResponse], error) {
153 return respond(f, "CloseMergeRequest", r, f.closeMR)
154}
155func (f *fakeAPI) MergeMergeRequest(_ context.Context, r *connect.Request[ermite.MergeMergeRequestRequest]) (*connect.Response[ermite.MergeMergeRequestResponse], error) {
156 return respond(f, "MergeMergeRequest", r, f.mergeMR)
157}
158func (f *fakeAPI) EnableAutoMerge(_ context.Context, r *connect.Request[ermite.EnableAutoMergeRequest]) (*connect.Response[ermite.EnableAutoMergeResponse], error) {
159 return respond(f, "EnableAutoMerge", r, f.enableAuto)
160}
161func (f *fakeAPI) CancelAutoMerge(_ context.Context, r *connect.Request[ermite.CancelAutoMergeRequest]) (*connect.Response[ermite.CancelAutoMergeResponse], error) {
162 return respond(f, "CancelAutoMerge", r, f.cancelAuto)
163}
164func (f *fakeAPI) MarkRead(_ context.Context, r *connect.Request[ermite.MarkReadRequest]) (*connect.Response[ermite.MarkReadResponse], error) {
165 return respond(f, "MarkRead", r, f.markRead)
166}
167func (f *fakeAPI) CreateIssue(_ context.Context, r *connect.Request[ermite.CreateIssueRequest]) (*connect.Response[ermite.CreateIssueResponse], error) {
168 return respond(f, "CreateIssue", r, f.createIssue)
169}
170func (f *fakeAPI) ListIssues(_ context.Context, r *connect.Request[ermite.ListIssuesRequest]) (*connect.Response[ermite.ListIssuesResponse], error) {
171 return respond(f, "ListIssues", r, f.listIssues)
172}
173func (f *fakeAPI) GetIssue(_ context.Context, r *connect.Request[ermite.GetIssueRequest]) (*connect.Response[ermite.GetIssueResponse], error) {
174 return respond(f, "GetIssue", r, f.getIssue)
175}
176func (f *fakeAPI) AddIssueComment(_ context.Context, r *connect.Request[ermite.AddIssueCommentRequest]) (*connect.Response[ermite.AddIssueCommentResponse], error) {
177 return respond(f, "AddIssueComment", r, f.addIssueCmt)
178}
179func (f *fakeAPI) CloseIssue(_ context.Context, r *connect.Request[ermite.CloseIssueRequest]) (*connect.Response[ermite.CloseIssueResponse], error) {
180 return respond(f, "CloseIssue", r, f.closeIssue)
181}
182func (f *fakeAPI) ResolveIssue(_ context.Context, r *connect.Request[ermite.ResolveIssueRequest]) (*connect.Response[ermite.ResolveIssueResponse], error) {
183 return respond(f, "ResolveIssue", r, f.resolveIssue)
184}
185func (f *fakeAPI) ReopenIssue(_ context.Context, r *connect.Request[ermite.ReopenIssueRequest]) (*connect.Response[ermite.ReopenIssueResponse], error) {
186 return respond(f, "ReopenIssue", r, f.reopenIssue)
187}
188func (f *fakeAPI) GetSigningKey(_ context.Context, r *connect.Request[ermite.GetSigningKeyRequest]) (*connect.Response[ermite.GetSigningKeyResponse], error) {
189 return respond(f, "GetSigningKey", r, f.getSigningKey)
190}
191func (f *fakeAPI) ListPasskeys(_ context.Context, r *connect.Request[ermite.ListPasskeysRequest]) (*connect.Response[ermite.ListPasskeysResponse], error) {
192 return respond(f, "ListPasskeys", r, f.listPasskeys)
193}
194func (f *fakeAPI) RemovePasskey(_ context.Context, r *connect.Request[ermite.RemovePasskeyRequest]) (*connect.Response[ermite.RemovePasskeyResponse], error) {
195 return respond(f, "RemovePasskey", r, f.removePasskey)
196}
197func (f *fakeAPI) ListTrustedKeys(_ context.Context, r *connect.Request[ermite.ListTrustedKeysRequest]) (*connect.Response[ermite.ListTrustedKeysResponse], error) {
198 return respond(f, "ListTrustedKeys", r, f.listTrustedKeys)
199}
200func (f *fakeAPI) AddTrustedKey(_ context.Context, r *connect.Request[ermite.AddTrustedKeyRequest]) (*connect.Response[ermite.AddTrustedKeyResponse], error) {
201 return respond(f, "AddTrustedKey", r, f.addTrustedKey)
202}
203func (f *fakeAPI) RemoveTrustedKey(_ context.Context, r *connect.Request[ermite.RemoveTrustedKeyRequest]) (*connect.Response[ermite.RemoveTrustedKeyResponse], error) {
204 return respond(f, "RemoveTrustedKey", r, f.removeTrustedKey)
205}
206func (f *fakeAPI) Watch(ctx context.Context, r *connect.Request[ermite.WatchRequest], st *connect.ServerStream[ermite.WatchResponse]) error {
207 f.record("Watch", r.Header(), r.Msg)
208 if f.watch == nil {
209 return connect.NewError(connect.CodeUnimplemented, nil)
210 }
211 return f.watch(ctx, r.Msg, st)
212}
213
214// serve starts the fake under /-/api and returns its base URL.
215func (f *fakeAPI) serve(t *testing.T) string {
216 t.Helper()
217 mux := http.NewServeMux()
218 mount := func(path string, h http.Handler) { mux.Handle("/-/api"+path, http.StripPrefix("/-/api", h)) }
219 p, h := ermiteconnect.NewRepoServiceHandler(f)
220 mount(p, h)
221 p, h = ermiteconnect.NewTokenServiceHandler(f)
222 mount(p, h)
223 p, h = ermiteconnect.NewServerServiceHandler(f)
224 mount(p, h)
225 p, h = ermiteconnect.NewEventServiceHandler(f)
226 mount(p, h)
227 p, h = ermiteconnect.NewMergeRequestServiceHandler(f)
228 mount(p, h)
229 p, h = ermiteconnect.NewIssueServiceHandler(f)
230 mount(p, h)
231 p, h = ermiteconnect.NewSessionServiceHandler(f)
232 mount(p, h)
233 p, h = ermiteconnect.NewRunServiceHandler(f)
234 mount(p, h)
235 p, h = ermiteconnect.NewTrustedKeyServiceHandler(f)
236 mount(p, h)
237 srv := httptest.NewServer(mux)
238 t.Cleanup(srv.Close)
239 return srv.URL
240}
241
242var testNow = time.Date(2026, 10, 10, 12, 0, 0, 0, time.UTC)
243
244// result is the outcome of one CLI run.
245type result struct {
246 stdout, stderr string
247 err error
248}
249
250// testEnv is a cliEnv with the given variables, the fixed clock, and a config
251// file path inside a fresh temp dir (no file yet) so that a test never reads
252// the developer's own ~/.config/ermite/config. token.cmd must not run unless
253// the test sets runTokenCmd.
254func testEnv(t *testing.T, vars map[string]string) *cliEnv {
255 t.Helper()
256 path := filepath.Join(t.TempDir(), "ermite", "config")
257 return &cliEnv{
258 getenv: func(k string) string { return vars[k] },
259 now: func() time.Time { return testNow },
260 version: "v0.0.1",
261 sleep: func(ctx context.Context, _ time.Duration) error { return ctx.Err() },
262 configPath: func() (string, error) { return path, nil },
263 runTokenCmd: func(context.Context, string) (string, error) {
264 t.Error("token.cmd must not run")
265 return "", errors.New("token.cmd must not run")
266 },
267 }
268}
269
270// runCLI runs the command tree in process with the given environment.
271func runCLI(t *testing.T, vars map[string]string, stdin string, args ...string) result {
272 t.Helper()
273 return runCLIWith(t, testEnv(t, vars), stdin, args...)
274}
275
276// runCLIWith runs the command tree in process with env.
277func runCLIWith(t *testing.T, env *cliEnv, stdin string, args ...string) result {
278 t.Helper()
279 root := newRootCmdWith(env)
280 var out, errb bytes.Buffer
281 root.SetOut(&out)
282 root.SetErr(&errb)
283 root.SetIn(strings.NewReader(stdin))
284 root.SetArgs(args)
285 ctx, cancel := context.WithTimeout(t.Context(), 20*time.Second)
286 defer cancel()
287 err := root.ExecuteContext(ctx)
288 return result{stdout: out.String(), stderr: errb.String(), err: err}
289}
290
291func (r result) ok(t *testing.T) result {
292 t.Helper()
293 require.NoError(t, r.err, "stderr: %s", r.stderr)
294 return r
295}
296
297func ts(s string) *timestamppb.Timestamp {
298 v, err := time.Parse(time.RFC3339, s)
299 if err != nil {
300 panic(err)
301 }
302 return timestamppb.New(v)
303}