diff options
Diffstat (limited to 'internal/server/server_test.go')
| -rw-r--r-- | internal/server/server_test.go | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/internal/server/server_test.go b/internal/server/server_test.go index e59a131..42938d1 100644 --- a/internal/server/server_test.go +++ b/internal/server/server_test.go @@ -2,6 +2,7 @@ package server import ( "log/slog" + "context" "testing" "time" @@ -15,7 +16,10 @@ func testServer(t *testing.T) *Server { resolver.WithRootAddresses([]string{"127.0.0.1:1"}), resolver.WithTimeout(50*time.Millisecond), ) - return &Server{logger: slog.Default(), resolver: r} + baseCtx, cancel := context.WithCancel(context.Background()) + + t.Cleanup(cancel) + return &Server{logger: slog.Default(), resolver: r, baseCtx: baseCtx} } func TestBuildResponse(t *testing.T) { @@ -73,7 +77,9 @@ func TestBuildResponseWithQuery(t *testing.T) { } func FuzzBuildResponse(f *testing.F) { - s := &Server{logger: slog.Default()} + baseCtx, cancel := context.WithCancel(context.Background()) + defer cancel() + s := &Server{logger: slog.Default(), baseCtx: baseCtx} // For fuzz, use a resolver that won't make real network calls s.resolver = resolver.New( resolver.WithRootAddresses([]string{"127.0.0.1:1"}), |