blockworks-foundation
/
solana-with-rpc-optimizations
mirror of https://github.com/blockworks-foundation/solana-with-rpc-optimizations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cleanup test

This commit is contained in:
Greg Fitzgerald 2018-06-19 11:40:29 -07:00 committed by Greg Fitzgerald
parent ceafc29040
commit 2f4a92e352
1 changed files with 19 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/crdt.rs
View File

@ -825,7 +825,6 @@ mod tests {
use packet::BlobRecycler; use packet::BlobRecycler;
use result::Error; use result::Error;
use signature::{KeyPair, KeyPairUtil}; use signature::{KeyPair, KeyPairUtil};
use std;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::mpsc::channel; use std::sync::mpsc::channel;
use std::sync::{Arc, RwLock}; use std::sync::{Arc, RwLock};
@ -1131,49 +1130,35 @@ mod tests {
fn run_window_request_with_backoff() { fn run_window_request_with_backoff() {
let window = default_window(); let window = default_window();
let mut me = ReplicatedData::new( let mut me = ReplicatedData::new_leader(&"127.0.0.1:1234".parse().unwrap());
KeyPair::new().pubkey(),
"127.0.0.1:1234".parse().unwrap(),
"127.0.0.1:1235".parse().unwrap(),
"127.0.0.1:1236".parse().unwrap(),
"127.0.0.1:1237".parse().unwrap(),
"127.0.0.1:1238".parse().unwrap(),
);
let mock_peer = ReplicatedData::new(
KeyPair::new().pubkey(),
"127.0.0.1:1234".parse().unwrap(),
"127.0.0.1:1235".parse().unwrap(),
"127.0.0.1:1236".parse().unwrap(),
"127.0.0.1:1237".parse().unwrap(),
"127.0.0.1:1238".parse().unwrap(),
);
me.current_leader_id = me.id; me.current_leader_id = me.id;
let mock_peer = ReplicatedData::new_leader(&"127.0.0.1:1234".parse().unwrap());
let recycler = BlobRecycler::default(); let recycler = BlobRecycler::default();
let num_requests: u32 = 64;
// Simulate handling a repair request from mock_peer // Simulate handling a repair request from mock_peer
let rv = Crdt::run_window_request(&window, &me, &mock_peer, 0, &recycler); let rv = Crdt::run_window_request(&window, &me, &mock_peer, 0, &recycler);
assert!(rv.is_none()); assert!(rv.is_none());
let out = recycler.allocate(); let blob = recycler.allocate();
out.write().unwrap().meta.size = 200; let blob_size = 200;
window.write().unwrap()[0] = Some(out); blob.write().unwrap().meta.size = blob_size;
let range: std::ops::Range<u32> = 0..num_requests; window.write().unwrap()[0] = Some(blob);
for i in range { let num_requests: u32 = 64;
let rv = Crdt::run_window_request(&window, &me, &mock_peer, 0, &recycler); for i in 0..num_requests {
assert!(rv.is_some()); let shared_blob =
let v = rv.unwrap(); Crdt::run_window_request(&window, &me, &mock_peer, 0, &recycler).unwrap();
let blob = v.read().unwrap(); let blob = shared_blob.read().unwrap();
// Test we copied the blob // Test we copied the blob
assert_eq!(blob.meta.size, 200); assert_eq!(blob.meta.size, blob_size);
if i != 0 && !(i.is_power_of_two()) { let id = if i == 0 || i.is_power_of_two() {
assert_eq!(blob.get_id().unwrap(), mock_peer.id); me.id
} else { } else {
assert_eq!(blob.get_id().unwrap(), me.id); mock_peer.id
} };
assert_eq!(blob.get_id().unwrap(), id);
} }
} }
} }