mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 23:48:06 -05:00
ee0a453b7b
* Move domain function and all signing root functions from beacon-chain/core/helpers to beacon-chain/core * @terencechain suggestion to put these methods under core/signing
38 lines
1.3 KiB
Go
38 lines
1.3 KiB
Go
package signing
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/eth2-types"
|
|
"github.com/prysmaticlabs/prysm/encoding/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
)
|
|
|
|
func TestDomain_OK(t *testing.T) {
|
|
state := ð.BeaconState{
|
|
Fork: ð.Fork{
|
|
Epoch: 3,
|
|
PreviousVersion: []byte{0, 0, 0, 2},
|
|
CurrentVersion: []byte{0, 0, 0, 3},
|
|
},
|
|
}
|
|
tests := []struct {
|
|
epoch types.Epoch
|
|
domainType [4]byte
|
|
result []byte
|
|
}{
|
|
{epoch: 1, domainType: bytesutil.ToBytes4(bytesutil.Bytes4(4)), result: bytesutil.ToBytes(947067381421703172, 32)},
|
|
{epoch: 2, domainType: bytesutil.ToBytes4(bytesutil.Bytes4(4)), result: bytesutil.ToBytes(947067381421703172, 32)},
|
|
{epoch: 2, domainType: bytesutil.ToBytes4(bytesutil.Bytes4(5)), result: bytesutil.ToBytes(947067381421703173, 32)},
|
|
{epoch: 3, domainType: bytesutil.ToBytes4(bytesutil.Bytes4(4)), result: bytesutil.ToBytes(9369798235163459588, 32)},
|
|
{epoch: 3, domainType: bytesutil.ToBytes4(bytesutil.Bytes4(5)), result: bytesutil.ToBytes(9369798235163459589, 32)},
|
|
}
|
|
for _, tt := range tests {
|
|
domain, err := Domain(state.Fork, tt.epoch, tt.domainType, nil)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, tt.result[:8], domain[:8], "Unexpected domain version")
|
|
}
|
|
}
|