adc007b7b0
- kei-router — keyword-dispatch meta-tool (CfC ML fallback removed) - kei-sage — Obsidian-style knowledge graph, FTS5 + BFS + PageRank - kei-task — task DAG with deps, milestones, dependency-chain queries - kei-chat-store — Claude conversation session persistence + FTS search - kei-crossdomain — typed-edge store + BFS cross-domain glue - kei-search-core — 3-wave deep research with microcent budget cap - kei-content-store — asset + prompt + campaign registry - kei-social-store — people + interactions CRM (lite) - kei-curator — edge-decay graph hygiene utility - kei-auth — multi-tenant session tokens (replaces single-bearer) Genesis-scan pre-import pass: skipped pkg/mxl1/*, pkg/inference/*, pkg/trainer/*, pkg/nc01/*, internal/ml/* (all Genesis/CfC adjacent, sensitive IP). Security: skipped tools_threat/radio/protocol/med/mlreg (offensive/banned). Domain verticals skipped: hr/legal/infra/ops/api/osint/edu/geo/hw/finance. New 'mcp' profile in MANIFEST.toml bundles all 10 for MCP server deployment. Workspace now 24 crates, cargo check --workspace clean, 94 workspace tests pass.
110 lines
3.3 KiB
Rust
110 lines
3.3 KiB
Rust
//! kei-sage integration tests.
|
|
|
|
use kei_sage::bfs::bfs;
|
|
use kei_sage::edges::{add_edge, list_outgoing};
|
|
use kei_sage::import::import_vault;
|
|
use kei_sage::pagerank::pagerank;
|
|
use kei_sage::search::fts_search;
|
|
use kei_sage::{Store, Unit};
|
|
use std::fs;
|
|
use tempfile::tempdir;
|
|
|
|
fn mkstore() -> Store { Store::open_memory().unwrap() }
|
|
|
|
fn mkunit(title: &str, body: &str, vault: &str) -> Unit {
|
|
Unit {
|
|
unit_type: "note".into(), title: title.into(), content: body.into(),
|
|
evidence_grade: "E2".into(), vault_path: vault.into(),
|
|
..Default::default()
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn crud_roundtrip() {
|
|
let s = mkstore();
|
|
let id = s.add_unit(&mkunit("hello", "world", "a.md")).unwrap();
|
|
assert!(id > 0);
|
|
let u = s.get_unit(id).unwrap().unwrap();
|
|
assert_eq!(u.title, "hello");
|
|
s.delete_unit(id).unwrap();
|
|
assert!(s.get_unit(id).unwrap().is_none());
|
|
}
|
|
|
|
#[test]
|
|
fn fts_search_matches() {
|
|
let s = mkstore();
|
|
s.add_unit(&mkunit("rust async", "tokio runtime details", "a.md")).unwrap();
|
|
s.add_unit(&mkunit("python sync", "flask wsgi server", "b.md")).unwrap();
|
|
let hits = fts_search(&s, "tokio", 10).unwrap();
|
|
assert_eq!(hits.len(), 1);
|
|
assert_eq!(hits[0].title, "rust async");
|
|
}
|
|
|
|
#[test]
|
|
fn bfs_depth_limit() {
|
|
let s = mkstore();
|
|
add_edge(&s, "a", "b", "rel", 1.0).unwrap();
|
|
add_edge(&s, "b", "c", "rel", 1.0).unwrap();
|
|
add_edge(&s, "c", "d", "rel", 1.0).unwrap();
|
|
let out = bfs(&s, "a", 2).unwrap();
|
|
let paths: Vec<&str> = out.iter().map(|r| r.path.as_str()).collect();
|
|
assert!(paths.contains(&"b"));
|
|
assert!(paths.contains(&"c"));
|
|
assert!(!paths.contains(&"d"));
|
|
}
|
|
|
|
#[test]
|
|
fn pagerank_orders_by_popularity() {
|
|
let s = mkstore();
|
|
add_edge(&s, "a", "hub", "rel", 1.0).unwrap();
|
|
add_edge(&s, "b", "hub", "rel", 1.0).unwrap();
|
|
add_edge(&s, "c", "hub", "rel", 1.0).unwrap();
|
|
add_edge(&s, "d", "hub", "rel", 1.0).unwrap();
|
|
add_edge(&s, "e", "hub", "rel", 1.0).unwrap();
|
|
let ranks = pagerank(&s).unwrap();
|
|
assert_eq!(ranks[0].0, "hub");
|
|
}
|
|
|
|
#[test]
|
|
fn edges_crud() {
|
|
let s = mkstore();
|
|
let id = add_edge(&s, "x", "y", "cites", 0.8).unwrap();
|
|
assert!(id > 0);
|
|
let out = list_outgoing(&s, "x").unwrap();
|
|
assert_eq!(out.len(), 1);
|
|
assert_eq!(out[0].dst_path, "y");
|
|
}
|
|
|
|
#[test]
|
|
fn import_idempotency() {
|
|
let tmp = tempdir().unwrap();
|
|
let p = tmp.path().join("one.md");
|
|
fs::write(&p, "# title one\nhello").unwrap();
|
|
let s = mkstore();
|
|
let first = import_vault(&s, tmp.path()).unwrap();
|
|
let second = import_vault(&s, tmp.path()).unwrap();
|
|
assert_eq!(first.imported, 1);
|
|
assert_eq!(second.imported, 1);
|
|
assert_eq!(s.count_units().unwrap(), 1);
|
|
}
|
|
|
|
#[test]
|
|
fn edges_cross_reference_validates() {
|
|
let s = mkstore();
|
|
s.add_unit(&mkunit("note a", "", "a.md")).unwrap();
|
|
s.add_unit(&mkunit("note b", "", "b.md")).unwrap();
|
|
add_edge(&s, "a.md", "b.md", "refs", 1.0).unwrap();
|
|
let out = list_outgoing(&s, "a.md").unwrap();
|
|
assert_eq!(out.len(), 1);
|
|
}
|
|
|
|
#[test]
|
|
fn fts5_respects_limit() {
|
|
let s = mkstore();
|
|
for i in 0..25 {
|
|
let t = format!("rust note {i}");
|
|
s.add_unit(&mkunit(&t, "rust rust rust", &format!("n{i}.md"))).unwrap();
|
|
}
|
|
let hits = fts_search(&s, "rust", 5).unwrap();
|
|
assert_eq!(hits.len(), 5);
|
|
}
|