Upstream plugin prototype

upstream/upstream_test.go

@@ -0,0 +1,86 @@
+package upstream
+
+import (
+	"github.com/miekg/dns"
+	"log"
+	"net"
+	"testing"
+)
+
+func TestDnsUpstream(t *testing.T) {
+
+	u, err := NewDnsUpstream("8.8.8.8:53")
+
+	if err != nil {
+		t.Errorf("cannot create a DNS upstream")
+	}
+
+	testUpstream(t, u)
+}
+
+func TestHttpsUpstream(t *testing.T) {
+
+	testCases := []string{
+		"https://cloudflare-dns.com/dns-query",
+		"https://dns.google.com/experimental",
+		"https://doh.cleanbrowsing.org/doh/security-filter/",
+	}
+
+	for _, url := range testCases {
+		u, err := NewHttpsUpstream(url)
+
+		if err != nil {
+			t.Errorf("cannot create a DNS-over-HTTPS upstream")
+		}
+
+		testUpstream(t, u)
+	}
+}
+
+func TestDnsOverTlsUpstream(t *testing.T) {
+
+	var tests = []struct {
+		endpoint      string
+		tlsServerName string
+	}{
+		{"1.1.1.1:853", ""},
+		{"8.8.8.8:853", ""},
+		{"185.228.168.10:853", "security-filter-dns.cleanbrowsing.org"},
+	}
+
+	for _, test := range tests {
+		u, err := NewDnsOverTlsUpstream(test.endpoint, test.tlsServerName)
+
+		if err != nil {
+			t.Errorf("cannot create a DNS-over-TLS upstream")
+		}
+
+		testUpstream(t, u)
+	}
+}
+
+func testUpstream(t *testing.T, u Upstream) {
+	req := dns.Msg{}
+	req.Id = dns.Id()
+	req.RecursionDesired = true
+	req.Question = []dns.Question{
+		{Name: "google-public-dns-a.google.com.", Qtype: dns.TypeA, Qclass: dns.ClassINET},
+	}
+
+	resp, err := u.Exchange(nil, &req)
+
+	if err != nil {
+		t.Errorf("error while making an upstream request: %s", err)
+	}
+
+	if len(resp.Answer) != 1 {
+		t.Errorf("no answer section in the response")
+	}
+	if answer, ok := resp.Answer[0].(*dns.A); ok {
+		if !net.IPv4(8, 8, 8, 8).Equal(answer.A) {
+			t.Errorf("wrong IP in the response: %v", answer.A)
+		}
+	}
+
+	log.Printf("response: %v", resp)
+}