github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

remove tests from darkfid

Browse Source
This commit is contained in:
ghassmo
2021-08-02 01:52:14 +03:00
parent 9f35f9c14e
commit e3a338100b
1 changed files with 0 additions and 302 deletions
Download Patch File Download Diff File Expand all files Collapse all files

302
src/bin/darkfid.rs
View File

@@ -273,305 +273,3 @@ fn main() -> Result<()> {
result
}
//// $ cargo test test_ten_clients_simultaneously --bin darkfid
//this will run 10 clients simultaneously
//// $ cargo test test_subscriber --bin darkfid
// Run Client A and send 10 slabs
// Client B should receive 10 slabs from subscriber
//// $ cargo test test_deposit --bin darkfid
// Run Client A and send 10 slabs
// Client B should receive 10 slabs from subscriber
#[cfg(test)]
mod test {
use std::net::SocketAddr;
use std::path::Path;
use std::path::PathBuf;
use std::sync::Arc;
use drk::blockchain::{rocks::columns, Rocks, RocksColumn, Slab};
use drk::service::{GatewayClient, GatewaySlabsSubscriber};
use drk::util::join_config_path;
use async_executor::Executor;
use easy_parallel::Parallel;
use log::*;
use rand::Rng;
use simplelog::*;
pub async fn subscribe(gateway_slabs_sub: GatewaySlabsSubscriber, id: String) {
loop {
gateway_slabs_sub.recv().await.unwrap();
info!("Client {}: update state", id);
}
}
fn setup_log() {
let logger_config = ConfigBuilder::new().set_time_format_str("%T%.6f").build();
CombinedLogger::init(vec![
TermLogger::new(LevelFilter::Debug, logger_config, TerminalMode::Mixed).unwrap(),
WriteLogger::new(
LevelFilter::Debug,
Config::default(),
std::fs::File::create(Path::new("/tmp/dar.log")).unwrap(),
),
])
.unwrap();
}
#[test]
fn test_ten_clients_simultaneously() {
setup_log();
let mut thread_pools: Vec<std::thread::JoinHandle<()>> = vec![];
for _ in 0..10 {
let thread = std::thread::spawn(|| {
let ex = Arc::new(Executor::new());
let (signal, shutdown) = async_channel::unbounded::<()>();
let ex2 = ex.clone();
let (_, _) = Parallel::new()
// Run four executor threads.
.each(0..3, |_| smol::future::block_on(ex2.run(shutdown.recv())))
// Run the main future on the current thread.
.finish(|| {
smol::future::block_on(async move {
let connect_addr: SocketAddr = "127.0.0.1:3333".parse().unwrap();
let sub_addr: SocketAddr = "127.0.0.1:4444".parse().unwrap();
let mut rng = rand::thread_rng();
let rnd: u32 = rng.gen();
let path_str = format!("database_{}.db", rnd);
let database_path = PathBuf::from(path_str.as_str());
let database_path = join_config_path(&database_path).unwrap();
let rocks = Rocks::new(&database_path).unwrap();
let slabstore = RocksColumn::<columns::Slabs>::new(rocks.clone());
// create gateway client
let mut client = GatewayClient::new(connect_addr, slabstore).unwrap();
// start subscribing
let gateway_slabs_sub: GatewaySlabsSubscriber =
client.start_subscriber(sub_addr, ex.clone()).await.unwrap();
ex.clone()
.spawn(subscribe(gateway_slabs_sub, rnd.clone().to_string()))
.detach();
// start gateway client
client.start().await.unwrap();
let slab = Slab::new(rnd.to_le_bytes().to_vec());
client.put_slab(slab).await.unwrap();
});
drop(signal);
Ok::<(), drk::Error>(())
});
});
thread_pools.push(thread);
}
for t in thread_pools {
t.join().unwrap();
}
}
#[test]
fn test_subscriber() {
setup_log();
let mut thread_pools: Vec<std::thread::JoinHandle<()>> = vec![];
// Client A
let thread = std::thread::spawn(|| {
smol::future::block_on(async move {
let connect_addr: SocketAddr = "127.0.0.1:3333".parse().unwrap();
let mut rng = rand::thread_rng();
let rnd: u32 = rng.gen();
let path_str = format!("database_{}.db", rnd);
let database_path = PathBuf::from(path_str.as_str());
let database_path = join_config_path(&database_path).unwrap();
let rocks = Rocks::new(&database_path).unwrap();
let slabstore = RocksColumn::<columns::Slabs>::new(rocks.clone());
// create gateway client
let mut client = GatewayClient::new(connect_addr, slabstore).unwrap();
// start gateway client
client.start().await.unwrap();
let slab = Slab::new(rnd.to_le_bytes().to_vec());
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
});
});
// Client B
let thread2 = std::thread::spawn(|| {
let ex = Arc::new(Executor::new());
let (signal, shutdown) = async_channel::unbounded::<()>();
let ex2 = ex.clone();
let (_, _) = Parallel::new()
// Run four executor threads.
.each(0..3, |_| smol::future::block_on(ex2.run(shutdown.recv())))
// Run the main future on the current thread.
.finish(|| {
smol::future::block_on(async move {
let connect_addr: SocketAddr = "127.0.0.1:3333".parse().unwrap();
let sub_addr: SocketAddr = "127.0.0.1:4444".parse().unwrap();
let mut rng = rand::thread_rng();
let rnd: u32 = rng.gen();
let path_str = format!("database_{}.db", rnd);
let database_path = PathBuf::from(path_str.as_str());
let database_path = join_config_path(&database_path).unwrap();
let rocks = Rocks::new(&database_path).unwrap();
let slabstore = RocksColumn::<columns::Slabs>::new(rocks.clone());
// create gateway client
let mut client = GatewayClient::new(connect_addr, slabstore).unwrap();
// start subscribing
let gateway_slabs_sub: GatewaySlabsSubscriber =
client.start_subscriber(sub_addr, ex.clone()).await.unwrap();
ex.clone()
.spawn(subscribe(gateway_slabs_sub, "B".to_string()))
.detach();
// start gateway client
client.start().await.unwrap();
// sleep for 2 seconds
std::thread::sleep(std::time::Duration::from_secs(2));
});
drop(signal);
Ok::<(), drk::Error>(())
});
});
thread_pools.push(thread);
thread_pools.push(thread2);
for t in thread_pools {
t.join().unwrap();
}
}
#[test]
fn test_deposit() {
setup_log();
let mut thread_pools: Vec<std::thread::JoinHandle<()>> = vec![];
// Client A: User
let thread = std::thread::spawn(|| {
smol::future::block_on(async move {
let connect_addr: SocketAddr = "127.0.0.1:3333".parse().unwrap();
let mut rng = rand::thread_rng();
let rnd: u32 = rng.gen();
let path_str = format!("database_{}.db", rnd);
let database_path = PathBuf::from(path_str.as_str());
let database_path = join_config_path(&database_path).unwrap();
let rocks = Rocks::new(&database_path).unwrap();
let slabstore = RocksColumn::<columns::Slabs>::new(rocks.clone());
// create gateway client
let mut client = GatewayClient::new(connect_addr, slabstore).unwrap();
// start gateway client
client.start().await.unwrap();
let slab = Slab::new(rnd.to_le_bytes().to_vec());
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
client.put_slab(slab.clone()).await.unwrap();
});
});
// Client B: Cashier
let thread2 = std::thread::spawn(|| {
let ex = Arc::new(Executor::new());
let (signal, shutdown) = async_channel::unbounded::<()>();
let ex2 = ex.clone();
let (_, _) = Parallel::new()
// Run four executor threads.
.each(0..3, |_| smol::future::block_on(ex2.run(shutdown.recv())))
// Run the main future on the current thread.
.finish(|| {
smol::future::block_on(async move {
let connect_addr: SocketAddr = "127.0.0.1:3333".parse().unwrap();
let sub_addr: SocketAddr = "127.0.0.1:4444".parse().unwrap();
let mut rng = rand::thread_rng();
let rnd: u32 = rng.gen();
let path_str = format!("database_{}.db", rnd);
let database_path = PathBuf::from(path_str.as_str());
let database_path = join_config_path(&database_path).unwrap();
let rocks = Rocks::new(&database_path).unwrap();
let slabstore = RocksColumn::<columns::Slabs>::new(rocks.clone());
// create gateway client
let mut client = GatewayClient::new(connect_addr, slabstore).unwrap();
// start subscribing
let gateway_slabs_sub: GatewaySlabsSubscriber =
client.start_subscriber(sub_addr, ex.clone()).await.unwrap();
ex.clone()
.spawn(subscribe(gateway_slabs_sub, "B".to_string()))
.detach();
// start gateway client
client.start().await.unwrap();
// sleep for 2 seconds
std::thread::sleep(std::time::Duration::from_secs(2));
});
drop(signal);
Ok::<(), drk::Error>(())
});
});
thread_pools.push(thread);
thread_pools.push(thread2);
for t in thread_pools {
t.join().unwrap();
}
}
}