add reject votes from jailed finality provider

babylonlabs-io · Sep 11, 2024 · fe7bac6 · fe7bac6
1 parent 7ec426d
commit fe7bac6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/x/finality/keeper/msg_server_test.go b/x/finality/keeper/msg_server_test.go
@@ -228,6 +228,13 @@ func FuzzAddFinalitySig(f *testing.F) {
 		bsKeeper.EXPECT().GetFinalityProvider(gomock.Any(), gomock.Eq(fpBTCPKBytes)).Return(fp, nil).Times(1)
 		_, err = ms.AddFinalitySig(ctx, msg)
 		require.ErrorIs(t, err, bstypes.ErrFpAlreadySlashed)
+
+		// Case 7: jailed finality provider cannot vote
+		fp.Jailed = true
+		fp.SlashedBabylonHeight = 0
+		bsKeeper.EXPECT().GetFinalityProvider(gomock.Any(), gomock.Eq(fpBTCPKBytes)).Return(fp, nil).Times(1)
+		_, err = ms.AddFinalitySig(ctx, msg)
+		require.ErrorIs(t, err, bstypes.ErrFpAlreadyJailed)
 	})
 }