fake_test.go RawHistoryFolder

1// SPDX-License-Identifier: AGPL-3.0-or-later
2
3package client_test
4
5import (
6	"context"
7	"net/http"
8	"net/http/httptest"
9	"sync"
10	"testing"
11
12	"connectrpc.com/connect"
13
14	"bnema.dev/ermite-api/gen/ermite"
15	"bnema.dev/ermite-api/gen/ermite/ermiteconnect"
16	"bnema.dev/ermite/internal/adapters/client"
17)
18
19// fake is a Connect server with stub services.
20type fake struct {
21	ermiteconnect.UnimplementedServerServiceHandler
22	ermiteconnect.UnimplementedEventServiceHandler
23
24	mu      sync.Mutex
25	headers []http.Header
26	version func(context.Context, *connect.Request[ermite.GetVersionRequest]) (*connect.Response[ermite.GetVersionResponse], error)
27	watch   func(context.Context, *connect.Request[ermite.WatchRequest], *connect.ServerStream[ermite.WatchResponse]) error
28}
29
30func (f *fake) record(h http.Header) {
31	f.mu.Lock()
32	defer f.mu.Unlock()
33	f.headers = append(f.headers, h.Clone())
34}
35
36func (f *fake) GetVersion(ctx context.Context, req *connect.Request[ermite.GetVersionRequest]) (*connect.Response[ermite.GetVersionResponse], error) {
37	f.record(req.Header())
38	return f.version(ctx, req)
39}
40
41func (f *fake) Watch(ctx context.Context, req *connect.Request[ermite.WatchRequest], st *connect.ServerStream[ermite.WatchResponse]) error {
42	f.record(req.Header())
43	return f.watch(ctx, req, st)
44}
45
46// start serves f under /-/api/ and returns a client for it.
47func start(t *testing.T, f *fake, cfg client.Config) *client.Client {
48	t.Helper()
49	mux := http.NewServeMux()
50	p, h := ermiteconnect.NewServerServiceHandler(f)
51	mux.Handle("/-/api"+p, http.StripPrefix("/-/api", h))
52	p, h = ermiteconnect.NewEventServiceHandler(f)
53	mux.Handle("/-/api"+p, http.StripPrefix("/-/api", h))
54	srv := httptest.NewServer(mux)
55	t.Cleanup(srv.Close)
56	cfg.URL = srv.URL
57	c, err := client.New(cfg)
58	if err != nil {
59		t.Fatal(err)
60	}
61	return c
62}