prysm/testing/slasher/simulator/attestation_generator_test.go

package simulator

import (
	"testing"

	"github.com/OffchainLabs/prysm/v7/config/params"
	"github.com/OffchainLabs/prysm/v7/proto/prysm/v1alpha1/slashings"
	"github.com/OffchainLabs/prysm/v7/runtime/version"
	"github.com/OffchainLabs/prysm/v7/testing/require"
)

func TestGenerateAttestationsForSlot_Slashing(t *testing.T) {
	ctx := t.Context()
	simParams := &Parameters{
		SecondsPerSlot:         params.BeaconConfig().SecondsPerSlot,
		SlotsPerEpoch:          params.BeaconConfig().SlotsPerEpoch,
		AggregationPercent:     1,
		NumValidators:          64,
		AttesterSlashingProbab: 1,
	}
	srv := setupService(t, simParams)

	for _, v := range []int{version.Phase0, version.Electra} {
		t.Run(version.String(v), func(t *testing.T) {
			epoch3Atts, _, err := srv.generateAttestationsForSlot(ctx, v, params.BeaconConfig().SlotsPerEpoch*3)
			require.NoError(t, err)
			epoch4Atts, _, err := srv.generateAttestationsForSlot(ctx, v, params.BeaconConfig().SlotsPerEpoch*4)
			require.NoError(t, err)
			for i := 0; i < len(epoch3Atts); i += 2 {
				goodAtt := epoch3Atts[i]
				surroundAtt := epoch4Atts[i+1]
				require.Equal(t, true, slashings.IsSurround(surroundAtt, goodAtt))
			}
		})
	}
}

func TestGenerateAttestationsForSlot_CorrectIndices(t *testing.T) {
	ctx := t.Context()
	simParams := &Parameters{
		SecondsPerSlot:         params.BeaconConfig().SecondsPerSlot,
		SlotsPerEpoch:          params.BeaconConfig().SlotsPerEpoch,
		AggregationPercent:     1,
		NumValidators:          16384,
		AttesterSlashingProbab: 0,
	}
	srv := setupService(t, simParams)

	for _, v := range []int{version.Phase0, version.Electra} {
		t.Run(version.String(v), func(t *testing.T) {
			slot0Atts, _, err := srv.generateAttestationsForSlot(ctx, v, 0)
			require.NoError(t, err)
			slot1Atts, _, err := srv.generateAttestationsForSlot(ctx, v, 1)
			require.NoError(t, err)
			slot2Atts, _, err := srv.generateAttestationsForSlot(ctx, v, 2)
			require.NoError(t, err)
			var validatorIndices []uint64
			for _, att := range append(slot0Atts, slot1Atts...) {
				validatorIndices = append(validatorIndices, att.GetAttestingIndices()...)
			}
			for _, att := range slot2Atts {
				validatorIndices = append(validatorIndices, att.GetAttestingIndices()...)
			}

			// Making sure indices are one after the other for attestations.
			var validatorIndex uint64
			for _, ii := range validatorIndices {
				require.Equal(t, validatorIndex, ii)
				validatorIndex++
			}
		})
	}
}