Add test case for domain not found

Author: varunsh-coder

dnsproxy.go

@@ -53,7 +53,6 @@ func (proxy *DNSProxy) getResponse(requestMsg *dns.Msg) (*dns.Msg, error) {
 
 			answer, err := proxy.processTypeA(&question, requestMsg)
 			if err != nil {
-
 				return responseMsg, err
 			}
 			responseMsg.Answer = append(responseMsg.Answer, *answer)

dnsproxy_test.go

@@ -35,6 +35,9 @@ func TestDNSProxy_getResponse(t *testing.T) {
 	httpmock.RegisterResponder("GET", "https://dns.google/resolve?name=allowed.com.&type=a",
 		httpmock.NewStringResponder(200, `{"Status":0,"TC":false,"RD":true,"RA":true,"AD":false,"CD":false,"Question":[{"name":"allowed.com.","type":1}],"Answer":[{"name":"allowed.com.","type":1,"TTL":3080,"data":"67.225.146.248"}]}`))
 
+	httpmock.RegisterResponder("GET", "https://dns.google/resolve?name=notfound.com.&type=a",
+		httpmock.NewStringResponder(200, `{"Status":3,"TC":false,"RD":true,"RA":true,"AD":false,"CD":false,"Question":[{"name":"notfound.com.","type":1}],"Authority":[{"name":"com.","type":6,"TTL":900,"data":"a.gtld-servers.net. nstld.verisign-grs.com. 1640040308 1800 900 604800 86400"}],"Comment":"Response from 2001:503:231d::2:30."}`))
+
 	tests := []struct {
 		name    string
 		fields  fields
@@ -78,6 +81,12 @@ func TestDNSProxy_getResponse(t *testing.T) {
 			want:    &dns.Msg{Answer: []dns.RR{rrDnsAllowed}},
 			wantErr: false,
 		},
+		{name: "type A notfound.com",
+			fields:  fields{Cache: &Cache, EgressPolicy: EgressPolicyAudit},
+			args:    args{requestMsg: &dns.Msg{Question: []dns.Question{{Name: "notfound.com.", Qtype: dns.TypeA}}}},
+			want:    &dns.Msg{},
+			wantErr: true,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {