(t *testing.T, store *enrolltokens.MemoryStore, nodeIDs ...string)
| 15 | ) |
| 16 | |
| 17 | func newEnrollTokenTestMux(t *testing.T, store *enrolltokens.MemoryStore, nodeIDs ...string) *http.ServeMux { |
| 18 | t.Helper() |
| 19 | ns := nodes.NewMemoryStore() |
| 20 | for _, id := range nodeIDs { |
| 21 | if _, err := ns.Upsert(nodes.Node{ID: id, Name: id}); err != nil { |
| 22 | t.Fatalf("seed node %q: %v", id, err) |
| 23 | } |
| 24 | } |
| 25 | mux := http.NewServeMux() |
| 26 | RegisterNodeEnrollTokenEndpoint(mux, ns, store, func(*http.Request) string { |
| 27 | return "https://panel.example.com" |
| 28 | }) |
| 29 | return mux |
| 30 | } |
| 31 | |
| 32 | func doEnrollToken(t *testing.T, mux *http.ServeMux, nodeID string, body any) *httptest.ResponseRecorder { |
| 33 | t.Helper() |
no test coverage detected