mirror of
https://github.com/aljazceru/turso.git
synced 2026-01-03 00:14:21 +01:00
be4014a1df
This adds a "limbo_stress" tool for stress testing Limbo in non-deterministic way together with support code to run the tests under Antithesis (which makes them deterministic). The stress tester does not really do anything useful yet, this is just a step to make sure we can run tests under Antithesis.
45 lines
1.2 KiB
Rust
45 lines
1.2 KiB
Rust
mod opts;
|
|
|
|
use antithesis_sdk::*;
|
|
use clap::Parser;
|
|
use limbo::{Builder, Value};
|
|
use opts::Opts;
|
|
use serde_json::json;
|
|
use std::sync::Arc;
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let (num_nodes, main_id) = (1, "n-001");
|
|
let startup_data = json!({
|
|
"num_nodes": num_nodes,
|
|
"main_node_id": main_id,
|
|
});
|
|
lifecycle::setup_complete(&startup_data);
|
|
antithesis_init();
|
|
|
|
let opts = Opts::parse();
|
|
let mut handles = Vec::new();
|
|
|
|
for _ in 0..opts.nr_threads {
|
|
// TODO: share the database between threads
|
|
let db = Arc::new(Builder::new_local(":memory:").build().await.unwrap());
|
|
let nr_iterations = opts.nr_iterations;
|
|
let db = db.clone();
|
|
let handle = tokio::spawn(async move {
|
|
let conn = db.connect().unwrap();
|
|
|
|
for _ in 0..nr_iterations {
|
|
let mut rows = conn.query("select 1", ()).await.unwrap();
|
|
let row = rows.next().await.unwrap().unwrap();
|
|
let value = row.get_value(0).unwrap();
|
|
assert_always!(matches!(value, Value::Integer(1)), "value is incorrect");
|
|
}
|
|
});
|
|
handles.push(handle);
|
|
}
|
|
for handle in handles {
|
|
handle.await.unwrap();
|
|
}
|
|
println!("Done.");
|
|
}
|