prysm-pulse/beacon-chain/blockchain/store/setter_getter_test.go

package store

import (
	"testing"

	ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
	"github.com/prysmaticlabs/prysm/testing/require"
)

func Test_store_PrevJustifiedCheckpt(t *testing.T) {
	s := &Store{}
	var cp *ethpb.Checkpoint
	require.Equal(t, cp, s.PrevJustifiedCheckpt())
	cp = &ethpb.Checkpoint{Epoch: 1, Root: []byte{'a'}}
	s.SetPrevJustifiedCheckpt(cp)
	require.Equal(t, cp, s.PrevJustifiedCheckpt())
}

func Test_store_BestJustifiedCheckpt(t *testing.T) {
	s := &Store{}
	var cp *ethpb.Checkpoint
	require.Equal(t, cp, s.BestJustifiedCheckpt())
	cp = &ethpb.Checkpoint{Epoch: 1, Root: []byte{'a'}}
	s.SetBestJustifiedCheckpt(cp)
	require.Equal(t, cp, s.BestJustifiedCheckpt())
}

func Test_store_JustifiedCheckpt(t *testing.T) {
	s := &Store{}
	var cp *ethpb.Checkpoint
	require.Equal(t, cp, s.JustifiedCheckpt())
	cp = &ethpb.Checkpoint{Epoch: 1, Root: []byte{'a'}}
	h := [32]byte{'b'}
	s.SetJustifiedCheckptAndPayloadHash(cp, h)
	require.Equal(t, cp, s.JustifiedCheckpt())
	require.Equal(t, h, s.JustifiedPayloadBlockHash())
}

func Test_store_FinalizedCheckpt(t *testing.T) {
	s := &Store{}
	var cp *ethpb.Checkpoint
	require.Equal(t, cp, s.FinalizedCheckpt())
	cp = &ethpb.Checkpoint{Epoch: 1, Root: []byte{'a'}}
	h := [32]byte{'b'}
	s.SetFinalizedCheckptAndPayloadHash(cp, h)
	require.Equal(t, cp, s.FinalizedCheckpt())
	require.Equal(t, h, s.FinalizedPayloadBlockHash())
}

func Test_store_PrevFinalizedCheckpt(t *testing.T) {
	s := &Store{}
	var cp *ethpb.Checkpoint
	require.Equal(t, cp, s.PrevFinalizedCheckpt())
	cp = &ethpb.Checkpoint{Epoch: 1, Root: []byte{'a'}}
	s.SetPrevFinalizedCheckpt(cp)
	require.Equal(t, cp, s.PrevFinalizedCheckpt())
}