satoridb 0.1.2

Embedded vector database for approximate nearest neighbor search (experimental).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

#[cfg(test)]
mod tests {
    use super::RebalanceState;
    use crate::router::RoutingTable;
    use crate::storage::Storage;
    use crate::wal::runtime::Walrus;
    use std::sync::Arc;
    use tempfile::TempDir;

    fn init_wal(tempdir: &TempDir) -> Arc<Walrus> {
        std::env::set_var("WALRUS_DATA_DIR", tempdir.path());
        Arc::new(Walrus::new().expect("walrus init"))
    }

    #[test]
    fn load_bucket_skips_corrupted_chunk() {
        let tmp = tempfile::tempdir().unwrap();
        let wal = init_wal(&tmp);
        let storage = Storage::new(wal.clone());
        let routing = Arc::new(RoutingTable::new());
        let state = RebalanceState::new(storage.clone(), routing.clone());

        let topic = crate::storage::Storage::topic_for(123);
        wal.append_batch(&topic, &[vec![1u8, 2, 3]]).unwrap();

        let loaded = state.load_bucket(123);
        assert!(loaded.is_none(), "corrupted chunk should not decode into a bucket");
    }
}