mirror of
https://github.com/Start9Labs/patch-db.git
synced 2026-03-26 02:11:54 +00:00
add a proptest
This commit is contained in:
@@ -1,9 +1,10 @@
|
||||
use super::*;
|
||||
use proptest::prelude::*;
|
||||
use tokio::runtime::Builder;
|
||||
use tokio::{fs, runtime::Builder};
|
||||
|
||||
async fn init_test_env() -> PatchDb {
|
||||
let db = PatchDb::open("test.db").await.unwrap();
|
||||
async fn init_db(db_name: String) -> PatchDb {
|
||||
cleanup_db(&db_name).await;
|
||||
let db = PatchDb::open(db_name).await.unwrap();
|
||||
db.put(
|
||||
&JsonPointer::<&'static str>::default(),
|
||||
&Sample {
|
||||
@@ -18,6 +19,10 @@ async fn init_test_env() -> PatchDb {
|
||||
db
|
||||
}
|
||||
|
||||
async fn cleanup_db(db_name: &String) {
|
||||
fs::remove_file(db_name).await.ok();
|
||||
}
|
||||
|
||||
async fn put_string_into_root(db: PatchDb, s: String) -> Arc<Revision> {
|
||||
println!("trying string: {}", s);
|
||||
db.put(
|
||||
@@ -30,25 +35,26 @@ async fn put_string_into_root(db: PatchDb, s: String) -> Arc<Revision> {
|
||||
|
||||
#[tokio::test]
|
||||
async fn basic() {
|
||||
let db = init_test_env().await;
|
||||
let db = init_db("basic".to_string()).await;
|
||||
let ptr: JsonPointer = "/b/b".parse().unwrap();
|
||||
let mut get_res: Value = db.get(&ptr).await.unwrap();
|
||||
assert_eq!(get_res, 1);
|
||||
db.put(&ptr, &"hello").await.unwrap();
|
||||
get_res = db.get(&ptr).await.unwrap();
|
||||
assert_eq!(get_res, "hello");
|
||||
cleanup_db(&"basic".to_string()).await;
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn transaction() {
|
||||
let db = init_test_env().await;
|
||||
let db = init_db("transaction".to_string()).await;
|
||||
let mut tx = db.begin();
|
||||
let ptr: JsonPointer = "/b/b".parse().unwrap();
|
||||
tx.put(&ptr, &(2 as usize)).await.unwrap();
|
||||
tx.put(&ptr, &(1 as usize)).await.unwrap();
|
||||
let _res = tx.commit().await.unwrap();
|
||||
println!("res = {:?}", _res);
|
||||
|
||||
cleanup_db(&"transaction".to_string()).await;
|
||||
}
|
||||
|
||||
proptest! {
|
||||
@@ -61,9 +67,11 @@ proptest! {
|
||||
.thread_stack_size(3 * 1024 * 1024)
|
||||
.build()
|
||||
.unwrap();
|
||||
let db = runtime.block_on(init_test_env());
|
||||
let db = runtime.block_on(init_db("doesnt_crash".to_string()));
|
||||
let put_future = put_string_into_root(db, s);
|
||||
runtime.block_on(put_future);
|
||||
runtime.block_on(cleanup_db(&"doesnt_crash".to_string()));
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user