make if error for unsupported query type

1 files changed, 17 insertions, 4 deletions

diff --git a/internal/resolver/resolver_test.go b/internal/resolver/resolver_test.go
index daa8a98..3c44cb4 100644
--- a/internal/resolver/resolver_test.go
+++ b/internal/resolver/resolver_test.go
@@ -1,17 +1,30 @@
 package resolver
 
 import (
+	"codeberg.org/miekg/dns"
+	"codeberg.org/miekg/dns/dnsutil"
+	"codeberg.org/miekg/dns/rdata"
 	"context"
+	"errors"
 	"io"
 	"net/netip"
 	"testing"
 	"time"
-
-	"codeberg.org/miekg/dns"
-	"codeberg.org/miekg/dns/dnsutil"
-	"codeberg.org/miekg/dns/rdata"
 )
 
+func TestLookupUnknownType(t *testing.T) {
+	r := New(WithForwarders([]string{"127.0.0.1:1"}), WithTimeout(50*time.Millisecond))
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+
+	_, err := r.Lookup(ctx, "example.com.", 0xaa58)
+	if err == nil {
+		t.Fatal("expected error for unsupported qtype")
+	}
+	if !errors.Is(err, ErrUnsupportedType) {
+		t.Fatalf("expected ErrUnsupportedType, got %v", err)
+	}
+}
 func startTestServer(t *testing.T, addr string, handler dns.Handler) *dns.Server {
 	t.Helper()
 	ready := make(chan struct{})