mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-04 00:44:27 +00:00
9bd6147027
* bls spec tests * add more bls tests * use ioutil instead of bazel runfiles * dont read bytes * skip tests that overflow uint64 * manually fix input data * add tests * lint and gaz * add all new changes * some refactoring, cleanup, remove new API methods that only exist for tests * gaz * Remove yamls, skip test
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package spectest
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/ghodss/yaml"
|
|
"github.com/phoreproject/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
)
|
|
|
|
// Note: This actually tests the underlying library as we don't have a need for
|
|
// HashG2Compressed in our local BLS API.
|
|
func TestG2CompressedHash(t *testing.T) {
|
|
file, err := loadBlsYaml("msg_hash_g2_compressed/g2_compressed.yaml")
|
|
if err != nil {
|
|
t.Fatalf("Failed to read file: %v", err)
|
|
}
|
|
|
|
test := &G2CompressedTest{}
|
|
if err := yaml.Unmarshal(file, test); err != nil {
|
|
t.Fatalf("Failed to unmarshal: %v", err)
|
|
}
|
|
|
|
for i, tt := range test.TestCases {
|
|
t.Run(fmt.Sprintf("Test %d", i), func(t *testing.T) {
|
|
|
|
projective := bls.HashG2WithDomain(
|
|
bytesutil.ToBytes32(tt.Input.Message),
|
|
tt.Input.Domain,
|
|
)
|
|
hash := bls.CompressG2(projective.ToAffine())
|
|
|
|
var buf []byte
|
|
for _, slice := range tt.Output {
|
|
buf = append(buf, slice...)
|
|
}
|
|
if !bytes.Equal(buf, hash[:]) {
|
|
t.Logf("Domain=%d", tt.Input.Domain)
|
|
t.Fatalf("Hash does not match the expected output. "+
|
|
"Expected %#x but received %#x", buf, hash)
|
|
}
|
|
t.Logf("Success. Domain=%d", tt.Input.Domain)
|
|
})
|
|
}
|
|
}
|