mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
47 lines
1.4 KiB
Go
47 lines
1.4 KiB
Go
|
package v1
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
||
|
"github.com/prysmaticlabs/prysm/testing/require"
|
||
|
)
|
||
|
|
||
|
func TestBeaconState_PreviousEpochAttestations(t *testing.T) {
|
||
|
s, err := InitializeFromProto(ðpb.BeaconState{})
|
||
|
require.NoError(t, err)
|
||
|
atts, err := s.PreviousEpochAttestations()
|
||
|
require.NoError(t, err)
|
||
|
require.DeepEqual(t, []*ethpb.PendingAttestation(nil), atts)
|
||
|
|
||
|
want := []*ethpb.PendingAttestation{{ProposerIndex: 100}}
|
||
|
s, err = InitializeFromProto(ðpb.BeaconState{PreviousEpochAttestations: want})
|
||
|
require.NoError(t, err)
|
||
|
got, err := s.PreviousEpochAttestations()
|
||
|
require.NoError(t, err)
|
||
|
require.DeepEqual(t, want, got)
|
||
|
|
||
|
// Test copy does not mutate.
|
||
|
got[0].ProposerIndex = 101
|
||
|
require.DeepNotEqual(t, want, got)
|
||
|
}
|
||
|
|
||
|
func TestBeaconState_CurrentEpochAttestations(t *testing.T) {
|
||
|
s, err := InitializeFromProto(ðpb.BeaconState{})
|
||
|
require.NoError(t, err)
|
||
|
atts, err := s.CurrentEpochAttestations()
|
||
|
require.NoError(t, err)
|
||
|
require.DeepEqual(t, []*ethpb.PendingAttestation(nil), atts)
|
||
|
|
||
|
want := []*ethpb.PendingAttestation{{ProposerIndex: 101}}
|
||
|
s, err = InitializeFromProto(ðpb.BeaconState{CurrentEpochAttestations: want})
|
||
|
require.NoError(t, err)
|
||
|
got, err := s.CurrentEpochAttestations()
|
||
|
require.NoError(t, err)
|
||
|
require.DeepEqual(t, want, got)
|
||
|
|
||
|
// Test copy does not mutate.
|
||
|
got[0].ProposerIndex = 102
|
||
|
require.DeepNotEqual(t, want, got)
|
||
|
}
|