wehub-resource-sync
80 lines
2.2 KiB
Rust
80 lines
2.2 KiB
Rust
use memvid_core::{Memvid, MemvidError};
|
|
use std::path::{Path, PathBuf};
|
|
use tempfile::TempDir;
|
|
|
|
type TestResult = Result<(), Box<dyn std::error::Error>>;
|
|
|
|
fn create_tmp_memvid() -> Result<(TempDir, PathBuf), Box<dyn std::error::Error>> {
|
|
let dir = TempDir::new()?;
|
|
let path = dir.path().join("test.mv2");
|
|
// Create new memory
|
|
{
|
|
let mut mem = Memvid::create(&path)?;
|
|
mem.commit()?;
|
|
}
|
|
Ok((dir, path))
|
|
}
|
|
|
|
fn open_tmp_memvid(path: &Path) -> Result<Memvid, Box<dyn std::error::Error>> {
|
|
Ok(Memvid::open(path)?)
|
|
}
|
|
|
|
#[test]
|
|
fn test_vec_model_consistency() -> TestResult {
|
|
let (_dir, path) = create_tmp_memvid()?;
|
|
|
|
// 1. Create index and set model "A"
|
|
{
|
|
let mut memvid = open_tmp_memvid(&path)?;
|
|
memvid.enable_vec()?;
|
|
memvid.set_vec_model("model-a")?;
|
|
memvid.commit()?;
|
|
}
|
|
|
|
// 2. Open and verify "model-a" matches and "model-b" fails
|
|
{
|
|
let mut memvid = open_tmp_memvid(&path)?;
|
|
// Should succeed (idempotent)
|
|
memvid.set_vec_model("model-a")?;
|
|
|
|
// Should fail
|
|
let result = memvid.set_vec_model("model-b");
|
|
assert!(result.is_err());
|
|
match result {
|
|
Err(MemvidError::ModelMismatch { expected, actual }) => {
|
|
assert_eq!(expected, "model-a");
|
|
assert_eq!(actual, "model-b");
|
|
}
|
|
_ => panic!("Expected ModelMismatch error"),
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_vec_model_persistence() -> TestResult {
|
|
let (_dir, path) = create_tmp_memvid()?;
|
|
|
|
// 1. Create index with model
|
|
{
|
|
let mut memvid = open_tmp_memvid(&path)?;
|
|
memvid.enable_vec()?;
|
|
memvid.set_vec_model("persistent-model")?;
|
|
memvid.commit()?;
|
|
}
|
|
|
|
// 2. Open and check if model is loaded automatically
|
|
{
|
|
let mut memvid = open_tmp_memvid(&path)?;
|
|
// We can inspect internal state via debug or by trying to set a mismatch
|
|
let result = memvid.set_vec_model("wrong-model");
|
|
assert!(result.is_err());
|
|
|
|
// Verify set_vec_model("persistent-model") works (confirming loaded state)
|
|
memvid.set_vec_model("persistent-model")?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|