Function test_kg_edge_crud

atomic-core/src/pristine/txn/write/triples.rs:502–542

()

    #[test]
    fn test_kg_edge_crud() {
        let dir = tempdir().unwrap();
        let pristine = Pristine::open(dir.path().join("db")).unwrap();
        let mut txn = pristine.write_txn().unwrap();
        txn.init_kg().unwrap();

        // Create nodes
        txn.upsert_kg_node(&KgNode::new("change:abc", "change", "abc", "graph"))
            .unwrap();
        txn.upsert_kg_node(&KgNode::new("file:auth.rs", "file", "auth.rs", "graph"))
            .unwrap();
        txn.upsert_kg_node(&KgNode::new("file:main.rs", "file", "main.rs", "graph"))
            .unwrap();

        // Create edges
        txn.upsert_kg_edge(&KgEdge::new("change:abc", "file:auth.rs", "MODIFIES"))
            .unwrap();
        txn.upsert_kg_edge(&KgEdge::new("change:abc", "file:main.rs", "MODIFIES"))
            .unwrap();

        // Query outgoing
        let outgoing = txn.get_kg_edges_from("change:abc").unwrap();
        assert_eq!(outgoing.len(), 2);

        // Query incoming
        let incoming = txn.get_kg_edges_to("file:auth.rs").unwrap();
        assert_eq!(incoming.len(), 1);
        assert_eq!(incoming[0].from_id, "change:abc");

        // Delete edge
        assert!(txn
            .del_kg_edge("change:abc", "file:auth.rs", "MODIFIES")
            .unwrap());
        assert_eq!(txn.get_kg_edges_from("change:abc").unwrap().len(), 1);

        // Counts
        assert_eq!(txn.count_kg_nodes().unwrap(), 3);
        assert_eq!(txn.count_kg_edges().unwrap(), 1);

        txn.commit().unwrap();
    }

    #[test]
    fn test_kg_neighbors() {