snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-23 02:27:40 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Add two tests for the UpdateStore

Browse Source
This commit is contained in:
Clément Renault 2020-10-18 18:22:32 +02:00
parent edb8c99fbe
commit 5b4eda670b
No known key found for this signature in database
GPG Key ID: 92ADA4E935E71FA4
1 changed files with 63 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
src/update_store.rs
View File

@ -1,7 +1,7 @@
use std::path::Path;
use std::sync::Arc;
use crossbeam_channel::{bounded, Sender, Receiver};
use crossbeam_channel::{Sender, Receiver};
use heed::types::{OwnedType, DecodeIgnore, SerdeJson, ByteSlice};
use heed::{EnvOpenOptions, Env, Database};
use once_cell::sync::OnceCell;
@ -20,7 +20,7 @@ pub struct UpdateStore<M> {
impl<M: 'static + Send + Sync> UpdateStore<M> {
pub fn open<P, F>(
options: EnvOpenOptions,
mut options: EnvOpenOptions,
path: P,
mut update_function: F,
) -> heed::Result<Arc<UpdateStore<M>>>
@ -29,12 +29,13 @@ impl<M: 'static + Send + Sync> UpdateStore<M> {
F: FnMut(u64, M, &[u8]) -> heed::Result<M> + Send + 'static,
M: for<'a> Deserialize<'a> + Serialize,
{
options.max_dbs(3);
let env = options.open(path)?;
let pending_meta = env.create_database(Some("pending-meta"))?;
let pending = env.create_database(Some("pending"))?;
let processed_meta = env.create_database(Some("processed-meta"))?;
let (notification_sender, notification_receiver) = bounded(1);
let (notification_sender, notification_receiver) = crossbeam_channel::bounded(1);
let update_store = Arc::new(UpdateStore {
env,
pending,
@ -188,3 +189,62 @@ impl<M: 'static + Send + Sync> UpdateStore<M> {
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::thread;
use std::time::{Duration, Instant};
#[test]
fn simple() {
let dir = tempfile::tempdir().unwrap();
let options = EnvOpenOptions::new();
let update_store = UpdateStore::open(options, dir, |id, meta: String, content| {
Ok(meta + " processed")
}).unwrap();
let meta = String::from("kiki");
let update_id = update_store.register_update(&meta, &[]).unwrap();
thread::sleep(Duration::from_millis(100));
let meta = update_store.update_meta(update_id).unwrap().unwrap();
assert_eq!(meta, "kiki processed");
}
#[test]
fn long_running_update() {
let dir = tempfile::tempdir().unwrap();
let options = EnvOpenOptions::new();
let update_store = UpdateStore::open(options, dir, |id, meta: String, content| {
thread::sleep(Duration::from_millis(400));
Ok(meta + " processed")
}).unwrap();
let before_register = Instant::now();
let meta = String::from("kiki");
let update_id_kiki = update_store.register_update(&meta, &[]).unwrap();
assert!(before_register.elapsed() < Duration::from_millis(200));
let meta = String::from("coco");
let update_id_coco = update_store.register_update(&meta, &[]).unwrap();
assert!(before_register.elapsed() < Duration::from_millis(200));
let meta = String::from("cucu");
let update_id_cucu = update_store.register_update(&meta, &[]).unwrap();
assert!(before_register.elapsed() < Duration::from_millis(200));
thread::sleep(Duration::from_millis(400 * 3 + 100));
let meta = update_store.update_meta(update_id_kiki).unwrap().unwrap();
assert_eq!(meta, "kiki processed");
let meta = update_store.update_meta(update_id_coco).unwrap().unwrap();
assert_eq!(meta, "coco processed");
let meta = update_store.update_meta(update_id_cucu).unwrap().unwrap();
assert_eq!(meta, "cucu processed");
}
}