mirror of https://github.com/terhechte/postsack
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
3.2 KiB
Rust
104 lines
3.2 KiB
Rust
use ps_core::{
|
|
self,
|
|
model::{self, Engine, Rect},
|
|
Config, DatabaseLike, Field, Filter, FormatType, Importerlike, ValueField,
|
|
};
|
|
use ps_database::Database;
|
|
use ps_importer::mbox_importer;
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use std::sync::Once;
|
|
|
|
static INIT: Once = Once::new();
|
|
|
|
pub fn initialize() {
|
|
INIT.call_once(|| {
|
|
if std::env::var("RUST_LOG").is_err() {
|
|
std::env::set_var("RUST_LOG", "trace");
|
|
}
|
|
ps_core::setup_tracing();
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn test_engine_all() {
|
|
initialize();
|
|
let config = create_database();
|
|
let mut engine = Engine::new::<Database>(&config).expect("Expected Engine");
|
|
engine.start().expect("Expect to start engine");
|
|
engine.wait().expect("Expected working wait");
|
|
let segment = {
|
|
let segments =
|
|
model::segmentations::layouted_segments(&mut engine, default_rect()).unwrap();
|
|
assert_eq!(segments.len(), 5);
|
|
segments[0].clone()
|
|
};
|
|
|
|
// add another aggregation level
|
|
engine
|
|
.push(segment)
|
|
.expect("Expect being able to push another segment");
|
|
engine.wait().expect("Expected working wait");
|
|
|
|
// Validate
|
|
let segments =
|
|
model::segmentations::layouted_segments(&mut engine, default_rect()).unwrap();
|
|
assert_eq!(segments.len(), 2);
|
|
|
|
// Limit to only one segment
|
|
model::segmentations::set_segments_range(&mut engine, Some(0..=1));
|
|
// Validate
|
|
let segments =
|
|
model::segmentations::layouted_segments(&mut engine, default_rect()).unwrap();
|
|
assert_eq!(segments.len(), 1);
|
|
|
|
// Add a filter
|
|
let filter = Filter::Is(ValueField::bool(&Field::IsSend, true));
|
|
model::segmentations::set_filters(&mut engine, &[filter]).expect("Expect setting filters");
|
|
engine.wait().expect("");
|
|
|
|
let segments =
|
|
model::segmentations::layouted_segments(&mut engine, default_rect()).unwrap();
|
|
assert_eq!(segments.len(), 0);
|
|
|
|
// Remove filter
|
|
model::segmentations::set_filters(&mut engine, &[]).expect("");
|
|
engine.wait().expect("");
|
|
|
|
let segments =
|
|
model::segmentations::layouted_segments(&mut engine, default_rect()).unwrap();
|
|
assert_eq!(segments.len(), 2);
|
|
|
|
// Push a new segment
|
|
let segment = segments[0].clone();
|
|
engine.push(segment).unwrap();
|
|
engine.wait().expect("");
|
|
|
|
// Check the new segments
|
|
let segments =
|
|
model::segmentations::layouted_segments(&mut engine, default_rect()).unwrap();
|
|
assert_eq!(segments.len(), 1);
|
|
}
|
|
}
|
|
|
|
fn default_rect() -> Rect {
|
|
Rect {
|
|
left: 50.0,
|
|
top: 50.0,
|
|
width: 500.0,
|
|
height: 500.0,
|
|
}
|
|
}
|
|
|
|
fn create_database() -> Config {
|
|
let path = "tests/resources/mbox";
|
|
let config = Config::new(None, path, vec!["".to_string()], FormatType::Mbox).expect("Config");
|
|
let importer = mbox_importer(config.clone());
|
|
let database = Database::new(&config.database_path).unwrap();
|
|
let (_receiver, handle) = importer.import(database).unwrap();
|
|
handle.join().expect("").expect("");
|
|
config
|
|
}
|