nit

Signed-off-by: Joshua Kim <[email protected]>
ava-labs · Dec 12, 2023 · e314988 · e314988
1 parent 23f90d4
commit e314988
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/network/p2p/network_test.go b/network/p2p/network_test.go
@@ -58,8 +58,10 @@ func TestMessageRouting(t *testing.T) {
 		SentAppRequest:           make(chan []byte, 1),
 		SentCrossChainAppRequest: make(chan []byte, 1),
 	}
-	network := NewNetwork(logging.NoLog{}, sender, prometheus.NewRegistry(), "")
-	client, err := network.NewAppProtocol(1, testHandler)
+	network, err := NewNetwork(logging.NoLog{}, sender, prometheus.NewRegistry(), "")
+	require.NoError(err)
+	require.NoError(network.AddHandler(1, testHandler))
+	client, err := network.NewClient(1)
 	require.NoError(err)
 
 	require.NoError(client.AppGossip(ctx, wantMsg))