Skip to content

Commit 1ebb404

Browse files
MarcoPolosukunrt
authored andcommitted
test: fix failing test (#3141)
1 parent 8f09a3e commit 1ebb404

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

p2p/net/swarm/swarm_dial_test.go

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -399,8 +399,23 @@ func TestBlackHoledAddrBlocked(t *testing.T) {
399399
require.ErrorIs(t, err, ErrDialRefusedBlackHole)
400400
}
401401

402+
type mockDNSResolver struct {
403+
ipsToReturn []net.IPAddr
404+
txtsToReturn []string
405+
}
406+
407+
var _ madns.BasicResolver = (*mockDNSResolver)(nil)
408+
409+
func (m *mockDNSResolver) LookupIPAddr(_ context.Context, _ string) ([]net.IPAddr, error) {
410+
return m.ipsToReturn, nil
411+
}
412+
413+
func (m *mockDNSResolver) LookupTXT(_ context.Context, _ string) ([]string, error) {
414+
return m.txtsToReturn, nil
415+
}
416+
402417
func TestSkipDialingManyDNS(t *testing.T) {
403-
resolver, err := madns.NewResolver()
418+
resolver, err := madns.NewResolver(madns.WithDefaultResolver(&mockDNSResolver{ipsToReturn: []net.IPAddr{{IP: net.ParseIP("1.2.3.4")}, {IP: net.ParseIP("1.2.3.5")}}}))
404419
if err != nil {
405420
t.Fatal(err)
406421
}

0 commit comments

Comments
 (0)