From 2c61900dd5efd81a5351513a20fa65580c8a6616 Mon Sep 17 00:00:00 2001 From: radhitya Date: Tue, 23 Jun 2026 05:36:25 +0700 Subject: make if error for unsupported query type --- internal/server/server_test.go | 23 ++++++++++++++++++++--- 1 file changed, 20 insertions(+), 3 deletions(-) (limited to 'internal/server/server_test.go') diff --git a/internal/server/server_test.go b/internal/server/server_test.go index 2340acd..93e2a69 100644 --- a/internal/server/server_test.go +++ b/internal/server/server_test.go @@ -1,15 +1,32 @@ package server import ( + "codeberg.org/miekg/dns" + "codeberg.org/miekg/dns/rdata" "context" + "linum/internal/resolver" "log/slog" "testing" "time" - - "codeberg.org/miekg/dns" - "linum/internal/resolver" ) +func TestBuildResponseUnknownType(t *testing.T) { + s := testServer(t) + m := new(dns.Msg) + m.Question = []dns.RR{&dns.RFC3597{ + Hdr: dns.Header{Name: "example.com.", Class: dns.ClassINET}, + RFC3597: rdata.RFC3597{RRType: 0xaa58}, + }} + + resp, _ := s.buildResponse(m) + if resp == nil { + t.Fatal("buildResponse returned nil") + } + if resp.Rcode != dns.RcodeNotImplemented { + t.Errorf("expected NOTIMPL, got %d", resp.Rcode) + } +} + func testServer(t *testing.T) *Server { t.Helper() r := resolver.New( -- cgit v1.2.3