Add custom BlocktreeConfig for bad tests that break with the default

This commit is contained in:
Michael Vines 2019-02-14 09:48:14 -08:00
parent cd9dac4c7e
commit 3f6aba23dd
2 changed files with 21 additions and 8 deletions

View File

@ -1692,7 +1692,8 @@ mod tests {
pub fn test_insert_data_blobs_consecutive() { pub fn test_insert_data_blobs_consecutive() {
let blocktree_path = get_tmp_ledger_path("test_insert_data_blobs_consecutive"); let blocktree_path = get_tmp_ledger_path("test_insert_data_blobs_consecutive");
{ {
let blocktree = Blocktree::open(&blocktree_path).unwrap(); let config = BlocktreeConfig::new(32);
let blocktree = Blocktree::open_config(&blocktree_path, &config).unwrap();
let slot = 0; let slot = 0;
let parent_slot = 0; let parent_slot = 0;
// Write entries // Write entries

View File

@ -211,12 +211,13 @@ impl Service for Tvu {
pub mod tests { pub mod tests {
use super::*; use super::*;
use crate::bank::Bank; use crate::bank::Bank;
use crate::blocktree::get_tmp_ledger_path; use crate::blocktree::{get_tmp_ledger_path, BlocktreeConfig};
use crate::cluster_info::{ClusterInfo, Node}; use crate::cluster_info::{ClusterInfo, Node};
use crate::entry::next_entry_mut; use crate::entry::next_entry_mut;
use crate::entry::EntrySlice; use crate::entry::EntrySlice;
use crate::genesis_block::GenesisBlock; use crate::genesis_block::GenesisBlock;
use crate::gossip_service::GossipService; use crate::gossip_service::GossipService;
use crate::leader_scheduler::LeaderSchedulerConfig;
use crate::packet::index_blobs; use crate::packet::index_blobs;
use crate::storage_stage::STORAGE_ROTATE_TEST_COUNT; use crate::storage_stage::STORAGE_ROTATE_TEST_COUNT;
use crate::streamer; use crate::streamer;
@ -291,14 +292,14 @@ pub mod tests {
let target2 = Node::new_localhost(); let target2 = Node::new_localhost();
let exit = Arc::new(AtomicBool::new(false)); let exit = Arc::new(AtomicBool::new(false));
//start cluster_info_l // start cluster_info_l
let mut cluster_info_l = ClusterInfo::new(leader.info.clone()); let mut cluster_info_l = ClusterInfo::new(leader.info.clone());
cluster_info_l.set_leader(leader.info.id); cluster_info_l.set_leader(leader.info.id);
let cref_l = Arc::new(RwLock::new(cluster_info_l)); let cref_l = Arc::new(RwLock::new(cluster_info_l));
let dr_l = new_gossip(cref_l, leader.sockets.gossip, exit.clone()); let dr_l = new_gossip(cref_l, leader.sockets.gossip, exit.clone());
//start cluster_info2 // start cluster_info2
let mut cluster_info2 = ClusterInfo::new(target2.info.clone()); let mut cluster_info2 = ClusterInfo::new(target2.info.clone());
cluster_info2.insert_info(leader.info.clone()); cluster_info2.insert_info(leader.info.clone());
cluster_info2.set_leader(leader.info.id); cluster_info2.set_leader(leader.info.id);
@ -322,13 +323,22 @@ pub mod tests {
r_responder, r_responder,
); );
// TODO: Fix this test so it always works with the default
// LeaderSchedulerConfig/BlocktreeConfig configuration
let mut leader_scheduler_config = LeaderSchedulerConfig::default();
leader_scheduler_config.ticks_per_slot = 64;
let blocktree_config = BlocktreeConfig::new(leader_scheduler_config.ticks_per_slot);
let starting_balance = 10_000; let starting_balance = 10_000;
let (genesis_block, mint_keypair) = GenesisBlock::new(starting_balance); let (genesis_block, mint_keypair) = GenesisBlock::new(starting_balance);
let tvu_addr = target1.info.tvu; let tvu_addr = target1.info.tvu;
let bank = Arc::new(Bank::new(&genesis_block)); let bank = Arc::new(Bank::new_with_leader_scheduler_config(
&genesis_block,
&leader_scheduler_config,
));
assert_eq!(bank.get_balance(&mint_keypair.pubkey()), starting_balance); assert_eq!(bank.get_balance(&mint_keypair.pubkey()), starting_balance);
//start cluster_info1 // start cluster_info1
let mut cluster_info1 = ClusterInfo::new(target1.info.clone()); let mut cluster_info1 = ClusterInfo::new(target1.info.clone());
cluster_info1.insert_info(leader.info.clone()); cluster_info1.insert_info(leader.info.clone());
cluster_info1.set_leader(leader.info.id); cluster_info1.set_leader(leader.info.id);
@ -337,7 +347,9 @@ pub mod tests {
let mut cur_hash = Hash::default(); let mut cur_hash = Hash::default();
let blocktree_path = get_tmp_ledger_path("test_replay"); let blocktree_path = get_tmp_ledger_path("test_replay");
let (blocktree, l_sender, l_receiver) = Blocktree::open_with_signal(&blocktree_path)
let (blocktree, l_sender, l_receiver) =
Blocktree::open_with_config_signal(&blocktree_path, &blocktree_config)
.expect("Expected to successfully open ledger"); .expect("Expected to successfully open ledger");
let vote_account_keypair = Arc::new(Keypair::new()); let vote_account_keypair = Arc::new(Keypair::new());
let voting_keypair = VotingKeypair::new_local(&vote_account_keypair); let voting_keypair = VotingKeypair::new_local(&vote_account_keypair);