Skip to content

Commit d8291f6

Browse files
committed
test(admin): update tests for mandatory admin auth
1 parent 32bfd6f commit d8291f6

5 files changed

Lines changed: 16 additions & 3 deletions

File tree

internal/admin/resource_api_test.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,10 @@ func newTestServerWithResourceManagerAndLoggerAndOptions(
5656
) *Server {
5757
t.Helper()
5858

59+
if !authConfigured(opts) {
60+
opts.BearerToken = testAuthToken
61+
}
62+
5963
store := ir.NewSnapshotStore(logger)
6064
nodes := nodeinfo.NewRegistry(
6165
ir.NewNodeStatusStore(),
@@ -325,6 +329,7 @@ func performRequestWithBody(
325329
t.Helper()
326330

327331
req := httptest.NewRequest(method, path, bytes.NewReader(body))
332+
req.Header.Set("Authorization", "Bearer "+testAuthToken)
328333
recorder := httptest.NewRecorder()
329334
server.server.Handler.ServeHTTP(recorder, req)
330335

internal/admin/server_auth_metrics_test.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,7 @@ func TestAdminRateLimiterUsesBurstAndRecords429Metrics(t *testing.T) {
154154
})
155155

156156
req := httptest.NewRequest(http.MethodGet, "/v1/summary", nil)
157+
req.Header.Set("Authorization", "Bearer "+testAuthToken)
157158
req.RemoteAddr = "203.0.113.10:12345"
158159
recorder := httptest.NewRecorder()
159160
server.server.Handler.ServeHTTP(recorder, req)

internal/admin/server_backend_node_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,7 @@ func TestNodeEndpointsFallbackToSharedLeaseState(t *testing.T) {
425425
nodeinfo.NewRegistry(ir.NewNodeStatusStore(), nil, logger, nodeinfo.Options{PersistTimeout: time.Second}),
426426
NewResourceManager(client, logger),
427427
logger,
428-
Options{},
428+
Options{BearerToken: testAuthToken},
429429
)
430430

431431
var nodes []ir.NodeStatus
@@ -516,7 +516,7 @@ func TestNodeEndpointsIgnoreStaleSharedLeaseState(t *testing.T) {
516516
nodeinfo.NewRegistry(ir.NewNodeStatusStore(), nil, logger, nodeinfo.Options{PersistTimeout: time.Second}),
517517
NewResourceManager(client, logger),
518518
logger,
519-
Options{},
519+
Options{BearerToken: testAuthToken},
520520
)
521521

522522
var nodes []ir.NodeStatus

internal/admin/server_chatbot_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ func newChatbotTestServer(t *testing.T, k8sObjects ...client.Object) *Server {
5252
nodes,
5353
NewResourceManager(k8sClient, logger),
5454
logger,
55-
Options{},
55+
Options{BearerToken: testAuthToken},
5656
)
5757
}
5858

internal/admin/server_test.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ import (
3131
"github.com/nantian-gw/gateway/internal/nodeinfo"
3232
)
3333

34+
const testAuthToken = "test-admin-token"
35+
3436
func TestNewServerAppliesRuntimeTimeouts(t *testing.T) {
3537
t.Parallel()
3638

@@ -251,6 +253,10 @@ func newTestServerWithOptions(t *testing.T, opts Options) *Server {
251253
func newTestServerWithRepository(t *testing.T, repo nodeinfo.Repository, opts Options) *Server {
252254
t.Helper()
253255

256+
if !authConfigured(opts) {
257+
opts.BearerToken = testAuthToken
258+
}
259+
254260
logger := slog.New(slog.NewTextHandler(io.Discard, nil))
255261
store := ir.NewSnapshotStore(logger)
256262
nodes := nodeinfo.NewRegistry(
@@ -432,6 +438,7 @@ func performRequest(t *testing.T, server *Server, method, path string, target an
432438
t.Helper()
433439

434440
req := httptest.NewRequest(method, path, nil)
441+
req.Header.Set("Authorization", "Bearer "+testAuthToken)
435442
recorder := httptest.NewRecorder()
436443
server.server.Handler.ServeHTTP(recorder, req)
437444

0 commit comments

Comments
 (0)