0be354a920
Single-commit clean baseline after security scrub of niche-tells, project codenames, internal jargon, and contributor-email leaks. Contents: - 100 Rust crates (_primitives/_rust/) - 37 agent manifests (_manifests/) + generated specs (_generated/) - 67 user-invocable skills (skills/) - 33 hooks (hooks/) - Composition blocks (_blocks/) - Documentation (docs/, README.md) - TS adapter packages (_ts_packages/) - Assembler (_assembler/) - Roles (_roles/) - Templates (_templates/) - Forgejo CI (.forgejo/) Author: Denis Parfionovich <info@greendragon.info> License: see LICENSE.
100 lines
3.5 KiB
Rust
100 lines
3.5 KiB
Rust
use kei_chat_store::search::search;
|
|
use kei_chat_store::sessions::{archive_session, get_session, save_message, start_session, ChatMessage};
|
|
use kei_chat_store::stats::stats;
|
|
use kei_chat_store::Store;
|
|
|
|
fn mk() -> Store { Store::open_memory().unwrap() }
|
|
|
|
#[test]
|
|
fn save_and_retrieve() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "demo", "t", "claude-opus-4").unwrap();
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid.clone(), role: "user".into(),
|
|
content: "hello world".into(), tokens_in: 3, tokens_out: 0, cost: 0.001,
|
|
..Default::default()
|
|
}).unwrap();
|
|
let sess = get_session(&s, &sid).unwrap().unwrap();
|
|
assert_eq!(sess.message_count, 1);
|
|
assert_eq!(sess.total_tokens, 3);
|
|
}
|
|
|
|
#[test]
|
|
fn fts_search_finds_message() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "demo", "", "").unwrap();
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid, role: "user".into(),
|
|
content: "rust async tokio bench".into(),
|
|
..Default::default()
|
|
}).unwrap();
|
|
let hits = search(&s, "tokio", 10).unwrap();
|
|
assert_eq!(hits.len(), 1);
|
|
}
|
|
|
|
#[test]
|
|
fn archive_session_works() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "p", "", "").unwrap();
|
|
archive_session(&s, &sid).unwrap();
|
|
let sess = get_session(&s, &sid).unwrap().unwrap();
|
|
assert_eq!(sess.status, "archived");
|
|
}
|
|
|
|
#[test]
|
|
fn engine_migration_parity_smoke() {
|
|
// Layer-A convergence parity: fresh session opens cleanly through
|
|
// kei_entity_store::Store + CHAT_SCHEMA, start_session returns a
|
|
// UUID that get_session can retrieve with id > 0 chars and the
|
|
// engine-generated chat_messages table is writeable.
|
|
let s = mk();
|
|
let sid = start_session(&s, "conv-layer-a", "smoke", "claude-opus-4").unwrap();
|
|
assert!(!sid.is_empty(), "session id should be a non-empty UUID");
|
|
let mid = save_message(&s, &ChatMessage {
|
|
session_id: sid.clone(), role: "user".into(),
|
|
content: "migration-parity probe".into(),
|
|
tokens_in: 1, tokens_out: 0, cost: 0.0,
|
|
..Default::default()
|
|
}).unwrap();
|
|
assert!(mid > 0, "engine-backed message id must be > 0");
|
|
let sess = get_session(&s, &sid).unwrap().unwrap();
|
|
assert_eq!(sess.id, sid);
|
|
assert_eq!(sess.message_count, 1);
|
|
}
|
|
|
|
#[test]
|
|
fn cost_roundtrips_via_search() {
|
|
// Wave-8 re-migration: cost is re-instated as engine-managed
|
|
// RealDefault column. The value written via save_message must be
|
|
// visible on the ChatMessage returned from search (no longer 0.0).
|
|
let s = mk();
|
|
let sid = start_session(&s, "demo", "", "").unwrap();
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid, role: "user".into(),
|
|
content: "rust async tokio bench cost-marker".into(),
|
|
tokens_in: 1, tokens_out: 1, cost: 0.00777,
|
|
..Default::default()
|
|
}).unwrap();
|
|
let hits = search(&s, "cost-marker", 10).unwrap();
|
|
assert_eq!(hits.len(), 1);
|
|
assert!((hits[0].cost - 0.00777).abs() < 1e-9,
|
|
"cost should round-trip via engine RealDefault column, got {}",
|
|
hits[0].cost);
|
|
}
|
|
|
|
#[test]
|
|
fn stats_aggregates() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "p", "", "").unwrap();
|
|
for _ in 0..3 {
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid.clone(), role: "user".into(),
|
|
content: "x".into(), tokens_in: 5, tokens_out: 5, cost: 0.01,
|
|
..Default::default()
|
|
}).unwrap();
|
|
}
|
|
let st = stats(&s).unwrap();
|
|
assert_eq!(st.total_sessions, 1);
|
|
assert_eq!(st.total_messages, 3);
|
|
assert_eq!(st.total_tokens, 30);
|
|
}
|