mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-28 14:17:17 +00:00
7076a1ec4a
* Delete disable state lock and init sync verbose flags * Delete disable slashing broadcast * Remove disable wait for sync, noise, eth1 cache, static subnet * Remove enable broadcast recovery attemp and make it as default * Remove disable head update on per attestation * Revert disable att braodcast discovery attempt * gazelle * Fixed an anti pattern * Add enableAttBroadcastDiscoveryAttempts back * Add back WaitForSync * Remove extra lines * Use DisableDynamicCommitteeSubnets path per @prestonvanloon feedback Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
74 lines
2.3 KiB
Go
74 lines
2.3 KiB
Go
package beacon
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
mock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/operations/slashings"
|
|
mockp2p "github.com/prysmaticlabs/prysm/beacon-chain/p2p/testing"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestServer_SubmitProposerSlashing(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
st, privs := testutil.DeterministicGenesisState(t, 64)
|
|
slashedVal, err := st.ValidatorAtIndex(5)
|
|
require.NoError(t, err)
|
|
// We mark the validator at index 5 as already slashed.
|
|
slashedVal.Slashed = true
|
|
require.NoError(t, st.UpdateValidatorAtIndex(5, slashedVal))
|
|
|
|
mb := &mockp2p.MockBroadcaster{}
|
|
bs := &Server{
|
|
HeadFetcher: &mock.ChainService{
|
|
State: st,
|
|
},
|
|
SlashingsPool: slashings.NewPool(),
|
|
Broadcaster: mb,
|
|
}
|
|
|
|
// We want a proposer slashing for validator with index 2 to
|
|
// be included in the pool.
|
|
slashing, err := testutil.GenerateProposerSlashingForValidator(st, privs[2], uint64(2))
|
|
require.NoError(t, err)
|
|
|
|
_, err = bs.SubmitProposerSlashing(ctx, slashing)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, true, mb.BroadcastCalled, "Expected broadcast to be called")
|
|
}
|
|
|
|
func TestServer_SubmitAttesterSlashing(t *testing.T) {
|
|
ctx := context.Background()
|
|
// We mark the validators at index 5, 6 as already slashed.
|
|
st, privs := testutil.DeterministicGenesisState(t, 64)
|
|
slashedVal, err := st.ValidatorAtIndex(5)
|
|
require.NoError(t, err)
|
|
|
|
// We mark the validator at index 5 as already slashed.
|
|
slashedVal.Slashed = true
|
|
require.NoError(t, st.UpdateValidatorAtIndex(5, slashedVal))
|
|
|
|
mb := &mockp2p.MockBroadcaster{}
|
|
bs := &Server{
|
|
HeadFetcher: &mock.ChainService{
|
|
State: st,
|
|
},
|
|
SlashingsPool: slashings.NewPool(),
|
|
Broadcaster: mb,
|
|
}
|
|
|
|
slashing, err := testutil.GenerateAttesterSlashingForValidator(st, privs[2], uint64(2))
|
|
require.NoError(t, err)
|
|
|
|
// We want the intersection of the slashing attesting indices
|
|
// to be slashed, so we expect validators 2 and 3 to be in the response
|
|
// slashed indices.
|
|
_, err = bs.SubmitAttesterSlashing(ctx, slashing)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, true, mb.BroadcastCalled, "Expected broadcast to be called when flag is set")
|
|
}
|