Skip to content

Commit 388b9e2

Browse files
committed
update all node fixtures to use inspector builder
1 parent 9963de0 commit 388b9e2

File tree

3 files changed

+7
-31
lines changed

3 files changed

+7
-31
lines changed

network/internal/p2pfixtures/fixtures.go

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -32,12 +32,9 @@ import (
3232
"github.com/onflow/flow-go/network/p2p"
3333
p2pdht "github.com/onflow/flow-go/network/p2p/dht"
3434
"github.com/onflow/flow-go/network/p2p/distributor"
35-
"github.com/onflow/flow-go/network/p2p/inspector"
36-
"github.com/onflow/flow-go/network/p2p/inspector/validation"
3735
"github.com/onflow/flow-go/network/p2p/keyutils"
3836
"github.com/onflow/flow-go/network/p2p/p2pbuilder"
3937
inspectorbuilder "github.com/onflow/flow-go/network/p2p/p2pbuilder/inspector"
40-
"github.com/onflow/flow-go/network/p2p/p2pnode"
4138
"github.com/onflow/flow-go/network/p2p/tracer"
4239
"github.com/onflow/flow-go/network/p2p/unicast"
4340
"github.com/onflow/flow-go/network/p2p/unicast/protocols"
@@ -110,14 +107,8 @@ func CreateNode(t *testing.T, networkKey crypto.PrivateKey, sporkID flow.Identif
110107
idProvider,
111108
p2pbuilder.DefaultGossipSubConfig().LocalMeshLogInterval)
112109

113-
defaultRPCValidationInpectorCfg := inspectorbuilder.DefaultRPCValidationConfig()
114-
rpcInspectorNotifDistributor := distributor.DefaultGossipSubInspectorNotificationDistributor(logger)
115-
116-
gossipSubMetrics := p2pnode.NewGossipSubControlMessageMetrics(metrics.NewNoopCollector(), logger)
117-
rpcInspectors := []p2p.GossipSubRPCInspector{
118-
inspector.NewControlMsgMetricsInspector(logger, gossipSubMetrics, inspector.DefaultControlMsgMetricsInspectorNumberOfWorkers),
119-
validation.NewControlMsgValidationInspector(logger, sporkID, defaultRPCValidationInpectorCfg, rpcInspectorNotifDistributor),
120-
}
110+
rpcInspectors, err := inspectorbuilder.NewGossipSubInspectorBuilder(logger, sporkID, inspectorbuilder.DefaultGossipSubRPCInspectorsConfig(), distributor.DefaultGossipSubInspectorNotificationDistributor(logger)).Build()
111+
require.NoError(t, err)
121112

122113
builder := p2pbuilder.NewNodeBuilder(
123114
logger,

network/internal/testutils/testUtil.go

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,9 @@ import (
3838
"github.com/onflow/flow-go/network/p2p/connection"
3939
p2pdht "github.com/onflow/flow-go/network/p2p/dht"
4040
"github.com/onflow/flow-go/network/p2p/distributor"
41-
"github.com/onflow/flow-go/network/p2p/inspector"
42-
"github.com/onflow/flow-go/network/p2p/inspector/validation"
4341
"github.com/onflow/flow-go/network/p2p/middleware"
4442
"github.com/onflow/flow-go/network/p2p/p2pbuilder"
4543
inspectorbuilder "github.com/onflow/flow-go/network/p2p/p2pbuilder/inspector"
46-
"github.com/onflow/flow-go/network/p2p/p2pnode"
4744
"github.com/onflow/flow-go/network/p2p/subscription"
4845
"github.com/onflow/flow-go/network/p2p/translator"
4946
"github.com/onflow/flow-go/network/p2p/unicast"
@@ -452,14 +449,8 @@ func generateLibP2PNode(t *testing.T,
452449
connManager, err := NewTagWatchingConnManager(logger, noopMetrics, connection.DefaultConnManagerConfig())
453450
require.NoError(t, err)
454451

455-
defaultRPCValidationInpectorCfg := inspectorbuilder.DefaultRPCValidationConfig()
456-
rpcInspectorNotifDistributor := distributor.DefaultGossipSubInspectorNotificationDistributor(logger)
457-
458-
gossipSubMetrics := p2pnode.NewGossipSubControlMessageMetrics(metrics.NewNoopCollector(), logger)
459-
rpcInspectors := []p2p.GossipSubRPCInspector{
460-
inspector.NewControlMsgMetricsInspector(logger, gossipSubMetrics, inspector.DefaultControlMsgMetricsInspectorNumberOfWorkers),
461-
validation.NewControlMsgValidationInspector(logger, sporkID, defaultRPCValidationInpectorCfg, rpcInspectorNotifDistributor),
462-
}
452+
rpcInspectors, err := inspectorbuilder.NewGossipSubInspectorBuilder(logger, sporkID, inspectorbuilder.DefaultGossipSubRPCInspectorsConfig(), distributor.DefaultGossipSubInspectorNotificationDistributor(logger)).Build()
453+
require.NoError(t, err)
463454

464455
builder := p2pbuilder.NewNodeBuilder(
465456
logger,

network/p2p/test/fixtures.go

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,8 @@ import (
3030
"github.com/onflow/flow-go/network/p2p/connection"
3131
p2pdht "github.com/onflow/flow-go/network/p2p/dht"
3232
"github.com/onflow/flow-go/network/p2p/distributor"
33-
"github.com/onflow/flow-go/network/p2p/inspector"
34-
"github.com/onflow/flow-go/network/p2p/inspector/validation"
3533
"github.com/onflow/flow-go/network/p2p/p2pbuilder"
3634
inspectorbuilder "github.com/onflow/flow-go/network/p2p/p2pbuilder/inspector"
37-
"github.com/onflow/flow-go/network/p2p/p2pnode"
3835
"github.com/onflow/flow-go/network/p2p/unicast"
3936
"github.com/onflow/flow-go/network/p2p/unicast/protocols"
4037
"github.com/onflow/flow-go/network/p2p/utils"
@@ -61,11 +58,8 @@ func NodeFixture(
6158
) (p2p.LibP2PNode, flow.Identity) {
6259
// default parameters
6360
logger := unittest.Logger().Level(zerolog.ErrorLevel)
64-
defaultRPCValidationInpectorCfg := inspectorbuilder.DefaultRPCValidationConfig()
65-
rpcInspectorNotifDistributor := distributor.DefaultGossipSubInspectorNotificationDistributor(logger)
66-
gossipSubMetrics := p2pnode.NewGossipSubControlMessageMetrics(metrics.NewNoopCollector(), logger)
67-
metricsInspector := inspector.NewControlMsgMetricsInspector(logger, gossipSubMetrics, inspector.DefaultControlMsgMetricsInspectorNumberOfWorkers)
68-
validationInspector := validation.NewControlMsgValidationInspector(logger, sporkID, defaultRPCValidationInpectorCfg, rpcInspectorNotifDistributor)
61+
rpcInspectors, err := inspectorbuilder.NewGossipSubInspectorBuilder(logger, sporkID, inspectorbuilder.DefaultGossipSubRPCInspectorsConfig(), distributor.DefaultGossipSubInspectorNotificationDistributor(logger)).Build()
62+
require.NoError(t, err)
6963
parameters := &NodeFixtureParameters{
7064
HandlerFunc: func(network.Stream) {},
7165
Unicasts: nil,
@@ -77,7 +71,7 @@ func NodeFixture(
7771
Metrics: metrics.NewNoopCollector(),
7872
ResourceManager: testutils.NewResourceManager(t),
7973
GossipSubPeerScoreTracerInterval: 0, // disabled by default
80-
GossipSubRPCInspectors: []p2p.GossipSubRPCInspector{metricsInspector, validationInspector},
74+
GossipSubRPCInspectors: rpcInspectors,
8175
}
8276

8377
for _, opt := range opts {

0 commit comments

Comments
 (0)