use crate::common::test_key::TestKey; use crate::common::test_value::TestValue; use dharmadb::options::DharmaOpts; use std::fs::{create_dir, remove_dir_all, remove_file}; pub mod test_key; pub mod test_value; pub fn get_test_data(count: u32) -> Vec<(TestKey, TestValue)> { let mut vector = Vec::new(); for i in 0..count { let key = TestKey::from(i); let value = TestValue::from(format!("value is {}", i).as_str()); vector.push((key, value)); } vector } pub fn get_test_data_in_range(start: u32, end: u32) -> Vec<(TestKey, TestValue)> { let mut vector = Vec::new(); for i in start..end { let key = TestKey::from(i); let value = TestValue::from(format!("value is {}", i).as_str()); vector.push((key, value)); } vector } /// Clean any leftover log files from previous test executions. /// /// # Arguments /// - _options_ - The database config. pub fn cleanup_paths(options: &DharmaOpts) { let sstable_dir = format!("{0}/tables", options.path); let wal_path = format!("{0}/wal.log", options.path); let compaction_path = format!("{}/compaction", options.path); remove_dir_all(&sstable_dir); remove_dir_all(&compaction_path); create_dir(&sstable_dir); remove_file(&wal_path); }