[CORE-715] fix bridge client tests and add test on empty eth rpc endp…

…oint (#728) (#729) * fix bridge client tests and add test on empty eth rpc endpoint (cherry picked from commit 4d60e8d) Co-authored-by: Tian <[email protected]>
dydxprotocol · Oct 31, 2023 · be49763 · be49763
1 parent d08c8e0
commit be49763
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 3 deletions.
diff --git a/protocol/daemons/bridge/client/client.go b/protocol/daemons/bridge/client/client.go
@@ -36,6 +36,7 @@ func Start(
 		"Starting bridge daemon with flags",
 		"BridgeFlags", flags.Bridge,
 	)
+
 	// Panic if EthRpcEndpoint is empty.
 	if flags.Bridge.EthRpcEndpoint == "" {
 		return fmt.Errorf("flag %s is not set", daemonflags.FlagBridgeDaemonEthRpcEndpoint)

diff --git a/protocol/daemons/bridge/client/client_test.go b/protocol/daemons/bridge/client/client_test.go
@@ -10,7 +10,7 @@ import (
 	appflags "github.com/dydxprotocol/v4-chain/protocol/app/flags"
 	"github.com/dydxprotocol/v4-chain/protocol/daemons/bridge/client"
 	d_constants "github.com/dydxprotocol/v4-chain/protocol/daemons/constants"
-	"github.com/dydxprotocol/v4-chain/protocol/daemons/flags"
+	daemonflags "github.com/dydxprotocol/v4-chain/protocol/daemons/flags"
 	"github.com/dydxprotocol/v4-chain/protocol/mocks"
 	"github.com/dydxprotocol/v4-chain/protocol/testutil/appoptions"
 	"github.com/dydxprotocol/v4-chain/protocol/testutil/constants"
@@ -21,18 +21,38 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+func TestStart_EthRpcEndpointNotSet(t *testing.T) {
+	errorMsg := "flag bridge-daemon-eth-rpc-endpoint is not set"
+
+	require.EqualError(
+		t,
+		client.Start(
+			grpc.Ctx,
+			daemonflags.GetDefaultDaemonFlags(),
+			appflags.GetFlagValuesFromOptions(appoptions.GetDefaultTestAppOptions("", nil)),
+			log.NewNopLogger(),
+			&mocks.GrpcClient{},
+		),
+		errorMsg,
+	)
+}
+
 func TestStart_TcpConnectionFails(t *testing.T) {
 	errorMsg := "Failed to create connection"
 
 	// Mock the gRPC client to return an error when creating a TCP connection.
 	mockGrpcClient := &mocks.GrpcClient{}
 	mockGrpcClient.On("NewTcpConnection", grpc.Ctx, d_constants.DefaultGrpcEndpoint).Return(nil, errors.New(errorMsg))
 
+	// Override default daemon flags with a non-empty EthRpcEndpoint.
+	daemonFlagsWithEthRpcEndpoint := daemonflags.GetDefaultDaemonFlags()
+	daemonFlagsWithEthRpcEndpoint.Bridge.EthRpcEndpoint = "http://localhost:8545"
+
 	require.EqualError(
 		t,
 		client.Start(
 			grpc.Ctx,
-			flags.GetDefaultDaemonFlags(),
+			daemonFlagsWithEthRpcEndpoint,
 			appflags.GetFlagValuesFromOptions(appoptions.GetDefaultTestAppOptions("", nil)),
 			log.NewNopLogger(),
 			mockGrpcClient,
@@ -56,11 +76,15 @@ func TestStart_UnixSocketConnectionFails(t *testing.T) {
 	mockGrpcClient.On("NewGrpcConnection", grpc.Ctx, grpc.SocketPath).Return(nil, errors.New(errorMsg))
 	mockGrpcClient.On("CloseConnection", grpc.GrpcConn).Return(nil)
 
+	// Override default daemon flags with a non-empty EthRpcEndpoint.
+	daemonFlagsWithEthRpcEndpoint := daemonflags.GetDefaultDaemonFlags()
+	daemonFlagsWithEthRpcEndpoint.Bridge.EthRpcEndpoint = "http://localhost:8545"
+
 	require.EqualError(
 		t,
 		client.Start(
 			grpc.Ctx,
-			flags.GetDefaultDaemonFlags(),
+			daemonFlagsWithEthRpcEndpoint,
 			appflags.GetFlagValuesFromOptions(appoptions.GetDefaultTestAppOptions("", nil)),
 			log.NewNopLogger(),
 			mockGrpcClient,