dnsforward: imp code

annguyen0 · Apr 17, 2024 · a79f98f · a79f98f
1 parent b901a11
commit a79f98f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/dnsforward/dnsforward_test.go b/internal/dnsforward/dnsforward_test.go
@@ -583,12 +583,12 @@ func TestSafeSearch(t *testing.T) {
 				cname := testutil.RequireTypeAssert[*dns.CNAME](t, reply.Answer[0])
 				assert.Equal(t, tc.wantCNAME, cname.Target)
 
-				a := testutil.RequireTypeAssert[*dns.A](t, reply.Answer[len(reply.Answer)-1])
+				a := testutil.RequireTypeAssert[*dns.A](t, reply.Answer[1])
 				assert.NotEmpty(t, a.A)
 			} else {
 				require.Len(t, reply.Answer, 1)
 
-				a := testutil.RequireTypeAssert[*dns.A](t, reply.Answer[len(reply.Answer)-1])
+				a := testutil.RequireTypeAssert[*dns.A](t, reply.Answer[0])
 				assert.Equal(t, net.IP(tc.want.AsSlice()), a.A)
 			}
 		})