make if error for unsupported query type

1 files changed, 20 insertions, 3 deletions

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(