From 55d452e8cc8e782345ab36f2f5e57a45068067f1 Mon Sep 17 00:00:00 2001 From: radhitya Date: Sun, 14 Jun 2026 18:49:03 +0700 Subject: hmm --- internal/server/server_test.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'internal/server/server_test.go') 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"}), -- cgit v1.2.3