fix all tests

prysmaticlabs · Oct 5, 2020 · a5d6fff · a5d6fff
1 parent b7a0cf5
commit a5d6fff
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/beacon-chain/p2p/fork_test.go b/beacon-chain/p2p/fork_test.go
@@ -32,7 +32,7 @@ func TestStartDiscv5_DifferentForkDigests(t *testing.T) {
 	genesisTime := time.Now()
 	genesisValidatorsRoot := make([]byte, 32)
 	s := &Service{
-		cfg:                   &Config{UDPPort: uint(port)},
+		cfg:                   &Config{UDPPort: uint(port), LocalIP: ipAddr.String()},
 		genesisTime:           genesisTime,
 		genesisValidatorsRoot: genesisValidatorsRoot,
 	}
@@ -51,6 +51,7 @@ func TestStartDiscv5_DifferentForkDigests(t *testing.T) {
 		port = 3000 + i
 		cfg.UDPPort = uint(port)
 		ipAddr, pkey := createAddrAndPrivKey(t)
+		cfg.LocalIP = ipAddr.String()
 
 		// We give every peer a different genesis validators root, which
 		// will cause each peer to have a different ForkDigest, preventing
@@ -116,7 +117,7 @@ func TestStartDiscv5_SameForkDigests_DifferentNextForkData(t *testing.T) {
 	genesisTime := time.Now()
 	genesisValidatorsRoot := make([]byte, 32)
 	s := &Service{
-		cfg:                   &Config{UDPPort: uint(port)},
+		cfg:                   &Config{UDPPort: uint(port), LocalIP: ipAddr.String()},
 		genesisTime:           genesisTime,
 		genesisValidatorsRoot: genesisValidatorsRoot,
 	}
@@ -136,6 +137,7 @@ func TestStartDiscv5_SameForkDigests_DifferentNextForkData(t *testing.T) {
 		port = 3000 + i
 		cfg.UDPPort = uint(port)
 		ipAddr, pkey := createAddrAndPrivKey(t)
+		cfg.LocalIP = ipAddr.String()
 
 		c := params.BeaconConfig()
 		nextForkEpoch := uint64(i)

diff --git a/beacon-chain/p2p/service_test.go b/beacon-chain/p2p/service_test.go
@@ -132,6 +132,7 @@ func TestListenForNewNodes(t *testing.T) {
 	cfg.UDPPort = uint(port)
 	_, pkey := createAddrAndPrivKey(t)
 	ipAddr := net.ParseIP("127.0.0.1")
+	cfg.LocalIP = ipAddr.String()
 	genesisTime := time.Now()
 	genesisValidatorsRoot := make([]byte, 32)
 	s := &Service{
@@ -164,6 +165,7 @@ func TestListenForNewNodes(t *testing.T) {
 		h, pkey, ipAddr := createHost(t, port+i)
 		cfg.UDPPort = uint(port + i)
 		cfg.TCPPort = uint(port + i)
+		cfg.LocalIP = ipAddr.String()
 		s := &Service{
 			cfg:                   cfg,
 			genesisTime:           genesisTime,

diff --git a/beacon-chain/p2p/subnets_test.go b/beacon-chain/p2p/subnets_test.go
@@ -23,7 +23,7 @@ func TestStartDiscV5_DiscoverPeersWithSubnets(t *testing.T) {
 	genesisTime := time.Now()
 	genesisValidatorsRoot := make([]byte, 32)
 	s := &Service{
-		cfg:                   &Config{UDPPort: uint(port)},
+		cfg:                   &Config{UDPPort: uint(port), LocalIP: ipAddr.String()},
 		genesisTime:           genesisTime,
 		genesisValidatorsRoot: genesisValidatorsRoot,
 	}
@@ -47,6 +47,7 @@ func TestStartDiscV5_DiscoverPeersWithSubnets(t *testing.T) {
 			Discv5BootStrapAddr: []string{bootNode.String()},
 			MaxPeers:            30,
 			UDPPort:             uint(port),
+			LocalIP:             ipAddr.String(),
 		}
 		ipAddr, pkey := createAddrAndPrivKey(t)
 		s = &Service{