cleanup: test fixup after refactor

This commit is contained in:
William Banfield
2022-10-24 16:16:28 -04:00
parent d4fcfa127b
commit 60373be5ed
3 changed files with 9 additions and 5 deletions

View File

@@ -4,6 +4,8 @@ import (
"fmt"
golog "log"
"net"
"reflect"
"sync"
"testing"
"time"
@@ -98,8 +100,12 @@ func createOutboundPeerAndPerformHandshake(
if err != nil {
return nil, err
}
mlc := &metricsLabelCache{
mtx: &sync.RWMutex{},
messageLabelNames: make(map[reflect.Type]string),
}
p := newPeer(pc, mConfig, peerNodeInfo, reactorsByCh, msgTypeByChID, chDescs, func(p Peer, r interface{}) {})
p := newPeer(pc, mConfig, peerNodeInfo, reactorsByCh, msgTypeByChID, chDescs, func(p Peer, r interface{}) {}, mlc)
p.SetLogger(log.TestingLogger().With("peer", addr))
return p, nil
}

View File

@@ -697,8 +697,7 @@ func TestPexVectors(t *testing.T) {
for _, tc := range testCases {
tc := tc
w, err := tc.msg.(p2p.Wrapper).Wrap()
require.NoError(t, err)
w := tc.msg.(p2p.Wrapper).Wrap()
bz, err := proto.Marshal(w)
require.NoError(t, err)

View File

@@ -100,8 +100,7 @@ func TestStateSyncVectors(t *testing.T) {
for _, tc := range testCases {
tc := tc
w, err := tc.msg.(p2p.Wrapper).Wrap()
require.NoError(t, err)
w := tc.msg.(p2p.Wrapper).Wrap()
bz, err := proto.Marshal(w)
require.NoError(t, err)