Skip to content

Commit

Permalink
[test] real_epoch_managers doesn't need genesis.config (#10597)
Browse files Browse the repository at this point in the history
Buildup PRs

#10590
#10591
#10594

Next PR I'll try using the real epoch manager as default and let's see
what happens...
  • Loading branch information
Shreyan Gupta authored Feb 9, 2024
1 parent 68b1bbc commit c6afdd7
Show file tree
Hide file tree
Showing 35 changed files with 130 additions and 128 deletions.
7 changes: 3 additions & 4 deletions chain/client/src/test_utils/test_env_builder.rs
Original file line number Diff line number Diff line change
Expand Up @@ -245,14 +245,13 @@ impl TestEnvBuilder {
self
}

pub fn real_epoch_managers(self, genesis_config: &GenesisConfig) -> Self {
self.real_epoch_managers_with_test_overrides(genesis_config, None)
pub fn real_epoch_managers(self) -> Self {
self.real_epoch_managers_with_test_overrides(None)
}

/// Constructs real EpochManager implementations for each instance.
pub fn real_epoch_managers_with_test_overrides(
self,
genesis_config: &GenesisConfig,
test_overrides: Option<AllEpochConfigTestOverrides>,
) -> Self {
assert!(
Expand All @@ -264,7 +263,7 @@ impl TestEnvBuilder {
.map(|i| {
EpochManager::new_arc_handle_with_test_overrides(
ret.stores.as_ref().unwrap()[i].clone(),
genesis_config,
&ret.genesis_config,
test_overrides.clone(),
)
})
Expand Down
2 changes: 1 addition & 1 deletion integration-tests/src/tests/client/benchmarks.rs
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ fn benchmark_large_chunk_production_time() {

let genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down
6 changes: 3 additions & 3 deletions integration-tests/src/tests/client/block_corruption.rs
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ fn change_shard_id_to_invalid() {
let epoch_length = 5000000;
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -180,7 +180,7 @@ fn check_process_flipped_block_fails_on_bit(
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down
11 changes: 6 additions & 5 deletions integration-tests/src/tests/client/challenges.rs
Original file line number Diff line number Diff line change
Expand Up @@ -60,8 +60,8 @@ fn test_block_with_challenges() {
#[test]
fn test_invalid_chunk_state() {
let genesis = Genesis::test(vec!["test0".parse().unwrap()], 1);
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
env.produce_block(0, 1);
Expand Down Expand Up @@ -332,9 +332,10 @@ fn challenge(

#[test]
fn test_verify_chunk_invalid_state_challenge() {
let genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.min_gas_price = 0;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
let signer = InMemorySigner::from_seed("test0".parse().unwrap(), KeyType::ED25519, "test0");
Expand Down
10 changes: 5 additions & 5 deletions integration-tests/src/tests/client/cold_storage.rs
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,7 @@ fn test_storage_after_commit_of_cold_update() {
genesis.config.epoch_length = epoch_length;
genesis.config.min_gas_price = 0;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -233,7 +233,7 @@ fn test_cold_db_head_update() {
let cold_db = storage.cold_db().unwrap();
let mut env = TestEnv::builder(&genesis.config)
.stores(vec![hot_store.clone()])
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -268,7 +268,7 @@ fn test_cold_db_copy_with_height_skips() {
genesis.config.epoch_length = epoch_length;
genesis.config.min_gas_price = 0;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -365,7 +365,7 @@ fn test_initial_copy_to_cold(batch_size: usize) {
let mut genesis = Genesis::test(vec![test0(), test1()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -449,7 +449,7 @@ fn test_cold_loop_on_gc_boundary() {
.archive(true)
.save_trie_changes(true)
.stores(vec![hot_store.clone()])
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down
8 changes: 4 additions & 4 deletions integration-tests/src/tests/client/epoch_sync.rs
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ fn test_continuous_epoch_sync_info_population() {
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -247,7 +247,7 @@ fn test_epoch_sync_data_hash_from_epoch_sync_info() {
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down Expand Up @@ -327,11 +327,11 @@ fn test_node_after_simulated_sync() {
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;

let mut env = TestEnv::default_builder()
let mut env = TestEnv::builder(&genesis.config)
.clients_count(num_clients)
.real_stores()
.use_state_snapshots()
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,8 +52,8 @@ fn test_transaction_hash_collision() {
let epoch_length = 5;
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
let genesis_block = env.clients[0].chain.get_block_by_height(0).unwrap();
Expand Down Expand Up @@ -125,8 +125,8 @@ fn get_status_of_tx_hash_collision_for_near_implicit_account(
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = protocol_version;
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
let genesis_block = env.clients[0].chain.get_block_by_height(0).unwrap();
Expand Down Expand Up @@ -242,8 +242,9 @@ fn test_chunk_transaction_validity() {
let epoch_length = 5;
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
genesis.config.min_gas_price = 0;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
let genesis_block = env.clients[0].chain.get_block_by_height(0).unwrap();
Expand Down Expand Up @@ -276,8 +277,8 @@ fn test_transaction_nonce_too_large() {
let epoch_length = 5;
let mut genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
genesis.config.epoch_length = epoch_length;
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
let genesis_block = env.clients[0].chain.get_block_by_height(0).unwrap();
Expand Down Expand Up @@ -344,7 +345,7 @@ fn test_request_chunks_for_orphan() {
let mut env = TestEnv::builder(&genesis.config)
.clients_count(num_clients)
.validator_seats(num_validators as usize)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
Expand Down Expand Up @@ -484,7 +485,7 @@ fn test_processing_chunks_sanity() {
let mut env = TestEnv::builder(&genesis.config)
.clients_count(num_clients)
.validator_seats(num_validators as usize)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes(&genesis)
.build();
Expand Down Expand Up @@ -583,7 +584,7 @@ impl ChunkForwardingOptimizationTestData {
let env = TestEnv::builder(&genesis.config)
.clients_count(num_clients)
.validator_seats(num_validators as usize)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes(&genesis)
.build();
Expand Down Expand Up @@ -812,7 +813,7 @@ fn test_processing_blocks_async() {
let mut env = TestEnv::builder(&genesis.config)
.clients_count(num_clients)
.validator_seats(num_validators as usize)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes(&genesis)
.build();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ fn test_account_id_in_function_call_permission_upgrade() {
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = old_protocol_version;
TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
vec![RuntimeConfigStore::new(None)],
Expand Down Expand Up @@ -93,7 +93,7 @@ fn test_very_long_account_id() {
let mut env = {
let genesis = Genesis::test(vec!["test0".parse().unwrap(), "test1".parse().unwrap()], 1);
TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
vec![RuntimeConfigStore::new(None)],
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ impl AdversarialBehaviorTestData {
.clock(clock.clock())
.clients_count(num_clients)
.validator_seats(num_validators as usize)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes(&genesis)
.build();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ fn compare_node_counts() {
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = old_protocol_version;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
vec![RuntimeConfigStore::new(None)],
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -57,8 +57,8 @@ fn exec_meta_transaction(
Genesis::test(vec![validator, user.clone(), receiver.clone(), relayer.clone()], 1);
genesis.config.epoch_length = 1000;
genesis.config.protocol_version = protocol_version;
let mut env = TestEnv::default_builder()
.real_epoch_managers(&genesis.config)
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ fn prepare_env_with_contract(
genesis.config.protocol_version = protocol_version;
let runtime_config = near_parameters::RuntimeConfigStore::new(None);
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(&genesis, vec![runtime_config])
.build();
deploy_test_contract(&mut env, account, &contract, epoch_length, 1);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ fn process_blocks_with_storage_usage_fix(
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = ProtocolFeature::FixStorageUsage.protocol_version() - 1;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();
for i in 1..=16 {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ fn test_flat_storage_upgrade() {
genesis.config.protocol_version = old_protocol_version;
let runtime_config = near_parameters::RuntimeConfigStore::new(None);
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(&genesis, vec![runtime_config])
.build();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ fn test_in_memory_trie_node_consistency() {
let mut env = TestEnv::builder(&genesis.config)
.clients(vec!["account0".parse().unwrap(), "account1".parse().unwrap()])
.stores(stores.clone())
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes_with_trie_config(
&genesis,
Expand Down Expand Up @@ -179,7 +179,7 @@ fn test_in_memory_trie_node_consistency() {
let mut env = TestEnv::builder(&genesis.config)
.clients(vec!["account0".parse().unwrap(), "account1".parse().unwrap()])
.stores(stores.clone())
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes_with_trie_config(
&genesis,
Expand Down Expand Up @@ -210,7 +210,7 @@ fn test_in_memory_trie_node_consistency() {
let mut env = TestEnv::builder(&genesis.config)
.clients(vec!["account0".parse().unwrap(), "account1".parse().unwrap()])
.stores(stores)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes_with_trie_config(
&genesis,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ fn test_deploy_cost_increased() {
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = old_protocol_version;
TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
vec![RuntimeConfigStore::new(None)],
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ fn assert_compute_limit_reached(
genesis.config.protocol_version = old_protocol_version;
genesis.config.gas_limit = genesis.config.gas_limit / gas_divider;
TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(&genesis, vec![runtime_config_store])
.build()
};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ fn verify_contract_limits_upgrade(
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = old_protocol_version;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
vec![RuntimeConfigStore::new(None)],
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ fn protocol_upgrade() {
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = old_protocol_version;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.track_all_shards()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
Expand Down
2 changes: 1 addition & 1 deletion integration-tests/src/tests/client/features/nearvm.rs
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ fn test_nearvm_upgrade() {
genesis.config.epoch_length = epoch_length;
genesis.config.protocol_version = old_protocol_version;
let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes_with_runtime_config_store(
&genesis,
vec![RuntimeConfigStore::new(None)],
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ fn run_test(
let migration_data = load_migration_data(&genesis.config.chain_id);

let mut env = TestEnv::builder(&genesis.config)
.real_epoch_managers(&genesis.config)
.real_epoch_managers()
.nightshade_runtimes(&genesis)
.build();

Expand Down
Loading

0 comments on commit c6afdd7

Please sign in to comment.