initial

2026-06-25 21:37:31 +02:00 · 2026-06-05 03:56:25 +00:00
parent ff6e45fae0
commit aa781c6cc5
12 changed files with 599 additions and 0 deletions
@@ -0,0 +1,82 @@
+package restrictednet
+
+import (
+	"net"
+	"net/netip"
+	"testing"
+
+	"github.com/miekg/dns"
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_answersToNetipAddrs(t *testing.T) {
+	t.Parallel()
+
+	testCases := map[string]struct {
+		message    *dns.Msg
+		expected   []netip.Addr
+		errorIsNil bool
+	}{
+		"nil_message": {
+			message:    nil,
+			expected:   nil,
+			errorIsNil: true,
+		},
+		"no_answers": {
+			message:    &dns.Msg{},
+			expected:   []netip.Addr{},
+			errorIsNil: true,
+		},
+		"a_record": {
+			message: &dns.Msg{
+				Answer: []dns.RR{
+					&dns.A{
+						Hdr: dns.RR_Header{Name: "example.com.", Rrtype: dns.TypeA, Class: dns.ClassINET},
+						A:   net.IP{1, 1, 1, 1},
+					},
+				},
+			},
+			expected:   []netip.Addr{netip.MustParseAddr("1.1.1.1")},
+			errorIsNil: true,
+		},
+		"aaaa_record": {
+			message: &dns.Msg{
+				Answer: []dns.RR{
+					&dns.AAAA{
+						Hdr:  dns.RR_Header{Name: "example.com.", Rrtype: dns.TypeAAAA, Class: dns.ClassINET},
+						AAAA: net.IP{0x20, 0x01, 0x48, 0x60, 0x48, 0x60, 0, 0, 0, 0, 0, 0, 0, 0, 0x88, 0x88},
+					},
+				},
+			},
+			expected:   []netip.Addr{netip.MustParseAddr("2001:4860:4860::8888")},
+			errorIsNil: true,
+		},
+		"mixed_records": {
+			message: &dns.Msg{
+				Answer: []dns.RR{
+					&dns.A{
+						Hdr: dns.RR_Header{Name: "example.com.", Rrtype: dns.TypeA, Class: dns.ClassINET},
+						A:   net.IP{1, 1, 1, 1},
+					},
+					&dns.AAAA{
+						Hdr:  dns.RR_Header{Name: "example.com.", Rrtype: dns.TypeAAAA, Class: dns.ClassINET},
+						AAAA: net.IP{0x20, 0x01, 0x48, 0x60, 0x48, 0x60, 0, 0, 0, 0, 0, 0, 0, 0, 0x88, 0x88},
+					},
+				},
+			},
+			expected:   []netip.Addr{netip.MustParseAddr("1.1.1.1"), netip.MustParseAddr("2001:4860:4860::8888")},
+			errorIsNil: true,
+		},
+	}
+
+	for testName, testCase := range testCases {
+		testCase := testCase
+		t.Run(testName, func(t *testing.T) {
+			t.Parallel()
+
+			addresses := answersToNetipAddrs(testCase.message)
+
+			assert.Equal(t, testCase.expected, addresses)
+		})
+	}
+}