diff options
Diffstat (limited to 'internal/server/server_test.go')
| -rw-r--r-- | internal/server/server_test.go | 23 |
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( |