fix: Pass transport protocol as an argument

internal/pkg/service/stream/storage/level/local/diskwriter/network/connection/connection.go

@@ -63,13 +63,15 @@ func NewManager(d dependencies, cfg network.Config, nodeID string) (*Manager, er
 		nodeID: nodeID,
 	}
 
-	// Create transport client
-	var err error
-	m.client, err = transport.NewClient(m.logger.WithComponent("client"), cfg, nodeID)
+	// Create transport
+	tr, err := transport.NewProtocol(cfg)
 	if err != nil {
 		return nil, err
 	}
 
+	// Create transport client
+	m.client = transport.NewClient(m.logger.WithComponent("client"), cfg, nodeID, tr)
+
 	// Graceful shutdown
 	wg := &sync.WaitGroup{}
 	ctx, cancel := context.WithCancel(context.Background())

internal/pkg/service/stream/storage/level/local/diskwriter/network/transport/client.go

@@ -25,20 +25,15 @@ type Client struct {
 	connections map[string]*ClientConnection
 }
 
-func NewClient(logger log.Logger, config network.Config, nodeID string) (*Client, error) {
-	transport, err := NewProtocol(config)
-	if err != nil {
-		return nil, err
-	}
-
+func NewClient(logger log.Logger, config network.Config, nodeID string, transport Protocol) *Client {
 	return &Client{
 		logger:      logger.WithComponent("transport"),
 		config:      config,
 		nodeID:      nodeID,
 		transport:   transport,
 		closed:      make(chan struct{}),
 		connections: make(map[string]*ClientConnection),
-	}, nil
+	}
 }
 
 // OpenConnection starts a connection dial loop to the target address.

internal/pkg/service/stream/storage/level/local/diskwriter/network/transport/protocol_test.go

@@ -76,8 +76,7 @@ func testTransportSmallData(t *testing.T, transportFactory func(cfg network.Conf
 	// Setup client
 	clientLogger := log.NewDebugLogger()
 	clientLogger.ConnectTo(os.Stdout)
-	client, err := transport.NewClient(clientLogger, cfg, "client-node")
-	require.NoError(t, err)
+	client := transport.NewClient(clientLogger, cfg, "client-node", tr)
 	conn, err := client.OpenConnectionOrErr(ctx, "srv", addr)
 	require.NoError(t, err)
 
@@ -164,8 +163,7 @@ func testTransportBiggerData(t *testing.T, transportFactory func(cfg network.Con
 
 	// Setup client
 	clientLogger := log.NewDebugLogger()
-	client, err := transport.NewClient(clientLogger, cfg, "client-node")
-	require.NoError(t, err)
+	client := transport.NewClient(clientLogger, cfg, "client-node", tr)
 	conn, err := client.OpenConnectionOrErr(ctx, "srv", addr)
 	require.NoError(t, err)