Fix conflicts rebasing eip4844
This commit is contained in:
parent
994990063a
commit
d292a3a6a8
@ -27,7 +27,7 @@ use parking_lot::RwLock;
|
|||||||
use proto_array::ReOrgThreshold;
|
use proto_array::ReOrgThreshold;
|
||||||
use slasher::Slasher;
|
use slasher::Slasher;
|
||||||
use slog::{crit, error, info, Logger};
|
use slog::{crit, error, info, Logger};
|
||||||
use slot_clock::{SlotClock, TestingSlotClock};
|
use slot_clock::SlotClock;
|
||||||
use std::marker::PhantomData;
|
use std::marker::PhantomData;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@ -954,13 +954,14 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<TEth1Backend, TEthSpec, THotStore, TColdStore>
|
impl<TEth1Backend, TEthSpec, THotStore, TColdStore, TSlotClock>
|
||||||
BeaconChainBuilder<Witness<TestingSlotClock, TEth1Backend, TEthSpec, THotStore, TColdStore>>
|
BeaconChainBuilder<Witness<TSlotClock, TEth1Backend, TEthSpec, THotStore, TColdStore>>
|
||||||
where
|
where
|
||||||
THotStore: ItemStore<TEthSpec> + 'static,
|
THotStore: ItemStore<TEthSpec> + 'static,
|
||||||
TColdStore: ItemStore<TEthSpec> + 'static,
|
TColdStore: ItemStore<TEthSpec> + 'static,
|
||||||
TEth1Backend: Eth1ChainBackend<TEthSpec> + 'static,
|
TEth1Backend: Eth1ChainBackend<TEthSpec> + 'static,
|
||||||
TEthSpec: EthSpec + 'static,
|
TEthSpec: EthSpec + 'static,
|
||||||
|
TSlotClock: SlotClock + 'static,
|
||||||
{
|
{
|
||||||
/// Sets the `BeaconChain` slot clock to `TestingSlotClock`.
|
/// Sets the `BeaconChain` slot clock to `TestingSlotClock`.
|
||||||
///
|
///
|
||||||
@ -970,7 +971,7 @@ where
|
|||||||
.genesis_time
|
.genesis_time
|
||||||
.ok_or("testing_slot_clock requires an initialized state")?;
|
.ok_or("testing_slot_clock requires an initialized state")?;
|
||||||
|
|
||||||
let slot_clock = TestingSlotClock::new(
|
let slot_clock = TSlotClock::new(
|
||||||
Slot::new(0),
|
Slot::new(0),
|
||||||
Duration::from_secs(genesis_time),
|
Duration::from_secs(genesis_time),
|
||||||
slot_duration,
|
slot_duration,
|
||||||
|
@ -208,7 +208,7 @@ impl BlockShufflingIds {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::test_utils::EphemeralHarnessType;
|
use crate::test_utils::EphemeralTestingSlotClockHarnessType;
|
||||||
use types::*;
|
use types::*;
|
||||||
|
|
||||||
type BeaconChainHarness =
|
type BeaconChainHarness =
|
||||||
|
@ -365,13 +365,13 @@ impl<T: EthSpec> SnapshotCache<T> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use crate::test_utils::{BeaconChainHarness, EphemeralTestingSlotClockHarnessType};
|
||||||
use types::{
|
use types::{
|
||||||
test_utils::generate_deterministic_keypair, BeaconBlock, Epoch, MainnetEthSpec,
|
test_utils::generate_deterministic_keypair, BeaconBlock, Epoch, MainnetEthSpec,
|
||||||
SignedBeaconBlock, Slot,
|
SignedBeaconBlock, Slot,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn get_harness() -> BeaconChainHarness<EphemeralHarnessType<MainnetEthSpec>> {
|
fn get_harness() -> BeaconChainHarness<EphemeralTestingSlotClockHarnessType<MainnetEthSpec>> {
|
||||||
let harness = BeaconChainHarness::builder(MainnetEthSpec)
|
let harness = BeaconChainHarness::builder(MainnetEthSpec)
|
||||||
.default_spec()
|
.default_spec()
|
||||||
.deterministic_keypairs(1)
|
.deterministic_keypairs(1)
|
||||||
|
@ -35,7 +35,7 @@ use rand::SeedableRng;
|
|||||||
use rayon::prelude::*;
|
use rayon::prelude::*;
|
||||||
use sensitive_url::SensitiveUrl;
|
use sensitive_url::SensitiveUrl;
|
||||||
use slog::Logger;
|
use slog::Logger;
|
||||||
use slot_clock::{SlotClock, TestingSlotClock};
|
use slot_clock::{SlotClock, SystemTimeSlotClock, TestingSlotClock};
|
||||||
use state_processing::per_block_processing::compute_timestamp_at_slot;
|
use state_processing::per_block_processing::compute_timestamp_at_slot;
|
||||||
use state_processing::{
|
use state_processing::{
|
||||||
state_advance::{complete_state_advance, partial_state_advance},
|
state_advance::{complete_state_advance, partial_state_advance},
|
||||||
@ -65,16 +65,25 @@ const FORK_NAME_ENV_VAR: &str = "FORK_NAME";
|
|||||||
// a different value.
|
// a different value.
|
||||||
pub const DEFAULT_TARGET_AGGREGATORS: u64 = u64::max_value();
|
pub const DEFAULT_TARGET_AGGREGATORS: u64 = u64::max_value();
|
||||||
|
|
||||||
pub type BaseHarnessType<TEthSpec, THotStore, TColdStore> =
|
pub type BaseHarnessType<TEthSpec, THotStore, TColdStore, TSlotClock> =
|
||||||
Witness<TestingSlotClock, CachingEth1Backend<TEthSpec>, TEthSpec, THotStore, TColdStore>;
|
Witness<TSlotClock, CachingEth1Backend<TEthSpec>, TEthSpec, THotStore, TColdStore>;
|
||||||
|
|
||||||
pub type DiskHarnessType<E> = BaseHarnessType<E, LevelDB<E>, LevelDB<E>>;
|
pub type DiskHarnessType<E, TSlotClock> = BaseHarnessType<E, LevelDB<E>, LevelDB<E>, TSlotClock>;
|
||||||
pub type EphemeralHarnessType<E> = BaseHarnessType<E, MemoryStore<E>, MemoryStore<E>>;
|
pub type EphemeralHarnessType<E, TSlotClock> =
|
||||||
|
BaseHarnessType<E, MemoryStore<E>, MemoryStore<E>, TSlotClock>;
|
||||||
|
|
||||||
pub type BoxedMutator<E, Hot, Cold> = Box<
|
pub type EphemeralTestingSlotClockHarnessType<E> =
|
||||||
|
BaseHarnessType<E, MemoryStore<E>, MemoryStore<E>, TestingSlotClock>;
|
||||||
|
pub type EphemeralSystemTimeSlotClockHarnessType<E> =
|
||||||
|
BaseHarnessType<E, MemoryStore<E>, MemoryStore<E>, SystemTimeSlotClock>;
|
||||||
|
|
||||||
|
pub type TestingSlotClockHarnessType<E, THotStore, TColdStore> =
|
||||||
|
BaseHarnessType<E, THotStore, TColdStore, TestingSlotClock>;
|
||||||
|
|
||||||
|
pub type BoxedMutator<E, Hot, Cold, S> = Box<
|
||||||
dyn FnOnce(
|
dyn FnOnce(
|
||||||
BeaconChainBuilder<BaseHarnessType<E, Hot, Cold>>,
|
BeaconChainBuilder<BaseHarnessType<E, Hot, Cold, S>>,
|
||||||
) -> BeaconChainBuilder<BaseHarnessType<E, Hot, Cold>>,
|
) -> BeaconChainBuilder<BaseHarnessType<E, Hot, Cold, S>>,
|
||||||
>;
|
>;
|
||||||
|
|
||||||
pub type AddBlocksResult<E> = (
|
pub type AddBlocksResult<E> = (
|
||||||
@ -146,7 +155,7 @@ pub fn test_spec<E: EthSpec>() -> ChainSpec {
|
|||||||
spec
|
spec
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct Builder<T: BeaconChainTypes> {
|
pub struct Builder<T: BeaconChainTypes, S: SlotClock + 'static> {
|
||||||
eth_spec_instance: T::EthSpec,
|
eth_spec_instance: T::EthSpec,
|
||||||
spec: Option<ChainSpec>,
|
spec: Option<ChainSpec>,
|
||||||
validator_keypairs: Option<Vec<Keypair>>,
|
validator_keypairs: Option<Vec<Keypair>>,
|
||||||
@ -155,17 +164,19 @@ pub struct Builder<T: BeaconChainTypes> {
|
|||||||
store_config: Option<StoreConfig>,
|
store_config: Option<StoreConfig>,
|
||||||
#[allow(clippy::type_complexity)]
|
#[allow(clippy::type_complexity)]
|
||||||
store: Option<Arc<HotColdDB<T::EthSpec, T::HotStore, T::ColdStore>>>,
|
store: Option<Arc<HotColdDB<T::EthSpec, T::HotStore, T::ColdStore>>>,
|
||||||
initial_mutator: Option<BoxedMutator<T::EthSpec, T::HotStore, T::ColdStore>>,
|
#[allow(clippy::type_complexity)]
|
||||||
store_mutator: Option<BoxedMutator<T::EthSpec, T::HotStore, T::ColdStore>>,
|
initial_mutator: Option<BoxedMutator<T::EthSpec, T::HotStore, T::ColdStore, S>>,
|
||||||
|
#[allow(clippy::type_complexity)]
|
||||||
|
store_mutator: Option<BoxedMutator<T::EthSpec, T::HotStore, T::ColdStore, S>>,
|
||||||
execution_layer: Option<ExecutionLayer<T::EthSpec>>,
|
execution_layer: Option<ExecutionLayer<T::EthSpec>>,
|
||||||
mock_execution_layer: Option<MockExecutionLayer<T::EthSpec>>,
|
mock_execution_layer: Option<MockExecutionLayer<T::EthSpec>>,
|
||||||
mock_builder: Option<TestingBuilder<T::EthSpec>>,
|
mock_builder: Option<TestingBuilder<T::EthSpec>>,
|
||||||
testing_slot_clock: Option<TestingSlotClock>,
|
testing_slot_clock: Option<S>,
|
||||||
runtime: TestRuntime,
|
runtime: TestRuntime,
|
||||||
log: Logger,
|
log: Logger,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<E: EthSpec> Builder<EphemeralHarnessType<E>> {
|
impl<E: EthSpec, S: SlotClock> Builder<EphemeralHarnessType<E, S>, S> {
|
||||||
pub fn fresh_ephemeral_store(mut self) -> Self {
|
pub fn fresh_ephemeral_store(mut self) -> Self {
|
||||||
let spec = self.spec.as_ref().expect("cannot build without spec");
|
let spec = self.spec.as_ref().expect("cannot build without spec");
|
||||||
let validator_keypairs = self
|
let validator_keypairs = self
|
||||||
@ -234,7 +245,7 @@ impl<E: EthSpec> Builder<EphemeralHarnessType<E>> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<E: EthSpec> Builder<DiskHarnessType<E>> {
|
impl<E: EthSpec, S: SlotClock> Builder<DiskHarnessType<E, S>, S> {
|
||||||
/// Disk store, start from genesis.
|
/// Disk store, start from genesis.
|
||||||
pub fn fresh_disk_store(mut self, store: Arc<HotColdDB<E, LevelDB<E>, LevelDB<E>>>) -> Self {
|
pub fn fresh_disk_store(mut self, store: Arc<HotColdDB<E, LevelDB<E>, LevelDB<E>>>) -> Self {
|
||||||
let validator_keypairs = self
|
let validator_keypairs = self
|
||||||
@ -271,11 +282,12 @@ impl<E: EthSpec> Builder<DiskHarnessType<E>> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<E, Hot, Cold> Builder<BaseHarnessType<E, Hot, Cold>>
|
impl<E, Hot, Cold, S> Builder<BaseHarnessType<E, Hot, Cold, S>, S>
|
||||||
where
|
where
|
||||||
E: EthSpec,
|
E: EthSpec,
|
||||||
Hot: ItemStore<E>,
|
Hot: ItemStore<E>,
|
||||||
Cold: ItemStore<E>,
|
Cold: ItemStore<E>,
|
||||||
|
S: SlotClock + 'static,
|
||||||
{
|
{
|
||||||
pub fn new(eth_spec_instance: E) -> Self {
|
pub fn new(eth_spec_instance: E) -> Self {
|
||||||
let runtime = TestRuntime::default();
|
let runtime = TestRuntime::default();
|
||||||
@ -351,7 +363,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// This mutator will be run before the `store_mutator`.
|
/// This mutator will be run before the `store_mutator`.
|
||||||
pub fn initial_mutator(mut self, mutator: BoxedMutator<E, Hot, Cold>) -> Self {
|
pub fn initial_mutator(mut self, mutator: BoxedMutator<E, Hot, Cold, S>) -> Self {
|
||||||
assert!(
|
assert!(
|
||||||
self.initial_mutator.is_none(),
|
self.initial_mutator.is_none(),
|
||||||
"initial mutator already set"
|
"initial mutator already set"
|
||||||
@ -361,14 +373,14 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// This mutator will be run after the `initial_mutator`.
|
/// This mutator will be run after the `initial_mutator`.
|
||||||
pub fn store_mutator(mut self, mutator: BoxedMutator<E, Hot, Cold>) -> Self {
|
pub fn store_mutator(mut self, mutator: BoxedMutator<E, Hot, Cold, S>) -> Self {
|
||||||
assert!(self.store_mutator.is_none(), "store mutator already set");
|
assert!(self.store_mutator.is_none(), "store mutator already set");
|
||||||
self.store_mutator = Some(mutator);
|
self.store_mutator = Some(mutator);
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Purposefully replace the `store_mutator`.
|
/// Purposefully replace the `store_mutator`.
|
||||||
pub fn override_store_mutator(mut self, mutator: BoxedMutator<E, Hot, Cold>) -> Self {
|
pub fn override_store_mutator(mut self, mutator: BoxedMutator<E, Hot, Cold, S>) -> Self {
|
||||||
assert!(self.store_mutator.is_some(), "store mutator not set");
|
assert!(self.store_mutator.is_some(), "store mutator not set");
|
||||||
self.store_mutator = Some(mutator);
|
self.store_mutator = Some(mutator);
|
||||||
self
|
self
|
||||||
@ -506,12 +518,7 @@ where
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn testing_slot_clock(mut self, slot_clock: TestingSlotClock) -> Self {
|
pub fn build(self) -> BeaconChainHarness<BaseHarnessType<E, Hot, Cold, S>> {
|
||||||
self.testing_slot_clock = Some(slot_clock);
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn build(self) -> BeaconChainHarness<BaseHarnessType<E, Hot, Cold>> {
|
|
||||||
let (shutdown_tx, shutdown_receiver) = futures::channel::mpsc::channel(1);
|
let (shutdown_tx, shutdown_receiver) = futures::channel::mpsc::channel(1);
|
||||||
|
|
||||||
let log = self.log;
|
let log = self.log;
|
||||||
@ -615,13 +622,391 @@ pub type HarnessSyncContributions<E> = Vec<(
|
|||||||
Option<SignedContributionAndProof<E>>,
|
Option<SignedContributionAndProof<E>>,
|
||||||
)>;
|
)>;
|
||||||
|
|
||||||
impl<E, Hot, Cold> BeaconChainHarness<BaseHarnessType<E, Hot, Cold>>
|
impl<E, Hot, Cold> BeaconChainHarness<TestingSlotClockHarnessType<E, Hot, Cold>>
|
||||||
where
|
where
|
||||||
E: EthSpec,
|
E: EthSpec,
|
||||||
Hot: ItemStore<E>,
|
Hot: ItemStore<E>,
|
||||||
Cold: ItemStore<E>,
|
Cold: ItemStore<E>,
|
||||||
{
|
{
|
||||||
pub fn builder(eth_spec_instance: E) -> Builder<BaseHarnessType<E, Hot, Cold>> {
|
pub fn set_current_slot(&self, slot: Slot) {
|
||||||
|
let current_slot = self.chain.slot().unwrap();
|
||||||
|
let current_epoch = current_slot.epoch(E::slots_per_epoch());
|
||||||
|
let epoch = slot.epoch(E::slots_per_epoch());
|
||||||
|
assert!(
|
||||||
|
epoch >= current_epoch,
|
||||||
|
"Jumping backwards to an earlier epoch isn't well defined. \
|
||||||
|
Please generate test blocks epoch-by-epoch instead."
|
||||||
|
);
|
||||||
|
self.chain.slot_clock.set_slot(slot.into());
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn process_block(
|
||||||
|
&self,
|
||||||
|
slot: Slot,
|
||||||
|
block_root: Hash256,
|
||||||
|
block: SignedBeaconBlock<E>,
|
||||||
|
) -> Result<SignedBeaconBlockHash, BlockError<E>> {
|
||||||
|
self.set_current_slot(slot);
|
||||||
|
let block_hash: SignedBeaconBlockHash = self
|
||||||
|
.chain
|
||||||
|
.process_block(
|
||||||
|
block_root,
|
||||||
|
Arc::new(block),
|
||||||
|
CountUnrealized::True,
|
||||||
|
NotifyExecutionLayer::Yes,
|
||||||
|
)
|
||||||
|
.await?
|
||||||
|
.into();
|
||||||
|
self.chain.recompute_head_at_current_slot().await;
|
||||||
|
Ok(block_hash)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn add_block_at_slot(
|
||||||
|
&self,
|
||||||
|
slot: Slot,
|
||||||
|
state: BeaconState<E>,
|
||||||
|
) -> Result<(SignedBeaconBlockHash, SignedBeaconBlock<E>, BeaconState<E>), BlockError<E>> {
|
||||||
|
self.set_current_slot(slot);
|
||||||
|
let (block, new_state) = self.make_block(state, slot).await;
|
||||||
|
let block_hash = self
|
||||||
|
.process_block(slot, block.canonical_root(), block.clone())
|
||||||
|
.await?;
|
||||||
|
Ok((block_hash, block, new_state))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn add_attested_block_at_slot(
|
||||||
|
&self,
|
||||||
|
slot: Slot,
|
||||||
|
state: BeaconState<E>,
|
||||||
|
state_root: Hash256,
|
||||||
|
validators: &[usize],
|
||||||
|
) -> Result<(SignedBeaconBlockHash, BeaconState<E>), BlockError<E>> {
|
||||||
|
let (block_hash, block, state) = self.add_block_at_slot(slot, state).await?;
|
||||||
|
self.attest_block(&state, state_root, block_hash, &block, validators);
|
||||||
|
Ok((block_hash, state))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_attested_blocks_at_slots_given_lbh(
|
||||||
|
&self,
|
||||||
|
mut state: BeaconState<E>,
|
||||||
|
state_root: Hash256,
|
||||||
|
slots: &[Slot],
|
||||||
|
validators: &[usize],
|
||||||
|
mut latest_block_hash: Option<SignedBeaconBlockHash>,
|
||||||
|
) -> AddBlocksResult<E> {
|
||||||
|
assert!(
|
||||||
|
slots.windows(2).all(|w| w[0] <= w[1]),
|
||||||
|
"Slots have to be sorted"
|
||||||
|
); // slice.is_sorted() isn't stabilized at the moment of writing this
|
||||||
|
let mut block_hash_from_slot: HashMap<Slot, SignedBeaconBlockHash> = HashMap::new();
|
||||||
|
let mut state_hash_from_slot: HashMap<Slot, BeaconStateHash> = HashMap::new();
|
||||||
|
for slot in slots {
|
||||||
|
let (block_hash, new_state) = self
|
||||||
|
.add_attested_block_at_slot(*slot, state, state_root, validators)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
state = new_state;
|
||||||
|
block_hash_from_slot.insert(*slot, block_hash);
|
||||||
|
state_hash_from_slot.insert(*slot, state.tree_hash_root().into());
|
||||||
|
latest_block_hash = Some(block_hash);
|
||||||
|
}
|
||||||
|
(
|
||||||
|
block_hash_from_slot,
|
||||||
|
state_hash_from_slot,
|
||||||
|
latest_block_hash.unwrap(),
|
||||||
|
state,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn add_attested_blocks_at_slots(
|
||||||
|
&self,
|
||||||
|
state: BeaconState<E>,
|
||||||
|
state_root: Hash256,
|
||||||
|
slots: &[Slot],
|
||||||
|
validators: &[usize],
|
||||||
|
) -> AddBlocksResult<E> {
|
||||||
|
assert!(!slots.is_empty());
|
||||||
|
self.add_attested_blocks_at_slots_given_lbh(state, state_root, slots, validators, None)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A monstrosity of great usefulness.
|
||||||
|
///
|
||||||
|
/// Calls `add_attested_blocks_at_slots` for each of the chains in `chains`,
|
||||||
|
/// taking care to batch blocks by epoch so that the slot clock gets advanced one
|
||||||
|
/// epoch at a time.
|
||||||
|
///
|
||||||
|
/// Chains is a vec of `(state, slots, validators)` tuples.
|
||||||
|
pub async fn add_blocks_on_multiple_chains(
|
||||||
|
&self,
|
||||||
|
chains: Vec<(BeaconState<E>, Vec<Slot>, Vec<usize>)>,
|
||||||
|
) -> Vec<AddBlocksResult<E>> {
|
||||||
|
let slots_per_epoch = E::slots_per_epoch();
|
||||||
|
|
||||||
|
let min_epoch = chains
|
||||||
|
.iter()
|
||||||
|
.map(|(_, slots, _)| slots.iter().min().unwrap())
|
||||||
|
.min()
|
||||||
|
.unwrap()
|
||||||
|
.epoch(slots_per_epoch);
|
||||||
|
let max_epoch = chains
|
||||||
|
.iter()
|
||||||
|
.map(|(_, slots, _)| slots.iter().max().unwrap())
|
||||||
|
.max()
|
||||||
|
.unwrap()
|
||||||
|
.epoch(slots_per_epoch);
|
||||||
|
|
||||||
|
let mut chains = chains
|
||||||
|
.into_iter()
|
||||||
|
.map(|(state, slots, validators)| {
|
||||||
|
(
|
||||||
|
state,
|
||||||
|
slots,
|
||||||
|
validators,
|
||||||
|
HashMap::new(),
|
||||||
|
HashMap::new(),
|
||||||
|
SignedBeaconBlockHash::from(Hash256::zero()),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
for epoch in min_epoch.as_u64()..=max_epoch.as_u64() {
|
||||||
|
let mut new_chains = vec![];
|
||||||
|
|
||||||
|
for (
|
||||||
|
mut head_state,
|
||||||
|
slots,
|
||||||
|
validators,
|
||||||
|
mut block_hashes,
|
||||||
|
mut state_hashes,
|
||||||
|
head_block,
|
||||||
|
) in chains
|
||||||
|
{
|
||||||
|
let epoch_slots = slots
|
||||||
|
.iter()
|
||||||
|
.filter(|s| s.epoch(slots_per_epoch).as_u64() == epoch)
|
||||||
|
.copied()
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
let head_state_root = head_state.update_tree_hash_cache().unwrap();
|
||||||
|
let (new_block_hashes, new_state_hashes, new_head_block, new_head_state) = self
|
||||||
|
.add_attested_blocks_at_slots_given_lbh(
|
||||||
|
head_state,
|
||||||
|
head_state_root,
|
||||||
|
&epoch_slots,
|
||||||
|
&validators,
|
||||||
|
Some(head_block),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
block_hashes.extend(new_block_hashes);
|
||||||
|
state_hashes.extend(new_state_hashes);
|
||||||
|
|
||||||
|
new_chains.push((
|
||||||
|
new_head_state,
|
||||||
|
slots,
|
||||||
|
validators,
|
||||||
|
block_hashes,
|
||||||
|
state_hashes,
|
||||||
|
new_head_block,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
chains = new_chains;
|
||||||
|
}
|
||||||
|
|
||||||
|
chains
|
||||||
|
.into_iter()
|
||||||
|
.map(|(state, _, _, block_hashes, state_hashes, head_block)| {
|
||||||
|
(block_hashes, state_hashes, head_block, state)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_finalized_checkpoints(&self) -> HashSet<SignedBeaconBlockHash> {
|
||||||
|
let chain_dump = self.chain.chain_dump().unwrap();
|
||||||
|
chain_dump
|
||||||
|
.iter()
|
||||||
|
.cloned()
|
||||||
|
.map(|checkpoint| checkpoint.beacon_state.finalized_checkpoint().root.into())
|
||||||
|
.filter(|block_hash| *block_hash != Hash256::zero().into())
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Deprecated: Do not modify the slot clock manually; rely on add_attested_blocks_at_slots()
|
||||||
|
/// instead
|
||||||
|
///
|
||||||
|
/// Advance the slot of the `BeaconChain`.
|
||||||
|
///
|
||||||
|
/// Does not produce blocks or attestations.
|
||||||
|
pub fn advance_slot(&self) {
|
||||||
|
self.chain.slot_clock.advance_slot();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Advance the clock to `lookahead` before the start of `slot`.
|
||||||
|
pub fn advance_to_slot_lookahead(&self, slot: Slot, lookahead: Duration) {
|
||||||
|
let time = self.chain.slot_clock.start_of(slot).unwrap() - lookahead;
|
||||||
|
self.chain.slot_clock.set_current_time(time);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Deprecated: Use make_block() instead
|
||||||
|
///
|
||||||
|
/// Returns a newly created block, signed by the proposer for the given slot.
|
||||||
|
pub async fn build_block(
|
||||||
|
&self,
|
||||||
|
state: BeaconState<E>,
|
||||||
|
slot: Slot,
|
||||||
|
_block_strategy: BlockStrategy,
|
||||||
|
) -> (SignedBeaconBlock<E>, BeaconState<E>) {
|
||||||
|
self.make_block(state, slot).await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Uses `Self::extend_chain` to build the chain out to the `target_slot`.
|
||||||
|
pub async fn extend_to_slot(&self, target_slot: Slot) -> Hash256 {
|
||||||
|
if self.chain.slot().unwrap() == self.chain.canonical_head.cached_head().head_slot() {
|
||||||
|
self.advance_slot();
|
||||||
|
}
|
||||||
|
|
||||||
|
let num_slots = target_slot
|
||||||
|
.as_usize()
|
||||||
|
.checked_sub(self.chain.slot().unwrap().as_usize())
|
||||||
|
.expect("target_slot must be >= current_slot")
|
||||||
|
.checked_add(1)
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
self.extend_slots(num_slots).await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Uses `Self::extend_chain` to `num_slots` blocks.
|
||||||
|
///
|
||||||
|
/// Utilizes:
|
||||||
|
///
|
||||||
|
/// - BlockStrategy::OnCanonicalHead,
|
||||||
|
/// - AttestationStrategy::AllValidators,
|
||||||
|
pub async fn extend_slots(&self, num_slots: usize) -> Hash256 {
|
||||||
|
if self.chain.slot().unwrap() == self.chain.canonical_head.cached_head().head_slot() {
|
||||||
|
self.advance_slot();
|
||||||
|
}
|
||||||
|
|
||||||
|
self.extend_chain(
|
||||||
|
num_slots,
|
||||||
|
BlockStrategy::OnCanonicalHead,
|
||||||
|
AttestationStrategy::AllValidators,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Deprecated: Use add_attested_blocks_at_slots() instead
|
||||||
|
///
|
||||||
|
/// Extend the `BeaconChain` with some blocks and attestations. Returns the root of the
|
||||||
|
/// last-produced block (the head of the chain).
|
||||||
|
///
|
||||||
|
/// Chain will be extended by `num_blocks` blocks.
|
||||||
|
///
|
||||||
|
/// The `block_strategy` dictates where the new blocks will be placed.
|
||||||
|
///
|
||||||
|
/// The `attestation_strategy` dictates which validators will attest to the newly created
|
||||||
|
/// blocks.
|
||||||
|
pub async fn extend_chain(
|
||||||
|
&self,
|
||||||
|
num_blocks: usize,
|
||||||
|
block_strategy: BlockStrategy,
|
||||||
|
attestation_strategy: AttestationStrategy,
|
||||||
|
) -> Hash256 {
|
||||||
|
let (mut state, slots) = match block_strategy {
|
||||||
|
BlockStrategy::OnCanonicalHead => {
|
||||||
|
let current_slot: u64 = self.get_current_slot().into();
|
||||||
|
let slots: Vec<Slot> = (current_slot..(current_slot + (num_blocks as u64)))
|
||||||
|
.map(Slot::new)
|
||||||
|
.collect();
|
||||||
|
let state = self.get_current_state();
|
||||||
|
(state, slots)
|
||||||
|
}
|
||||||
|
BlockStrategy::ForkCanonicalChainAt {
|
||||||
|
previous_slot,
|
||||||
|
first_slot,
|
||||||
|
} => {
|
||||||
|
let first_slot_: u64 = first_slot.into();
|
||||||
|
let slots: Vec<Slot> = (first_slot_..(first_slot_ + (num_blocks as u64)))
|
||||||
|
.map(Slot::new)
|
||||||
|
.collect();
|
||||||
|
let state = self
|
||||||
|
.chain
|
||||||
|
.state_at_slot(previous_slot, StateSkipConfig::WithStateRoots)
|
||||||
|
.unwrap();
|
||||||
|
(state, slots)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let validators = match attestation_strategy {
|
||||||
|
AttestationStrategy::AllValidators => self.get_all_validators(),
|
||||||
|
AttestationStrategy::SomeValidators(vals) => vals,
|
||||||
|
};
|
||||||
|
let state_root = state.update_tree_hash_cache().unwrap();
|
||||||
|
let (_, _, last_produced_block_hash, _) = self
|
||||||
|
.add_attested_blocks_at_slots(state, state_root, &slots, &validators)
|
||||||
|
.await;
|
||||||
|
last_produced_block_hash.into()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Deprecated: Use add_attested_blocks_at_slots() instead
|
||||||
|
///
|
||||||
|
/// Creates two forks:
|
||||||
|
///
|
||||||
|
/// - The "honest" fork: created by the `honest_validators` who have built `honest_fork_blocks`
|
||||||
|
/// on the head
|
||||||
|
/// - The "faulty" fork: created by the `faulty_validators` who skipped a slot and
|
||||||
|
/// then built `faulty_fork_blocks`.
|
||||||
|
///
|
||||||
|
/// Returns `(honest_head, faulty_head)`, the roots of the blocks at the top of each chain.
|
||||||
|
pub async fn generate_two_forks_by_skipping_a_block(
|
||||||
|
&self,
|
||||||
|
honest_validators: &[usize],
|
||||||
|
faulty_validators: &[usize],
|
||||||
|
honest_fork_blocks: usize,
|
||||||
|
faulty_fork_blocks: usize,
|
||||||
|
) -> (Hash256, Hash256) {
|
||||||
|
let initial_head_slot = self.chain.head_snapshot().beacon_block.slot();
|
||||||
|
|
||||||
|
// Move to the next slot so we may produce some more blocks on the head.
|
||||||
|
self.advance_slot();
|
||||||
|
|
||||||
|
// Extend the chain with blocks where only honest validators agree.
|
||||||
|
let honest_head = self
|
||||||
|
.extend_chain(
|
||||||
|
honest_fork_blocks,
|
||||||
|
BlockStrategy::OnCanonicalHead,
|
||||||
|
AttestationStrategy::SomeValidators(honest_validators.to_vec()),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
// Go back to the last block where all agreed, and build blocks upon it where only faulty nodes
|
||||||
|
// agree.
|
||||||
|
let faulty_head = self
|
||||||
|
.extend_chain(
|
||||||
|
faulty_fork_blocks,
|
||||||
|
BlockStrategy::ForkCanonicalChainAt {
|
||||||
|
previous_slot: initial_head_slot,
|
||||||
|
// `initial_head_slot + 2` means one slot is skipped.
|
||||||
|
first_slot: initial_head_slot + 2,
|
||||||
|
},
|
||||||
|
AttestationStrategy::SomeValidators(faulty_validators.to_vec()),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert_ne!(honest_head, faulty_head, "forks should be distinct");
|
||||||
|
|
||||||
|
(honest_head, faulty_head)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<E, Hot, Cold, S> BeaconChainHarness<BaseHarnessType<E, Hot, Cold, S>>
|
||||||
|
where
|
||||||
|
E: EthSpec,
|
||||||
|
Hot: ItemStore<E>,
|
||||||
|
Cold: ItemStore<E>,
|
||||||
|
S: SlotClock + 'static,
|
||||||
|
{
|
||||||
|
pub fn builder(eth_spec_instance: E) -> Builder<BaseHarnessType<E, Hot, Cold, S>, S> {
|
||||||
Builder::new(eth_spec_instance)
|
Builder::new(eth_spec_instance)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1654,27 +2039,6 @@ where
|
|||||||
(deposits, state)
|
(deposits, state)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn process_block(
|
|
||||||
&self,
|
|
||||||
slot: Slot,
|
|
||||||
block_root: Hash256,
|
|
||||||
block: SignedBeaconBlock<E>,
|
|
||||||
) -> Result<SignedBeaconBlockHash, BlockError<E>> {
|
|
||||||
self.set_current_slot(slot);
|
|
||||||
let block_hash: SignedBeaconBlockHash = self
|
|
||||||
.chain
|
|
||||||
.process_block(
|
|
||||||
block_root,
|
|
||||||
Arc::new(block),
|
|
||||||
CountUnrealized::True,
|
|
||||||
NotifyExecutionLayer::Yes,
|
|
||||||
)
|
|
||||||
.await?
|
|
||||||
.into();
|
|
||||||
self.chain.recompute_head_at_current_slot().await;
|
|
||||||
Ok(block_hash)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn process_block_result(
|
pub async fn process_block_result(
|
||||||
&self,
|
&self,
|
||||||
block: SignedBeaconBlock<E>,
|
block: SignedBeaconBlock<E>,
|
||||||
@ -1732,31 +2096,6 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn set_current_slot(&self, slot: Slot) {
|
|
||||||
let current_slot = self.chain.slot().unwrap();
|
|
||||||
let current_epoch = current_slot.epoch(E::slots_per_epoch());
|
|
||||||
let epoch = slot.epoch(E::slots_per_epoch());
|
|
||||||
assert!(
|
|
||||||
epoch >= current_epoch,
|
|
||||||
"Jumping backwards to an earlier epoch isn't well defined. \
|
|
||||||
Please generate test blocks epoch-by-epoch instead."
|
|
||||||
);
|
|
||||||
self.chain.slot_clock.set_slot(slot.into());
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn add_block_at_slot(
|
|
||||||
&self,
|
|
||||||
slot: Slot,
|
|
||||||
state: BeaconState<E>,
|
|
||||||
) -> Result<(SignedBeaconBlockHash, SignedBeaconBlock<E>, BeaconState<E>), BlockError<E>> {
|
|
||||||
self.set_current_slot(slot);
|
|
||||||
let (block, new_state) = self.make_block(state, slot).await;
|
|
||||||
let block_hash = self
|
|
||||||
.process_block(slot, block.canonical_root(), block.clone())
|
|
||||||
.await?;
|
|
||||||
Ok((block_hash, block, new_state))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn attest_block(
|
pub fn attest_block(
|
||||||
&self,
|
&self,
|
||||||
state: &BeaconState<E>,
|
state: &BeaconState<E>,
|
||||||
@ -1769,354 +2108,6 @@ where
|
|||||||
self.make_attestations(validators, state, state_root, block_hash, block.slot());
|
self.make_attestations(validators, state, state_root, block_hash, block.slot());
|
||||||
self.process_attestations(attestations);
|
self.process_attestations(attestations);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn add_attested_block_at_slot(
|
|
||||||
&self,
|
|
||||||
slot: Slot,
|
|
||||||
state: BeaconState<E>,
|
|
||||||
state_root: Hash256,
|
|
||||||
validators: &[usize],
|
|
||||||
) -> Result<(SignedBeaconBlockHash, BeaconState<E>), BlockError<E>> {
|
|
||||||
let (block_hash, block, state) = self.add_block_at_slot(slot, state).await?;
|
|
||||||
self.attest_block(&state, state_root, block_hash, &block, validators);
|
|
||||||
Ok((block_hash, state))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn add_attested_blocks_at_slots(
|
|
||||||
&self,
|
|
||||||
state: BeaconState<E>,
|
|
||||||
state_root: Hash256,
|
|
||||||
slots: &[Slot],
|
|
||||||
validators: &[usize],
|
|
||||||
) -> AddBlocksResult<E> {
|
|
||||||
assert!(!slots.is_empty());
|
|
||||||
self.add_attested_blocks_at_slots_given_lbh(state, state_root, slots, validators, None)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn add_attested_blocks_at_slots_given_lbh(
|
|
||||||
&self,
|
|
||||||
mut state: BeaconState<E>,
|
|
||||||
state_root: Hash256,
|
|
||||||
slots: &[Slot],
|
|
||||||
validators: &[usize],
|
|
||||||
mut latest_block_hash: Option<SignedBeaconBlockHash>,
|
|
||||||
) -> AddBlocksResult<E> {
|
|
||||||
assert!(
|
|
||||||
slots.windows(2).all(|w| w[0] <= w[1]),
|
|
||||||
"Slots have to be sorted"
|
|
||||||
); // slice.is_sorted() isn't stabilized at the moment of writing this
|
|
||||||
let mut block_hash_from_slot: HashMap<Slot, SignedBeaconBlockHash> = HashMap::new();
|
|
||||||
let mut state_hash_from_slot: HashMap<Slot, BeaconStateHash> = HashMap::new();
|
|
||||||
for slot in slots {
|
|
||||||
let (block_hash, new_state) = self
|
|
||||||
.add_attested_block_at_slot(*slot, state, state_root, validators)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
state = new_state;
|
|
||||||
block_hash_from_slot.insert(*slot, block_hash);
|
|
||||||
state_hash_from_slot.insert(*slot, state.tree_hash_root().into());
|
|
||||||
latest_block_hash = Some(block_hash);
|
|
||||||
}
|
|
||||||
(
|
|
||||||
block_hash_from_slot,
|
|
||||||
state_hash_from_slot,
|
|
||||||
latest_block_hash.unwrap(),
|
|
||||||
state,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// A monstrosity of great usefulness.
|
|
||||||
///
|
|
||||||
/// Calls `add_attested_blocks_at_slots` for each of the chains in `chains`,
|
|
||||||
/// taking care to batch blocks by epoch so that the slot clock gets advanced one
|
|
||||||
/// epoch at a time.
|
|
||||||
///
|
|
||||||
/// Chains is a vec of `(state, slots, validators)` tuples.
|
|
||||||
pub async fn add_blocks_on_multiple_chains(
|
|
||||||
&self,
|
|
||||||
chains: Vec<(BeaconState<E>, Vec<Slot>, Vec<usize>)>,
|
|
||||||
) -> Vec<AddBlocksResult<E>> {
|
|
||||||
let slots_per_epoch = E::slots_per_epoch();
|
|
||||||
|
|
||||||
let min_epoch = chains
|
|
||||||
.iter()
|
|
||||||
.map(|(_, slots, _)| slots.iter().min().unwrap())
|
|
||||||
.min()
|
|
||||||
.unwrap()
|
|
||||||
.epoch(slots_per_epoch);
|
|
||||||
let max_epoch = chains
|
|
||||||
.iter()
|
|
||||||
.map(|(_, slots, _)| slots.iter().max().unwrap())
|
|
||||||
.max()
|
|
||||||
.unwrap()
|
|
||||||
.epoch(slots_per_epoch);
|
|
||||||
|
|
||||||
let mut chains = chains
|
|
||||||
.into_iter()
|
|
||||||
.map(|(state, slots, validators)| {
|
|
||||||
(
|
|
||||||
state,
|
|
||||||
slots,
|
|
||||||
validators,
|
|
||||||
HashMap::new(),
|
|
||||||
HashMap::new(),
|
|
||||||
SignedBeaconBlockHash::from(Hash256::zero()),
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
for epoch in min_epoch.as_u64()..=max_epoch.as_u64() {
|
|
||||||
let mut new_chains = vec![];
|
|
||||||
|
|
||||||
for (
|
|
||||||
mut head_state,
|
|
||||||
slots,
|
|
||||||
validators,
|
|
||||||
mut block_hashes,
|
|
||||||
mut state_hashes,
|
|
||||||
head_block,
|
|
||||||
) in chains
|
|
||||||
{
|
|
||||||
let epoch_slots = slots
|
|
||||||
.iter()
|
|
||||||
.filter(|s| s.epoch(slots_per_epoch).as_u64() == epoch)
|
|
||||||
.copied()
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
let head_state_root = head_state.update_tree_hash_cache().unwrap();
|
|
||||||
let (new_block_hashes, new_state_hashes, new_head_block, new_head_state) = self
|
|
||||||
.add_attested_blocks_at_slots_given_lbh(
|
|
||||||
head_state,
|
|
||||||
head_state_root,
|
|
||||||
&epoch_slots,
|
|
||||||
&validators,
|
|
||||||
Some(head_block),
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
block_hashes.extend(new_block_hashes);
|
|
||||||
state_hashes.extend(new_state_hashes);
|
|
||||||
|
|
||||||
new_chains.push((
|
|
||||||
new_head_state,
|
|
||||||
slots,
|
|
||||||
validators,
|
|
||||||
block_hashes,
|
|
||||||
state_hashes,
|
|
||||||
new_head_block,
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
chains = new_chains;
|
|
||||||
}
|
|
||||||
|
|
||||||
chains
|
|
||||||
.into_iter()
|
|
||||||
.map(|(state, _, _, block_hashes, state_hashes, head_block)| {
|
|
||||||
(block_hashes, state_hashes, head_block, state)
|
|
||||||
})
|
|
||||||
.collect()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_finalized_checkpoints(&self) -> HashSet<SignedBeaconBlockHash> {
|
|
||||||
let chain_dump = self.chain.chain_dump().unwrap();
|
|
||||||
chain_dump
|
|
||||||
.iter()
|
|
||||||
.cloned()
|
|
||||||
.map(|checkpoint| checkpoint.beacon_state.finalized_checkpoint().root.into())
|
|
||||||
.filter(|block_hash| *block_hash != Hash256::zero().into())
|
|
||||||
.collect()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Deprecated: Do not modify the slot clock manually; rely on add_attested_blocks_at_slots()
|
|
||||||
/// instead
|
|
||||||
///
|
|
||||||
/// Advance the slot of the `BeaconChain`.
|
|
||||||
///
|
|
||||||
/// Does not produce blocks or attestations.
|
|
||||||
pub fn advance_slot(&self) {
|
|
||||||
self.chain.slot_clock.advance_slot();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Advance the clock to `lookahead` before the start of `slot`.
|
|
||||||
pub fn advance_to_slot_lookahead(&self, slot: Slot, lookahead: Duration) {
|
|
||||||
let time = self.chain.slot_clock.start_of(slot).unwrap() - lookahead;
|
|
||||||
self.chain.slot_clock.set_current_time(time);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Deprecated: Use make_block() instead
|
|
||||||
///
|
|
||||||
/// Returns a newly created block, signed by the proposer for the given slot.
|
|
||||||
pub async fn build_block(
|
|
||||||
&self,
|
|
||||||
state: BeaconState<E>,
|
|
||||||
slot: Slot,
|
|
||||||
_block_strategy: BlockStrategy,
|
|
||||||
) -> (SignedBeaconBlock<E>, BeaconState<E>) {
|
|
||||||
self.make_block(state, slot).await
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Uses `Self::extend_chain` to build the chain out to the `target_slot`.
|
|
||||||
pub async fn extend_to_slot(&self, target_slot: Slot) -> Hash256 {
|
|
||||||
if self.chain.slot().unwrap() == self.chain.canonical_head.cached_head().head_slot() {
|
|
||||||
self.advance_slot();
|
|
||||||
}
|
|
||||||
|
|
||||||
let num_slots = target_slot
|
|
||||||
.as_usize()
|
|
||||||
.checked_sub(self.chain.slot().unwrap().as_usize())
|
|
||||||
.expect("target_slot must be >= current_slot")
|
|
||||||
.checked_add(1)
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
self.extend_slots(num_slots).await
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Uses `Self::extend_chain` to `num_slots` blocks.
|
|
||||||
///
|
|
||||||
/// Utilizes:
|
|
||||||
///
|
|
||||||
/// - BlockStrategy::OnCanonicalHead,
|
|
||||||
/// - AttestationStrategy::AllValidators,
|
|
||||||
pub async fn extend_slots(&self, num_slots: usize) -> Hash256 {
|
|
||||||
if self.chain.slot().unwrap() == self.chain.canonical_head.cached_head().head_slot() {
|
|
||||||
self.advance_slot();
|
|
||||||
}
|
|
||||||
|
|
||||||
self.extend_chain(
|
|
||||||
num_slots,
|
|
||||||
BlockStrategy::OnCanonicalHead,
|
|
||||||
AttestationStrategy::AllValidators,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Deprecated: Use add_attested_blocks_at_slots() instead
|
|
||||||
///
|
|
||||||
/// Extend the `BeaconChain` with some blocks and attestations. Returns the root of the
|
|
||||||
/// last-produced block (the head of the chain).
|
|
||||||
///
|
|
||||||
/// Chain will be extended by `num_blocks` blocks.
|
|
||||||
///
|
|
||||||
/// The `block_strategy` dictates where the new blocks will be placed.
|
|
||||||
///
|
|
||||||
/// The `attestation_strategy` dictates which validators will attest to the newly created
|
|
||||||
/// blocks.
|
|
||||||
pub async fn extend_chain(
|
|
||||||
&self,
|
|
||||||
num_blocks: usize,
|
|
||||||
block_strategy: BlockStrategy,
|
|
||||||
attestation_strategy: AttestationStrategy,
|
|
||||||
) -> Hash256 {
|
|
||||||
let (mut state, slots) = match block_strategy {
|
|
||||||
BlockStrategy::OnCanonicalHead => {
|
|
||||||
let current_slot: u64 = self.get_current_slot().into();
|
|
||||||
let slots: Vec<Slot> = (current_slot..(current_slot + (num_blocks as u64)))
|
|
||||||
.map(Slot::new)
|
|
||||||
.collect();
|
|
||||||
let state = self.get_current_state();
|
|
||||||
(state, slots)
|
|
||||||
}
|
|
||||||
BlockStrategy::ForkCanonicalChainAt {
|
|
||||||
previous_slot,
|
|
||||||
first_slot,
|
|
||||||
} => {
|
|
||||||
let first_slot_: u64 = first_slot.into();
|
|
||||||
let slots: Vec<Slot> = (first_slot_..(first_slot_ + (num_blocks as u64)))
|
|
||||||
.map(Slot::new)
|
|
||||||
.collect();
|
|
||||||
let state = self
|
|
||||||
.chain
|
|
||||||
.state_at_slot(previous_slot, StateSkipConfig::WithStateRoots)
|
|
||||||
.unwrap();
|
|
||||||
(state, slots)
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let validators = match attestation_strategy {
|
|
||||||
AttestationStrategy::AllValidators => self.get_all_validators(),
|
|
||||||
AttestationStrategy::SomeValidators(vals) => vals,
|
|
||||||
};
|
|
||||||
let state_root = state.update_tree_hash_cache().unwrap();
|
|
||||||
let (_, _, last_produced_block_hash, _) = self
|
|
||||||
.add_attested_blocks_at_slots(state, state_root, &slots, &validators)
|
|
||||||
.await;
|
|
||||||
last_produced_block_hash.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Deprecated: Use add_attested_blocks_at_slots() instead
|
|
||||||
///
|
|
||||||
/// Creates two forks:
|
|
||||||
///
|
|
||||||
/// - The "honest" fork: created by the `honest_validators` who have built `honest_fork_blocks`
|
|
||||||
/// on the head
|
|
||||||
/// - The "faulty" fork: created by the `faulty_validators` who skipped a slot and
|
|
||||||
/// then built `faulty_fork_blocks`.
|
|
||||||
///
|
|
||||||
/// Returns `(honest_head, faulty_head)`, the roots of the blocks at the top of each chain.
|
|
||||||
pub async fn generate_two_forks_by_skipping_a_block(
|
|
||||||
&self,
|
|
||||||
honest_validators: &[usize],
|
|
||||||
faulty_validators: &[usize],
|
|
||||||
honest_fork_blocks: usize,
|
|
||||||
faulty_fork_blocks: usize,
|
|
||||||
) -> (Hash256, Hash256) {
|
|
||||||
let initial_head_slot = self.chain.head_snapshot().beacon_block.slot();
|
|
||||||
|
|
||||||
// Move to the next slot so we may produce some more blocks on the head.
|
|
||||||
self.advance_slot();
|
|
||||||
|
|
||||||
// Extend the chain with blocks where only honest validators agree.
|
|
||||||
let honest_head = self
|
|
||||||
.extend_chain(
|
|
||||||
honest_fork_blocks,
|
|
||||||
BlockStrategy::OnCanonicalHead,
|
|
||||||
AttestationStrategy::SomeValidators(honest_validators.to_vec()),
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
// Go back to the last block where all agreed, and build blocks upon it where only faulty nodes
|
|
||||||
// agree.
|
|
||||||
let faulty_head = self
|
|
||||||
.extend_chain(
|
|
||||||
faulty_fork_blocks,
|
|
||||||
BlockStrategy::ForkCanonicalChainAt {
|
|
||||||
previous_slot: initial_head_slot,
|
|
||||||
// `initial_head_slot + 2` means one slot is skipped.
|
|
||||||
first_slot: initial_head_slot + 2,
|
|
||||||
},
|
|
||||||
AttestationStrategy::SomeValidators(faulty_validators.to_vec()),
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
assert_ne!(honest_head, faulty_head, "forks should be distinct");
|
|
||||||
|
|
||||||
(honest_head, faulty_head)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn process_sync_contributions(
|
|
||||||
&self,
|
|
||||||
sync_contributions: HarnessSyncContributions<E>,
|
|
||||||
) -> Result<(), SyncCommitteeError> {
|
|
||||||
let mut verified_contributions = Vec::with_capacity(sync_contributions.len());
|
|
||||||
|
|
||||||
for (_, contribution_and_proof) in sync_contributions {
|
|
||||||
let signed_contribution_and_proof = contribution_and_proof.unwrap();
|
|
||||||
|
|
||||||
let verified_contribution = self
|
|
||||||
.chain
|
|
||||||
.verify_sync_contribution_for_gossip(signed_contribution_and_proof)?;
|
|
||||||
|
|
||||||
verified_contributions.push(verified_contribution);
|
|
||||||
}
|
|
||||||
|
|
||||||
for verified_contribution in verified_contributions {
|
|
||||||
self.chain
|
|
||||||
.add_contribution_to_block_inclusion_pool(verified_contribution)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Junk `Debug` impl to satistfy certain trait bounds during testing.
|
// Junk `Debug` impl to satistfy certain trait bounds during testing.
|
||||||
|
@ -191,14 +191,14 @@ impl DatabasePubkey {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use crate::test_utils::{BeaconChainHarness, EphemeralTestingSlotClockHarnessType};
|
||||||
use logging::test_logger;
|
use logging::test_logger;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use store::HotColdDB;
|
use store::HotColdDB;
|
||||||
use types::{BeaconState, EthSpec, Keypair, MainnetEthSpec};
|
use types::{BeaconState, EthSpec, Keypair, MainnetEthSpec};
|
||||||
|
|
||||||
type E = MainnetEthSpec;
|
type E = MainnetEthSpec;
|
||||||
type T = EphemeralHarnessType<E>;
|
type T = EphemeralTestingSlotClockHarnessType<E>;
|
||||||
|
|
||||||
fn get_state(validator_count: usize) -> (BeaconState<E>, Vec<Keypair>) {
|
fn get_state(validator_count: usize) -> (BeaconState<E>, Vec<Keypair>) {
|
||||||
let harness = BeaconChainHarness::builder(MainnetEthSpec)
|
let harness = BeaconChainHarness::builder(MainnetEthSpec)
|
||||||
|
@ -48,7 +48,7 @@ async fn merge_with_terminal_block_hash_override() {
|
|||||||
|
|
||||||
spec.terminal_block_hash = genesis_pow_block_hash;
|
spec.terminal_block_hash = genesis_pow_block_hash;
|
||||||
|
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::builder(E::default(), TestingSlotClock)
|
||||||
.spec(spec)
|
.spec(spec)
|
||||||
.logger(logging::test_logger())
|
.logger(logging::test_logger())
|
||||||
.deterministic_keypairs(VALIDATOR_COUNT)
|
.deterministic_keypairs(VALIDATOR_COUNT)
|
||||||
@ -105,7 +105,7 @@ async fn base_altair_merge_with_terminal_block_after_fork() {
|
|||||||
|
|
||||||
let mut execution_payloads = vec![];
|
let mut execution_payloads = vec![];
|
||||||
|
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::builder(E::default(), TestingSlotClock)
|
||||||
.spec(spec)
|
.spec(spec)
|
||||||
.logger(logging::test_logger())
|
.logger(logging::test_logger())
|
||||||
.deterministic_keypairs(VALIDATOR_COUNT)
|
.deterministic_keypairs(VALIDATOR_COUNT)
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
use beacon_chain::test_utils::EphemeralHarnessType;
|
use beacon_chain::test_utils::EphemeralTestingSlotClockHarnessType;
|
||||||
use environment::null_logger;
|
use environment::null_logger;
|
||||||
use http_metrics::Config;
|
use http_metrics::Config;
|
||||||
use reqwest::StatusCode;
|
use reqwest::StatusCode;
|
||||||
@ -7,7 +7,7 @@ use std::sync::Arc;
|
|||||||
use tokio::sync::oneshot;
|
use tokio::sync::oneshot;
|
||||||
use types::MainnetEthSpec;
|
use types::MainnetEthSpec;
|
||||||
|
|
||||||
type Context = http_metrics::Context<EphemeralHarnessType<MainnetEthSpec>>;
|
type Context = http_metrics::Context<EphemeralTestingSlotClockHarnessType<MainnetEthSpec>>;
|
||||||
|
|
||||||
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
||||||
async fn returns_200_ok() {
|
async fn returns_200_ok() {
|
||||||
|
@ -4,28 +4,26 @@ use crate::service::RequestId;
|
|||||||
use crate::sync::manager::RequestId as SyncId;
|
use crate::sync::manager::RequestId as SyncId;
|
||||||
use crate::NetworkMessage;
|
use crate::NetworkMessage;
|
||||||
|
|
||||||
use sloggers::{null::NullLoggerBuilder, Build};
|
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
use beacon_chain::{
|
use beacon_chain::{
|
||||||
builder::{BeaconChainBuilder, Witness},
|
builder::Witness,
|
||||||
eth1_chain::CachingEth1Backend,
|
eth1_chain::CachingEth1Backend,
|
||||||
test_utils::test_spec,
|
test_utils::{BeaconChainHarness, EphemeralSystemTimeSlotClockHarnessType as HarnessType},
|
||||||
};
|
};
|
||||||
|
pub use genesis::{interop_genesis_state, DEFAULT_ETH1_BLOCK_HASH};
|
||||||
use lighthouse_network::{NetworkGlobals, Request};
|
use lighthouse_network::{NetworkGlobals, Request};
|
||||||
use slog::{Drain, Level};
|
use slog::{Drain, Level};
|
||||||
use slot_clock::{SlotClock, SystemTimeSlotClock};
|
use slot_clock::SystemTimeSlotClock;
|
||||||
use std::time::{Duration, SystemTime};
|
use std::time::Duration;
|
||||||
use store::MemoryStore;
|
use store::MemoryStore;
|
||||||
use tokio::sync::mpsc;
|
use tokio::sync::mpsc;
|
||||||
use types::{
|
use types::{
|
||||||
test_utils::{SeedableRng, TestRandom, XorShiftRng},
|
test_utils::{SeedableRng, TestRandom, XorShiftRng},
|
||||||
EthSpec, MinimalEthSpec as E, SignedBeaconBlock,
|
MinimalEthSpec as E, SignedBeaconBlock,
|
||||||
};
|
};
|
||||||
|
|
||||||
type T = Witness<SystemTimeSlotClock, CachingEth1Backend<E>, E, MemoryStore<E>, MemoryStore<E>>;
|
type T = Witness<SystemTimeSlotClock, CachingEth1Backend<E>, E, MemoryStore<E>, MemoryStore<E>>;
|
||||||
const SLOT_DURATION_MILLIS: u64 = 400;
|
|
||||||
|
|
||||||
struct TestRig {
|
struct TestRig {
|
||||||
beacon_processor_rx: mpsc::Receiver<WorkEvent<T>>,
|
beacon_processor_rx: mpsc::Receiver<WorkEvent<T>>,
|
||||||
@ -37,31 +35,6 @@ const D: Duration = Duration::new(0, 0);
|
|||||||
|
|
||||||
impl TestRig {
|
impl TestRig {
|
||||||
fn test_setup(log_level: Option<Level>) -> (BlockLookups<T>, SyncNetworkContext<T>, Self) {
|
fn test_setup(log_level: Option<Level>) -> (BlockLookups<T>, SyncNetworkContext<T>, Self) {
|
||||||
let builder = NullLoggerBuilder;
|
|
||||||
let log = builder.build().expect("should build logger");
|
|
||||||
let store =
|
|
||||||
store::HotColdDB::open_ephemeral(store::StoreConfig::default(), E::default_spec(), log)
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
// Initialise a new beacon chain
|
|
||||||
let chain = BeaconChainBuilder::new(E)
|
|
||||||
.custom_spec(test_spec::<E>())
|
|
||||||
.store(Arc::new(store))
|
|
||||||
.dummy_eth1_backend()
|
|
||||||
.expect("should build dummy backend")
|
|
||||||
.slot_clock(SystemTimeSlotClock::new(
|
|
||||||
types::Slot::new(0),
|
|
||||||
Duration::from_secs(
|
|
||||||
SystemTime::now()
|
|
||||||
.duration_since(SystemTime::UNIX_EPOCH)
|
|
||||||
.unwrap()
|
|
||||||
.as_secs(),
|
|
||||||
),
|
|
||||||
Duration::from_millis(SLOT_DURATION_MILLIS),
|
|
||||||
))
|
|
||||||
.build()
|
|
||||||
.expect("should build");
|
|
||||||
|
|
||||||
let log = {
|
let log = {
|
||||||
let decorator = slog_term::TermDecorator::new().build();
|
let decorator = slog_term::TermDecorator::new().build();
|
||||||
let drain = slog_term::FullFormat::new(decorator).build().fuse();
|
let drain = slog_term::FullFormat::new(decorator).build().fuse();
|
||||||
@ -74,6 +47,16 @@ impl TestRig {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Initialise a new beacon chain
|
||||||
|
let harness = BeaconChainHarness::<HarnessType<E>>::builder(E::default())
|
||||||
|
.default_spec()
|
||||||
|
.logger(log.clone())
|
||||||
|
.deterministic_keypairs(8)
|
||||||
|
.fresh_ephemeral_store()
|
||||||
|
.build();
|
||||||
|
|
||||||
|
let chain = harness.chain;
|
||||||
|
|
||||||
let (beacon_processor_tx, beacon_processor_rx) = mpsc::channel(100);
|
let (beacon_processor_tx, beacon_processor_rx) = mpsc::channel(100);
|
||||||
let (network_tx, network_rx) = mpsc::unbounded_channel();
|
let (network_tx, network_rx) = mpsc::unbounded_channel();
|
||||||
let rng = XorShiftRng::from_seed([42; 16]);
|
let rng = XorShiftRng::from_seed([42; 16]);
|
||||||
@ -89,7 +72,7 @@ impl TestRig {
|
|||||||
network_tx,
|
network_tx,
|
||||||
globals,
|
globals,
|
||||||
beacon_processor_tx,
|
beacon_processor_tx,
|
||||||
Arc::new(chain),
|
chain,
|
||||||
log.new(slog::o!("component" => "network_context")),
|
log.new(slog::o!("component" => "network_context")),
|
||||||
)
|
)
|
||||||
};
|
};
|
||||||
|
@ -788,7 +788,7 @@ mod release_tests {
|
|||||||
validator_count: usize,
|
validator_count: usize,
|
||||||
spec: Option<ChainSpec>,
|
spec: Option<ChainSpec>,
|
||||||
) -> BeaconChainHarness<EphemeralHarnessType<E>> {
|
) -> BeaconChainHarness<EphemeralHarnessType<E>> {
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::builder(E::default(), TestingSlotClock)
|
||||||
.spec_or_default(spec)
|
.spec_or_default(spec)
|
||||||
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
||||||
.fresh_ephemeral_store()
|
.fresh_ephemeral_store()
|
||||||
|
@ -67,6 +67,12 @@ impl SlotClock for SystemTimeSlotClock {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl From<ManualSlotClock> for SystemTimeSlotClock {
|
||||||
|
fn from(clock: ManualSlotClock) -> Self {
|
||||||
|
SystemTimeSlotClock { clock }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
@ -5,7 +5,7 @@ use std::sync::Mutex;
|
|||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
use beacon_chain::test_utils::{
|
use beacon_chain::test_utils::{
|
||||||
AttestationStrategy, BeaconChainHarness, BlockStrategy, EphemeralHarnessType,
|
AttestationStrategy, BeaconChainHarness, BlockStrategy, EphemeralTestingSlotClockHarnessType,
|
||||||
};
|
};
|
||||||
use beacon_chain::{
|
use beacon_chain::{
|
||||||
BeaconChain, BeaconChainError, BeaconForkChoiceStore, ChainConfig, ForkChoiceError,
|
BeaconChain, BeaconChainError, BeaconForkChoiceStore, ChainConfig, ForkChoiceError,
|
||||||
|
@ -10,7 +10,9 @@ use crate::{
|
|||||||
per_block_processing::{process_operations, verify_exit::verify_exit},
|
per_block_processing::{process_operations, verify_exit::verify_exit},
|
||||||
BlockSignatureStrategy, ConsensusContext, VerifyBlockRoot, VerifySignatures,
|
BlockSignatureStrategy, ConsensusContext, VerifyBlockRoot, VerifySignatures,
|
||||||
};
|
};
|
||||||
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use beacon_chain::test_utils::{
|
||||||
|
BeaconChainHarness, EphemeralTestingSlotClockHarnessType as HarnessType,
|
||||||
|
};
|
||||||
use lazy_static::lazy_static;
|
use lazy_static::lazy_static;
|
||||||
use ssz_types::Bitfield;
|
use ssz_types::Bitfield;
|
||||||
use test_utils::generate_deterministic_keypairs;
|
use test_utils::generate_deterministic_keypairs;
|
||||||
@ -30,11 +32,11 @@ lazy_static! {
|
|||||||
async fn get_harness<E: EthSpec>(
|
async fn get_harness<E: EthSpec>(
|
||||||
epoch_offset: u64,
|
epoch_offset: u64,
|
||||||
num_validators: usize,
|
num_validators: usize,
|
||||||
) -> BeaconChainHarness<EphemeralHarnessType<E>> {
|
) -> BeaconChainHarness<HarnessType<E>> {
|
||||||
// Set the state and block to be in the last slot of the `epoch_offset`th epoch.
|
// Set the state and block to be in the last slot of the `epoch_offset`th epoch.
|
||||||
let last_slot_of_epoch =
|
let last_slot_of_epoch =
|
||||||
(MainnetEthSpec::genesis_epoch() + epoch_offset).end_slot(E::slots_per_epoch());
|
(MainnetEthSpec::genesis_epoch() + epoch_offset).end_slot(E::slots_per_epoch());
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::<HarnessType<E>>::builder(E::default())
|
||||||
.default_spec()
|
.default_spec()
|
||||||
.keypairs(KEYPAIRS[0..num_validators].to_vec())
|
.keypairs(KEYPAIRS[0..num_validators].to_vec())
|
||||||
.fresh_ephemeral_store()
|
.fresh_ephemeral_store()
|
||||||
|
@ -1,10 +1,10 @@
|
|||||||
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralTestingSlotClockHarnessType};
|
||||||
use types::{BeaconState, EthSpec, MainnetEthSpec};
|
use types::{BeaconState, EthSpec, MainnetEthSpec};
|
||||||
|
|
||||||
const TREE_HASH_LOOPS: usize = 1_000;
|
const TREE_HASH_LOOPS: usize = 1_000;
|
||||||
const VALIDATOR_COUNT: usize = 1_000;
|
const VALIDATOR_COUNT: usize = 1_000;
|
||||||
|
|
||||||
fn get_harness<T: EthSpec>() -> BeaconChainHarness<EphemeralHarnessType<T>> {
|
fn get_harness<T: EthSpec>() -> BeaconChainHarness<EphemeralTestingSlotClockHarnessType<T>> {
|
||||||
let harness = BeaconChainHarness::builder(T::default())
|
let harness = BeaconChainHarness::builder(T::default())
|
||||||
.default_spec()
|
.default_spec()
|
||||||
.deterministic_keypairs(VALIDATOR_COUNT)
|
.deterministic_keypairs(VALIDATOR_COUNT)
|
||||||
|
@ -1,6 +1,8 @@
|
|||||||
#![cfg(test)]
|
#![cfg(test)]
|
||||||
use crate::test_utils::*;
|
use crate::test_utils::*;
|
||||||
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use beacon_chain::test_utils::{
|
||||||
|
BeaconChainHarness, EphemeralTestingSlotClockHarnessType as HarnessType,
|
||||||
|
};
|
||||||
use beacon_chain::types::*;
|
use beacon_chain::types::*;
|
||||||
use swap_or_not_shuffle::shuffle_list;
|
use swap_or_not_shuffle::shuffle_list;
|
||||||
|
|
||||||
@ -11,7 +13,7 @@ lazy_static! {
|
|||||||
static ref KEYPAIRS: Vec<Keypair> = generate_deterministic_keypairs(VALIDATOR_COUNT);
|
static ref KEYPAIRS: Vec<Keypair> = generate_deterministic_keypairs(VALIDATOR_COUNT);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_harness<E: EthSpec>(validator_count: usize) -> BeaconChainHarness<EphemeralHarnessType<E>> {
|
fn get_harness<E: EthSpec>(validator_count: usize) -> BeaconChainHarness<HarnessType<E>> {
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::builder(E::default())
|
||||||
.default_spec()
|
.default_spec()
|
||||||
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
||||||
|
@ -2,8 +2,8 @@
|
|||||||
use crate::test_utils::*;
|
use crate::test_utils::*;
|
||||||
use crate::test_utils::{SeedableRng, XorShiftRng};
|
use crate::test_utils::{SeedableRng, XorShiftRng};
|
||||||
use beacon_chain::test_utils::{
|
use beacon_chain::test_utils::{
|
||||||
interop_genesis_state_with_eth1, test_spec, BeaconChainHarness, EphemeralHarnessType,
|
interop_genesis_state, test_spec, BeaconChainHarness,
|
||||||
DEFAULT_ETH1_BLOCK_HASH,
|
EphemeralTestingSlotClockHarnessType as HarnessType, DEFAULT_ETH1_BLOCK_HASH,
|
||||||
};
|
};
|
||||||
use beacon_chain::types::{
|
use beacon_chain::types::{
|
||||||
test_utils::TestRandom, BeaconState, BeaconStateAltair, BeaconStateBase, BeaconStateError,
|
test_utils::TestRandom, BeaconState, BeaconStateAltair, BeaconStateBase, BeaconStateError,
|
||||||
@ -28,7 +28,7 @@ lazy_static! {
|
|||||||
async fn get_harness<E: EthSpec>(
|
async fn get_harness<E: EthSpec>(
|
||||||
validator_count: usize,
|
validator_count: usize,
|
||||||
slot: Slot,
|
slot: Slot,
|
||||||
) -> BeaconChainHarness<EphemeralHarnessType<E>> {
|
) -> BeaconChainHarness<HarnessType<E>> {
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::builder(E::default())
|
||||||
.default_spec()
|
.default_spec()
|
||||||
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
||||||
|
@ -62,7 +62,7 @@
|
|||||||
//! --exclude-post-block-thc
|
//! --exclude-post-block-thc
|
||||||
//! ```
|
//! ```
|
||||||
use beacon_chain::{
|
use beacon_chain::{
|
||||||
test_utils::EphemeralHarnessType, validator_pubkey_cache::ValidatorPubkeyCache,
|
test_utils::EphemeralTestingSlotClockHarnessType, validator_pubkey_cache::ValidatorPubkeyCache,
|
||||||
};
|
};
|
||||||
use clap::ArgMatches;
|
use clap::ArgMatches;
|
||||||
use clap_utils::{parse_optional, parse_required};
|
use clap_utils::{parse_optional, parse_required};
|
||||||
@ -297,7 +297,7 @@ fn do_transition<T: EthSpec>(
|
|||||||
block: SignedBeaconBlock<T>,
|
block: SignedBeaconBlock<T>,
|
||||||
mut state_root_opt: Option<Hash256>,
|
mut state_root_opt: Option<Hash256>,
|
||||||
config: &Config,
|
config: &Config,
|
||||||
validator_pubkey_cache: &ValidatorPubkeyCache<EphemeralHarnessType<T>>,
|
validator_pubkey_cache: &ValidatorPubkeyCache<EphemeralTestingSlotClockHarnessType<T>>,
|
||||||
spec: &ChainSpec,
|
spec: &ChainSpec,
|
||||||
) -> Result<BeaconState<T>, String> {
|
) -> Result<BeaconState<T>, String> {
|
||||||
if !config.exclude_cache_builds {
|
if !config.exclude_cache_builds {
|
||||||
|
@ -6,7 +6,7 @@ use beacon_chain::{
|
|||||||
attestation_verification::{
|
attestation_verification::{
|
||||||
obtain_indexed_attestation_and_committees_per_slot, VerifiedAttestation,
|
obtain_indexed_attestation_and_committees_per_slot, VerifiedAttestation,
|
||||||
},
|
},
|
||||||
test_utils::{BeaconChainHarness, EphemeralHarnessType},
|
test_utils::{BeaconChainHarness, EphemeralTestingSlotClockHarnessType},
|
||||||
BeaconChainTypes, CachedHead, CountUnrealized, NotifyExecutionLayer,
|
BeaconChainTypes, CachedHead, CountUnrealized, NotifyExecutionLayer,
|
||||||
};
|
};
|
||||||
use execution_layer::{json_structures::JsonPayloadStatusV1Status, PayloadStatusV1};
|
use execution_layer::{json_structures::JsonPayloadStatusV1Status, PayloadStatusV1};
|
||||||
@ -286,7 +286,7 @@ impl<E: EthSpec> Case for ForkChoiceTest<E> {
|
|||||||
|
|
||||||
/// A testing rig used to execute a test case.
|
/// A testing rig used to execute a test case.
|
||||||
struct Tester<E: EthSpec> {
|
struct Tester<E: EthSpec> {
|
||||||
harness: BeaconChainHarness<EphemeralHarnessType<E>>,
|
harness: BeaconChainHarness<EphemeralTestingSlotClockHarnessType<E>>,
|
||||||
spec: ChainSpec,
|
spec: ChainSpec,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -306,14 +306,15 @@ impl<E: EthSpec> Tester<E> {
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness =
|
||||||
.spec(spec.clone())
|
BeaconChainHarness::<EphemeralTestingSlotClockHarnessType<E>>::builder(E::default())
|
||||||
.keypairs(vec![])
|
.spec(spec.clone())
|
||||||
.genesis_state_ephemeral_store(case.anchor_state.clone())
|
.keypairs(vec![])
|
||||||
.mock_execution_layer()
|
.genesis_state_ephemeral_store(case.anchor_state.clone())
|
||||||
.recalculate_fork_times_with_genesis(0)
|
.mock_execution_layer()
|
||||||
.mock_execution_layer_all_payloads_valid()
|
.recalculate_fork_times_with_genesis(0)
|
||||||
.build();
|
.mock_execution_layer_all_payloads_valid()
|
||||||
|
.build();
|
||||||
|
|
||||||
if harness.chain.genesis_block_root != case.anchor_block.canonical_root() {
|
if harness.chain.genesis_block_root != case.anchor_block.canonical_root() {
|
||||||
// This check will need to be removed if/when the fork-choice tests use a non-genesis
|
// This check will need to be removed if/when the fork-choice tests use a non-genesis
|
||||||
@ -469,11 +470,12 @@ impl<E: EthSpec> Tester<E> {
|
|||||||
.map_err(|e| {
|
.map_err(|e| {
|
||||||
Error::InternalError(format!("attestation indexing failed with {:?}", e))
|
Error::InternalError(format!("attestation indexing failed with {:?}", e))
|
||||||
})?;
|
})?;
|
||||||
let verified_attestation: ManuallyVerifiedAttestation<EphemeralHarnessType<E>> =
|
let verified_attestation: ManuallyVerifiedAttestation<
|
||||||
ManuallyVerifiedAttestation {
|
EphemeralTestingSlotClockHarnessType<E>,
|
||||||
attestation,
|
> = ManuallyVerifiedAttestation {
|
||||||
indexed_attestation,
|
attestation,
|
||||||
};
|
indexed_attestation,
|
||||||
|
};
|
||||||
|
|
||||||
self.harness
|
self.harness
|
||||||
.chain
|
.chain
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
use super::*;
|
use super::*;
|
||||||
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralTestingSlotClockHarnessType};
|
||||||
use state_processing::{
|
use state_processing::{
|
||||||
per_block_processing, per_block_processing::errors::ExitInvalid, BlockProcessingError,
|
per_block_processing, per_block_processing::errors::ExitInvalid, BlockProcessingError,
|
||||||
BlockSignatureStrategy, ConsensusContext, VerifyBlockRoot,
|
BlockSignatureStrategy, ConsensusContext, VerifyBlockRoot,
|
||||||
@ -18,8 +18,12 @@ struct ExitTest {
|
|||||||
#[allow(clippy::type_complexity)]
|
#[allow(clippy::type_complexity)]
|
||||||
state_modifier: Box<dyn FnOnce(&mut BeaconState<E>)>,
|
state_modifier: Box<dyn FnOnce(&mut BeaconState<E>)>,
|
||||||
#[allow(clippy::type_complexity)]
|
#[allow(clippy::type_complexity)]
|
||||||
block_modifier:
|
block_modifier: Box<
|
||||||
Box<dyn FnOnce(&BeaconChainHarness<EphemeralHarnessType<E>>, &mut BeaconBlock<E>)>,
|
dyn FnOnce(
|
||||||
|
&BeaconChainHarness<EphemeralTestingSlotClockHarnessType<E>>,
|
||||||
|
&mut BeaconBlock<E>,
|
||||||
|
),
|
||||||
|
>,
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
expected: Result<(), BlockProcessingError>,
|
expected: Result<(), BlockProcessingError>,
|
||||||
}
|
}
|
||||||
|
@ -2,7 +2,7 @@
|
|||||||
mod macros;
|
mod macros;
|
||||||
mod exit;
|
mod exit;
|
||||||
|
|
||||||
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralHarnessType};
|
use beacon_chain::test_utils::{BeaconChainHarness, EphemeralTestingSlotClockHarnessType};
|
||||||
use lazy_static::lazy_static;
|
use lazy_static::lazy_static;
|
||||||
use ssz::Encode;
|
use ssz::Encode;
|
||||||
use std::env;
|
use std::env;
|
||||||
@ -53,7 +53,7 @@ lazy_static! {
|
|||||||
async fn get_harness<E: EthSpec>(
|
async fn get_harness<E: EthSpec>(
|
||||||
slot: Slot,
|
slot: Slot,
|
||||||
validator_count: usize,
|
validator_count: usize,
|
||||||
) -> BeaconChainHarness<EphemeralHarnessType<E>> {
|
) -> BeaconChainHarness<EphemeralTestingSlotClockHarnessType<E>> {
|
||||||
let harness = BeaconChainHarness::builder(E::default())
|
let harness = BeaconChainHarness::builder(E::default())
|
||||||
.default_spec()
|
.default_spec()
|
||||||
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
.keypairs(KEYPAIRS[0..validator_count].to_vec())
|
||||||
|
Loading…
Reference in New Issue
Block a user