Testutil: Can generate attestation at epoch boundary (#8265)

prysmaticlabs · Jan 14, 2021 · 40155c9 · 40155c9
1 parent a2d4e33
commit 40155c9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/shared/testutil/attestation.go b/shared/testutil/attestation.go
@@ -41,7 +41,6 @@ func NewAttestation() *ethpb.Attestation {
 //
 // If you request 4 attestations, but there are 8 committees, you will get 4 fully aggregated attestations.
 func GenerateAttestations(bState *stateTrie.BeaconState, privs []bls.SecretKey, numToGen, slot uint64, randomRoot bool) ([]*ethpb.Attestation, error) {
-	currentEpoch := helpers.SlotToEpoch(slot)
 	var attestations []*ethpb.Attestation
 	generateHeadState := false
 	bState = bState.Copy()
@@ -50,6 +49,7 @@ func GenerateAttestations(bState *stateTrie.BeaconState, privs []bls.SecretKey,
 		slot--
 		generateHeadState = true
 	}
+	currentEpoch := helpers.SlotToEpoch(slot)
 
 	targetRoot := make([]byte, 32)
 	var headRoot []byte

diff --git a/shared/testutil/attestation_test.go b/shared/testutil/attestation_test.go
@@ -32,3 +32,9 @@ func TestHydrateIndexedAttestation(t *testing.T) {
 	_, err = a.Data.HashTreeRoot()
 	require.NoError(t, err)
 }
+
+func TestGenerateAttestations_EpochBoundary(t *testing.T) {
+	gs, pk := DeterministicGenesisState(t, 32)
+	_, err := GenerateAttestations(gs, pk, 1, params.BeaconConfig().SlotsPerEpoch, false)
+	require.NoError(t, err)
+}