increase test coverage

zeta-chain · Apr 4, 2024 · 6f4daa1 · 6f4daa1
1 parent 0fcdb89
commit 6f4daa1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/zetaclient/zetabridge/broadcast_test.go b/zetaclient/zetabridge/broadcast_test.go
@@ -82,7 +82,7 @@ func TestBroadcast(t *testing.T) {
 		msg := observerTypes.NewMsgAddBlockHeader(address.String(), chains.EthChain().ChainId, blockHash, 18495266, getHeaderData(t))
 		authzMsg, authzSigner, err := zetabridge.WrapMessageWithAuthz(msg)
 		require.NoError(t, err)
-		_, err = zetabridge.Broadcast(10000, authzMsg, authzSigner)
+		_, err = BroadcastToZetaCore(zetabridge, 10000, authzMsg, authzSigner)
 		require.NoError(t, err)
 	})
 
@@ -93,8 +93,19 @@ func TestBroadcast(t *testing.T) {
 		msg := observerTypes.NewMsgAddBlockHeader(address.String(), chains.EthChain().ChainId, blockHash, 18495266, getHeaderData(t))
 		authzMsg, authzSigner, err := zetabridge.WrapMessageWithAuthz(msg)
 		require.NoError(t, err)
-		_, err = zetabridge.Broadcast(10000, authzMsg, authzSigner)
+		_, err = BroadcastToZetaCore(zetabridge, 10000, authzMsg, authzSigner)
 		require.Error(t, err)
 	})
 
 }
+
+func TestZetaCoreBridge_GetContext(t *testing.T) {
+	address := types.AccAddress(stub.TestKeyringPair.PubKey().Address().Bytes())
+	zetabridge, err := setupCoreBridge()
+	require.NoError(t, err)
+	zetabridge.keys = keys.NewKeysWithKeybase(stub.NewKeyring(), address, "", "")
+
+	_, err = zetabridge.GetContext()
+	require.NoError(t, err)
+
+}
diff --git a/zetaclient/zetabridge/tx_test.go b/zetaclient/zetabridge/tx_test.go
@@ -353,7 +353,7 @@ func TestZetaCoreBridge_PostAddBlockHeader(t *testing.T) {
 func TestZetaCoreBridge_PostVoteInbound(t *testing.T) {
 	address := sdktypes.AccAddress(stub.TestKeyringPair.PubKey().Address().Bytes())
 
-	expectedOutput := observertypes.QueryHasVotedResponse{HasVoted: true}
+	expectedOutput := observertypes.QueryHasVotedResponse{HasVoted: false}
 	input := observertypes.QueryHasVotedRequest{
 		BallotIdentifier: "0x2d10e9b7ce7921fa6b61ada3020d1c797d5ec52424cdcf86ef31cbbbcd45db58",
 		VoterAddress:     address.String(),
@@ -374,7 +374,7 @@ func TestZetaCoreBridge_PostVoteInbound(t *testing.T) {
 			Creator: address.String(),
 		})
 		require.NoError(t, err)
-		require.Equal(t, "", hash)
+		require.Equal(t, sampleHash, hash)
 	})
 }
 
@@ -420,7 +420,7 @@ func TestZetaCoreBridge_MonitorVoteInboundTxResult(t *testing.T) {
 func TestZetaCoreBridge_PostVoteOutbound(t *testing.T) {
 	address := sdktypes.AccAddress(stub.TestKeyringPair.PubKey().Address().Bytes())
 
-	expectedOutput := observertypes.QueryHasVotedResponse{HasVoted: true}
+	expectedOutput := observertypes.QueryHasVotedResponse{HasVoted: false}
 	input := observertypes.QueryHasVotedRequest{
 		BallotIdentifier: "0xc507c67847209b403def6d944486ff888c442eccf924cf9ebdc48714b22b5347",
 		VoterAddress:     address.String(),
@@ -449,7 +449,7 @@ func TestZetaCoreBridge_PostVoteOutbound(t *testing.T) {
 		10001,
 		coin.CoinType_Gas)
 	require.NoError(t, err)
-	require.Equal(t, "", hash)
+	require.Equal(t, sampleHash, hash)
 	require.Equal(t, "0xc507c67847209b403def6d944486ff888c442eccf924cf9ebdc48714b22b5347", ballot)
 }