add one unit test

zeta-chain · Jan 21, 2024 · 233ced4 · 233ced4
1 parent e032572
commit 233ced4
Showing 1 changed file with 50 additions and 4 deletions.
diff --git a/app/ante/ante_test.go b/app/ante/ante_test.go
@@ -53,18 +53,32 @@ func TestIsSystemTx(t *testing.T) {
 	isAuthorized := func(_ string) bool {
 		return true
 	}
+	isAuthorizedFalse := func(_ string) bool {
+		return false
+	}
 
 	tests := []struct {
-		name   string
-		tx     sdk.Tx
-		wantIs bool
+		name         string
+		tx           sdk.Tx
+		isAuthorized func(string) bool
+		wantIs       bool
 	}{
 		{
 			"MsgCreateTSSVoter",
 			buildTxFromMsg(&crosschaintypes.MsgCreateTSSVoter{
 				Creator:   sample.AccAddress(),
 				TssPubkey: "pubkey1234",
 			}),
+			isAuthorizedFalse,
+			false,
+		},
+		{
+			"MsgCreateTSSVoter",
+			buildTxFromMsg(&crosschaintypes.MsgCreateTSSVoter{
+				Creator:   sample.AccAddress(),
+				TssPubkey: "pubkey1234",
+			}),
+			isAuthorized,
 			true,
 		},
 		{
@@ -73,21 +87,29 @@ func TestIsSystemTx(t *testing.T) {
 				Creator:   sample.AccAddress(),
 				TssPubkey: "pubkey1234",
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgSend",
 			buildTxFromMsg(&banktypes.MsgSend{}),
+			isAuthorized,
+
 			false,
 		},
 		{
 			"MsgExec{MsgSend}",
 			buildAuthzTxFromMsg(&banktypes.MsgSend{}),
+			isAuthorized,
+
 			false,
 		},
 		{
 			"MsgCreateValidator",
 			buildTxFromMsg(&stakingtypes.MsgCreateValidator{}),
+			isAuthorized,
+
 			false,
 		},
 
@@ -96,13 +118,17 @@ func TestIsSystemTx(t *testing.T) {
 			buildTxFromMsg(&crosschaintypes.MsgVoteOnObservedInboundTx{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgExec{MsgVoteOnObservedInboundTx}",
 			buildAuthzTxFromMsg(&crosschaintypes.MsgVoteOnObservedInboundTx{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 
@@ -111,75 +137,95 @@ func TestIsSystemTx(t *testing.T) {
 			buildTxFromMsg(&crosschaintypes.MsgVoteOnObservedOutboundTx{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgExec{MsgVoteOnObservedOutboundTx}",
 			buildAuthzTxFromMsg(&crosschaintypes.MsgVoteOnObservedOutboundTx{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgAddToOutTxTracker",
 			buildTxFromMsg(&crosschaintypes.MsgAddToOutTxTracker{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgExec{MsgAddToOutTxTracker}",
 			buildAuthzTxFromMsg(&crosschaintypes.MsgAddToOutTxTracker{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgCreateTSSVoter",
 			buildTxFromMsg(&crosschaintypes.MsgCreateTSSVoter{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgExec{MsgCreateTSSVoter}",
 			buildAuthzTxFromMsg(&crosschaintypes.MsgCreateTSSVoter{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgAddBlockHeader",
 			buildTxFromMsg(&observertypes.MsgAddBlockHeader{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgExec{MsgAddBlockHeader}",
 			buildAuthzTxFromMsg(&observertypes.MsgAddBlockHeader{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgAddBlameVote",
 			buildTxFromMsg(&observertypes.MsgAddBlameVote{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 		{
 			"MsgExec{MsgAddBlameVote}",
 			buildAuthzTxFromMsg(&observertypes.MsgAddBlameVote{
 				Creator: sample.AccAddress(),
 			}),
+			isAuthorized,
+
 			true,
 		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			is := ante.IsSystemTx(tt.tx, isAuthorized)
+			is := ante.IsSystemTx(tt.tx, tt.isAuthorized)
 			require.Equal(t, tt.wantIs, is)
 		})
 	}