test: refactor common mock DNS test code

Author: bryanchriswhite

internal/testutil/mockdns.go

@@ -0,0 +1,51 @@
+package testutil
+
+import (
+	"fmt"
+	"net"
+	"net/url"
+	"testing"
+
+	"github.com/foxcpp/go-mockdns"
+	"github.com/stretchr/testify/require"
+)
+
+func PrepareDNSMockFromServiceURLs(t *testing.T, serviceURLs []string) (done func()) {
+	zones := make(map[string]mockdns.Zone)
+	for i, u := range serviceURLs {
+		// Perpend `scheme://` as serviceURLs are currently scheme-less.
+		// Required for parsing to produce useful results.
+		// (see: https://pkg.go.dev/net/url@go1.20.2#URL)
+		serviceURL, err := url.Parse(fmt.Sprintf("scheme://%s", u))
+		require.NoError(t, err)
+
+		ipStr := fmt.Sprintf("10.0.0.%d", i+1)
+
+		if i >= 254 {
+			panic(fmt.Sprintf("would generate invalid IPv4 address: %s", ipStr))
+		}
+
+		zones[fmt.Sprintf("%s.", serviceURL.Hostname())] = mockdns.Zone{
+			A: []string{ipStr},
+		}
+	}
+
+	return PrepareDNSMock(zones)
+}
+
+func PrepareDNSMock(zones map[string]mockdns.Zone) (done func()) {
+	srv, _ := mockdns.NewServerWithLogger(zones, noopLogger{}, false)
+	srv.PatchNet(net.DefaultResolver)
+	return func() {
+		_ = srv.Close()
+		mockdns.UnpatchNet(net.DefaultResolver)
+	}
+}
+
+// NB: default logging behavior is too noisy.
+// noopLogger implements go-mockdns's `mockdns.Logger` interface.
+type noopLogger struct{}
+
+func (nl noopLogger) Printf(format string, args ...interface{}) {
+	// noop
+}

p2p/module_raintree_test.go

@@ -16,6 +16,7 @@ import (
 	"github.com/stretchr/testify/require"
 	"google.golang.org/protobuf/types/known/anypb"
 
+	"github.com/pokt-network/pocket/internal/testutil"
 	"github.com/pokt-network/pocket/p2p/protocol"
 )
 
@@ -233,7 +234,7 @@ func testRainTreeCalls(t *testing.T, origNode string, networkSimulationConfig Te
 		return iId < jId
 	})
 
-	prepareDNSResolverMock(t, valIds)
+	testutil.PrepareDNSMockFromServiceURLs(t, valIds)
 
 	// Create connection and bus mocks along with a shared WaitGroup to track the number of expected
 	// reads and writes throughout the mocked local network

p2p/raintree/peers_manager_test.go

@@ -3,7 +3,6 @@ package raintree
 import (
 	"encoding/hex"
 	"fmt"
-	"net"
 	"net/url"
 	"strings"
 	"testing"
@@ -14,6 +13,7 @@ import (
 	mocknet "github.com/libp2p/go-libp2p/p2p/net/mock"
 	"github.com/stretchr/testify/require"
 
+	"github.com/pokt-network/pocket/internal/testutil"
 	typesP2P "github.com/pokt-network/pocket/p2p/types"
 	mocksP2P "github.com/pokt-network/pocket/p2p/types/mocks"
 	"github.com/pokt-network/pocket/runtime/configs"
@@ -340,22 +340,6 @@ func mockAlphabetValidatorServiceURLsDNS(t *testing.T) (done func()) {
 			A: []string{fmt.Sprintf("10.0.0.%d", i+1)},
 		}
 	}
-	return prepareDNSMock(zones)
-}
-
-// TECHDEBT(#609): de-duplicate / refactor `prepaand reDNSMock` & `noopLogger`.
-func prepareDNSMock(zones map[string]mockdns.Zone) (done func()) {
-	srv, _ := mockdns.NewServerWithLogger(zones, noopLogger{}, false)
-	srv.PatchNet(net.DefaultResolver)
-	return func() {
-		_ = srv.Close()
-		mockdns.UnpatchNet(net.DefaultResolver)
-	}
-}
-
-// noopLogger implements go-mockdns's `mockdns.Logger` interface.
-type noopLogger struct{}
 
-func (nl noopLogger) Printf(format string, args ...interface{}) {
-	// noop
+	return testutil.PrepareDNSMock(zones)
 }

p2p/utils_test.go

@@ -4,14 +4,12 @@ import (
 	"fmt"
 	"log"
 	"net"
-	"net/url"
 	"sort"
 	"strconv"
 	"sync"
 	"testing"
 	"time"
 
-	"github.com/foxcpp/go-mockdns"
 	"github.com/golang/mock/gomock"
 	libp2pCrypto "github.com/libp2p/go-libp2p/core/crypto"
 	libp2pPeer "github.com/libp2p/go-libp2p/core/peer"
@@ -144,45 +142,6 @@ func setupMockNetPeers(t *testing.T, netMock mocknet.Mocknet, numPeers int) (pee
 	return peerIDs
 }
 
-// TECHDEBT(#609): this is one of a few places where we could de-duplicate test
-// code if we had a conventional place to store packages intended for import
-// only into tests.
-func prepareDNSResolverMock(t *testing.T, serviceURLs []string) (done func()) {
-	zones := make(map[string]mockdns.Zone)
-	for i, u := range serviceURLs {
-		// Perpend `scheme://` as serviceURLs are currently scheme-less.
-		// Required for parsing to produce useful results.
-		// (see: https://pkg.go.dev/net/url@go1.20.2#URL)
-		serviceURL, err := url.Parse(fmt.Sprintf("scheme://%s", u))
-		require.NoError(t, err)
-
-		ipStr := fmt.Sprintf("10.0.0.%d", i+1)
-
-		if i >= 254 {
-			panic(fmt.Sprintf("would generate invalid IPv4 address: %s", ipStr))
-		}
-
-		zones[fmt.Sprintf("%s.", serviceURL.Hostname())] = mockdns.Zone{
-			A: []string{ipStr},
-		}
-	}
-
-	srv, _ := mockdns.NewServerWithLogger(zones, noopLogger{}, false)
-	srv.PatchNet(net.DefaultResolver)
-	return func() {
-		_ = srv.Close()
-		mockdns.UnpatchNet(net.DefaultResolver)
-	}
-}
-
-// NB: default logging behavior is too noisy.
-// noopLogger implements go-mockdns's `mockdns.Logger` interface.
-type noopLogger struct{}
-
-func (nl noopLogger) Printf(format string, args ...interface{}) {
-	// noop
-}
-
 // createMockRuntimeMgrs creates `numValidators` instances of mocked `RuntimeMgr` that are essentially
 // representing the runtime environments of the validators that we will use in our tests
 func createMockRuntimeMgrs(t *testing.T, numValidators int) []modules.RuntimeMgr {