hmm

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"}),