fix bridge test

zeta-chain · Apr 10, 2024 · 3060eb3 · 3060eb3
1 parent fab7dce
commit 3060eb3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/zetaclient/zetabridge/query_test.go b/zetaclient/zetabridge/query_test.go
@@ -768,7 +768,7 @@ func TestZetaCoreBridge_GetBlockHeaderChainState(t *testing.T) {
 	}}
 	input := lightclienttypes.QueryGetChainStateRequest{ChainId: chainID}
 	method := "/zetachain.zetacore.lightclient.Query/ChainState"
-	server := setupMockServer(t, observertypes.RegisterQueryServer, method, input, expectedOutput)
+	server := setupMockServer(t, lightclienttypes.RegisterQueryServer, method, input, expectedOutput)
 	server.Serve()
 	defer closeMockServer(t, server)
 
@@ -846,7 +846,7 @@ func TestZetaCoreBridge_Prove(t *testing.T) {
 		TxIndex:   int64(txIndex),
 	}
 	method := "/zetachain.zetacore.lightclient.Query/Prove"
-	server := setupMockServer(t, observertypes.RegisterQueryServer, method, input, expectedOutput)
+	server := setupMockServer(t, lightclienttypes.RegisterQueryServer, method, input, expectedOutput)
 	server.Serve()
 	defer closeMockServer(t, server)
 

diff --git a/zetaclient/zetabridge/tx_test.go b/zetaclient/zetabridge/tx_test.go
@@ -19,6 +19,7 @@ import (
 	"github.com/zeta-chain/zetacore/pkg/coin"
 	"github.com/zeta-chain/zetacore/pkg/proofs"
 	crosschaintypes "github.com/zeta-chain/zetacore/x/crosschain/types"
+	lightclienttypes "github.com/zeta-chain/zetacore/x/lightclient/types"
 	observertypes "github.com/zeta-chain/zetacore/x/observer/types"
 	"github.com/zeta-chain/zetacore/zetaclient/authz"
 	"github.com/zeta-chain/zetacore/zetaclient/config"
@@ -204,6 +205,7 @@ func TestZetaCoreBridge_UpdateZetaCoreContext(t *testing.T) {
 		grpcmock.RegisterService(crosschaintypes.RegisterQueryServer),
 		grpcmock.RegisterService(upgradetypes.RegisterQueryServer),
 		grpcmock.RegisterService(observertypes.RegisterQueryServer),
+		grpcmock.RegisterService(lightclienttypes.RegisterQueryServer),
 		grpcmock.WithPlanner(planner.FirstMatch()),
 		grpcmock.WithListener(listener),
 		func(s *grpcmock.Server) {
@@ -286,6 +288,15 @@ func TestZetaCoreBridge_UpdateZetaCoreContext(t *testing.T) {
 					GasPriceIncreaseFlags:        nil,
 					BlockHeaderVerificationFlags: nil,
 				}})
+
+			method = "/zetachain.zetacore.lightclient.Query/VerificationFlags"
+			s.ExpectUnary(method).
+				UnlimitedTimes().
+				WithPayload(lightclienttypes.QueryVerificationFlagsRequest{}).
+				Return(lightclienttypes.QueryVerificationFlagsResponse{VerificationFlags: lightclienttypes.VerificationFlags{
+					EthTypeChainEnabled: true,
+					BtcTypeChainEnabled: false,
+				}})
 		},
 	)(t)