Add tests for the new methods

Author: thomaskrause

core/src/graph/mod.rs

@@ -1125,166 +1125,4 @@ impl<CT: ComponentType> Graph<CT> {
 }
 
 #[cfg(test)]
-mod tests {
-
-    use super::*;
-    use crate::types::{AnnoKey, Annotation, DefaultComponentType, Edge};
-
-    #[test]
-    fn create_writeable_gs() {
-        let mut db = Graph::<DefaultComponentType>::new(false).unwrap();
-
-        let anno_key = AnnoKey {
-            ns: "test".into(),
-            name: "edge_anno".into(),
-        };
-        let anno_val = "testValue".into();
-
-        let component = Component::new(DefaultComponentType::Edge, "test".into(), "dep".into());
-        let gs: &mut dyn WriteableGraphStorage = db.get_or_create_writable(&component).unwrap();
-
-        gs.add_edge(Edge {
-            source: 0,
-            target: 1,
-        })
-        .unwrap();
-
-        gs.add_edge_annotation(
-            Edge {
-                source: 0,
-                target: 1,
-            },
-            Annotation {
-                key: anno_key,
-                val: anno_val,
-            },
-        )
-        .unwrap();
-    }
-
-    #[test]
-    fn load_existing_graph_storage() {
-        let mut db = Graph::<DefaultComponentType>::new(false).unwrap();
-
-        let component = Component::new(DefaultComponentType::Edge, "test".into(), "dep".into());
-        db.get_or_create_writable(&component).unwrap();
-
-        let tmp = tempfile::tempdir().unwrap();
-
-        db.save_to(tmp.path()).unwrap();
-
-        let mut db = Graph::new(false).unwrap();
-        db.open(tmp.path()).unwrap();
-        assert_eq!(1, db.components.len());
-        let gs = db.components.get(&component);
-        assert_eq!(true, gs.is_some());
-        assert_eq!(false, gs.unwrap().is_some());
-
-        db.ensure_loaded(&component).unwrap();
-        assert_eq!(1, db.components.len());
-        let gs = db.components.get(&component);
-        assert_eq!(true, gs.is_some());
-        assert_eq!(true, gs.unwrap().is_some());
-    }
-
-    #[test]
-    fn load_existing_graph_storage_parallel() {
-        let mut db = Graph::<DefaultComponentType>::new(false).unwrap();
-
-        let component = Component::new(DefaultComponentType::Edge, "test".into(), "dep".into());
-        db.get_or_create_writable(&component).unwrap();
-
-        let tmp = tempfile::tempdir().unwrap();
-
-        db.save_to(tmp.path()).unwrap();
-
-        let mut db = Graph::new(false).unwrap();
-        db.open(tmp.path()).unwrap();
-        assert_eq!(1, db.components.len());
-        let gs = db.components.get(&component);
-        assert_eq!(true, gs.is_some());
-        assert_eq!(false, gs.unwrap().is_some());
-
-        db.ensure_loaded_parallel(&[component.clone()]).unwrap();
-        assert_eq!(1, db.components.len());
-        let gs = db.components.get(&component);
-        assert_eq!(true, gs.is_some());
-        assert_eq!(true, gs.unwrap().is_some());
-    }
-
-    #[test]
-    fn load_non_existing_graph_storage_parallel() {
-        let mut db = Graph::<DefaultComponentType>::new(false).unwrap();
-
-        let component = Component::new(DefaultComponentType::Edge, "test".into(), "dep".into());
-
-        let tmp = tempfile::tempdir().unwrap();
-
-        db.save_to(tmp.path()).unwrap();
-
-        let mut db = Graph::new(false).unwrap();
-        db.open(tmp.path()).unwrap();
-
-        db.ensure_loaded_parallel(&[component]).unwrap();
-        assert_eq!(0, db.components.len());
-    }
-
-    #[test]
-    fn load_with_wal_file() {
-        let mut db = Graph::<DefaultComponentType>::new(false).unwrap();
-        let example_node = 0;
-        db.node_annos
-            .insert(
-                example_node,
-                Annotation {
-                    key: NODE_TYPE_KEY.as_ref().clone(),
-                    val: "corpus".into(),
-                },
-            )
-            .unwrap();
-        db.node_annos
-            .insert(
-                example_node,
-                Annotation {
-                    key: NODE_NAME_KEY.as_ref().clone(),
-                    val: "root".into(),
-                },
-            )
-            .unwrap();
-
-        let tmp = tempfile::tempdir().unwrap();
-        // Save and remember the location, so that updates are recorded in a WAL
-        // file
-        db.persist_to(tmp.path()).unwrap();
-
-        // Add an node annotation with apply_update
-        let mut u = GraphUpdate::new();
-        u.add_event(UpdateEvent::AddNodeLabel {
-            node_name: "root".into(),
-            anno_ns: "example".into(),
-            anno_name: "anno-name".into(),
-            anno_value: "anno-value".into(),
-        })
-        .unwrap();
-        db.apply_update(&mut u, |_| {}).unwrap();
-
-        std::mem::drop(db);
-
-        // Check that loading the database again contains the changes
-        let mut db = Graph::<DefaultComponentType>::new(false).unwrap();
-        db.open(tmp.path()).unwrap();
-        db.ensure_loaded_all().unwrap();
-        let anno_value = db
-            .node_annos
-            .get_value_for_item(
-                &example_node,
-                &AnnoKey {
-                    name: "anno-name".into(),
-                    ns: "example".into(),
-                },
-            )
-            .unwrap()
-            .unwrap();
-        assert_eq!("anno-value", anno_value);
-    }
-}
+mod tests;