Update single signer test

strangelove-ventures · Jan 9, 2024 · 71b0045 · 71b0045
1 parent 1965316
commit 71b0045
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/signer/single_signer_validator_test.go b/signer/single_signer_validator_test.go
@@ -106,4 +106,21 @@ func TestSingleSignerValidator(t *testing.T) {
 	// signing higher block now should succeed
 	_, _, _, err = validator.Sign(ctx, testChainID, ProposalToBlock(testChainID, &proposal))
 	require.NoError(t, err)
+
+	precommit := cometproto.Vote{
+		Height:    2,
+		Round:     0,
+		Type:      cometproto.PrecommitType,
+		Timestamp: time.Now(),
+		Extension: []byte("test"),
+	}
+
+	block = VoteToBlock(testChainID, &precommit)
+	sig, voteExtSig, _, err := validator.Sign(ctx, testChainID, block)
+	require.NoError(t, err)
+
+	require.True(t, privateKey.PubKey().VerifySignature(block.SignBytes, sig), "signature verification failed")
+
+	require.True(t, privateKey.PubKey().VerifySignature(block.VoteExtensionSignBytes, voteExtSig), "vote extension signature verification failed")
+
 }
diff --git a/signer/threshold_validator_test.go b/signer/threshold_validator_test.go
@@ -280,7 +280,7 @@ func testThresholdValidator(t *testing.T, threshold, total uint8) {
 			}
 
 			if !pubKey.VerifySignature(block.SignBytes, sig) {
-				return fmt.Errorf("vote extension signature verification failed")
+				return fmt.Errorf("signature verification failed")
 			}
 
 			if !pubKey.VerifySignature(block.VoteExtensionSignBytes, voteExtSig) {
@@ -299,7 +299,7 @@ func testThresholdValidator(t *testing.T, threshold, total uint8) {
 			}
 
 			if !pubKey.VerifySignature(block.SignBytes, sig) {
-				return fmt.Errorf("vote extension signature verification failed")
+				return fmt.Errorf("signature verification failed")
 			}
 
 			if !pubKey.VerifySignature(block.VoteExtensionSignBytes, voteExtSig) {
@@ -318,7 +318,7 @@ func testThresholdValidator(t *testing.T, threshold, total uint8) {
 			}
 
 			if !pubKey.VerifySignature(block.SignBytes, sig) {
-				return fmt.Errorf("vote extension signature verification failed")
+				return fmt.Errorf("signature verification failed")
 			}
 
 			if !pubKey.VerifySignature(block.VoteExtensionSignBytes, voteExtSig) {