mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-09 03:01:19 +00:00
b4e72f1e19
* Update and use max per epoch churn limit * Update spec tests * Fix e2e test * deneb fork epoch condition * Fix lint and better casting * fix ordering * fix check * gaz * Fix more tests * Apply proposer boost to first block in equivocation * Increase timeout * Don't increase timeout, it's not the reason * implement deneb forkchoice spectests expose ReceiveBlob from the blockchain package * spin_off_helper * remove minimal tests * Terence's review * Add process register test for Deneb * Terence's suggestion Co-authored-by: terencechain <terence@prysmaticlabs.com> * fix forkchoice minimal * fix minimal sha * general sha * different repos * different repos --------- Co-authored-by: terence tsao <terence@prysmaticlabs.com> Co-authored-by: nisdas <nishdas93@gmail.com>
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package bls
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"errors"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/ghodss/yaml"
|
|
"github.com/prysmaticlabs/prysm/v4/crypto/bls"
|
|
"github.com/prysmaticlabs/prysm/v4/crypto/bls/common"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/spectest/utils"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/util"
|
|
)
|
|
|
|
func TestSign(t *testing.T) {
|
|
t.Run("blst", testSign)
|
|
}
|
|
|
|
func testSign(t *testing.T) {
|
|
testFolders, testFolderPath := utils.TestFolders(t, "general", "phase0", "bls/sign/bls")
|
|
if len(testFolders) == 0 {
|
|
t.Fatalf("No test folders found for %s/%s/%s", "general", "phase0", "bls/sign/bls")
|
|
}
|
|
for i, folder := range testFolders {
|
|
t.Run(folder.Name(), func(t *testing.T) {
|
|
file, err := util.BazelFileBytes(path.Join(testFolderPath, folder.Name(), "data.yaml"))
|
|
require.NoError(t, err)
|
|
test := &SignMsgTest{}
|
|
require.NoError(t, yaml.Unmarshal(file, test))
|
|
pkBytes, err := hex.DecodeString(test.Input.Privkey[2:])
|
|
require.NoError(t, err)
|
|
sk, err := bls.SecretKeyFromBytes(pkBytes)
|
|
if err != nil {
|
|
if test.Output == "" &&
|
|
(errors.Is(err, common.ErrZeroKey) || errors.Is(err, common.ErrSecretUnmarshal)) {
|
|
return
|
|
}
|
|
t.Fatalf("cannot unmarshal secret key: %v", err)
|
|
}
|
|
msgBytes, err := hex.DecodeString(test.Input.Message[2:])
|
|
require.NoError(t, err)
|
|
sig := sk.Sign(msgBytes)
|
|
|
|
if !sig.Verify(sk.PublicKey(), msgBytes) {
|
|
t.Fatal("could not verify signature")
|
|
}
|
|
|
|
outputBytes, err := hex.DecodeString(test.Output[2:])
|
|
require.NoError(t, err)
|
|
|
|
if !bytes.Equal(outputBytes, sig.Marshal()) {
|
|
t.Fatalf("Test Case %d: Signature does not match the expected output. "+
|
|
"Expected %#x but received %#x", i, outputBytes, sig.Marshal())
|
|
}
|
|
t.Log("Success")
|
|
})
|
|
}
|
|
}
|