mirror of
https://github.com/ModernRelay/omnigraph.git
synced 2026-06-09 01:35:18 +02:00
Merge pull request #40 from ModernRelay/refactor/extract-omnigraph-tests
Extract public-API tests from omnigraph.rs to integration tests
This commit is contained in:
Andrew Altshuler
GitHub
commit
789c0633e8
3 changed files with 288 additions and 268 deletions
|
|
@ -1500,9 +1500,7 @@ mod tests {
|
|||
use crate::db::is_internal_run_branch;
|
||||
use crate::db::manifest::ManifestCoordinator;
|
||||
use async_trait::async_trait;
|
||||
use omnigraph_compiler::{SchemaMigrationStep, SchemaTypeKind};
|
||||
use serde_json::Value;
|
||||
use std::fs;
|
||||
use std::sync::Mutex;
|
||||
|
||||
use crate::storage::{LocalStorageAdapter, StorageAdapter, join_uri};
|
||||
|
|
@ -1561,50 +1559,6 @@ edge WorksAt: Person -> Company
|
|||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_init_creates_repo() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
// Schema file written
|
||||
assert!(dir.path().join("_schema.pg").exists());
|
||||
assert!(dir.path().join("_schema.ir.json").exists());
|
||||
assert!(dir.path().join("__schema_state.json").exists());
|
||||
|
||||
// Manifest created with correct entries
|
||||
let snap = db.snapshot();
|
||||
assert!(snap.entry("node:Person").is_some());
|
||||
assert!(snap.entry("node:Company").is_some());
|
||||
assert!(snap.entry("edge:Knows").is_some());
|
||||
assert!(snap.entry("edge:WorksAt").is_some());
|
||||
|
||||
// Catalog is correct
|
||||
assert_eq!(db.catalog().node_types.len(), 2);
|
||||
assert_eq!(db.catalog().edge_types.len(), 2);
|
||||
assert_eq!(
|
||||
db.catalog().node_types["Person"].key_property(),
|
||||
Some("name")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_open_reads_existing_repo() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
|
||||
Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
// Re-open
|
||||
let db = Omnigraph::open(uri).await.unwrap();
|
||||
assert_eq!(db.catalog().node_types.len(), 2);
|
||||
assert_eq!(db.catalog().edge_types.len(), 2);
|
||||
let snap = db.snapshot();
|
||||
assert!(snap.entry("node:Person").is_some());
|
||||
assert!(snap.entry("edge:Knows").is_some());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_init_and_open_route_graph_metadata_through_storage_adapter() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
|
|
@ -1649,151 +1603,6 @@ edge WorksAt: Person -> Company
|
|||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_open_bootstraps_legacy_schema_state_for_main_only_repo() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
fs::remove_file(dir.path().join("_schema.ir.json")).unwrap();
|
||||
fs::remove_file(dir.path().join("__schema_state.json")).unwrap();
|
||||
|
||||
let db = Omnigraph::open(uri).await.unwrap();
|
||||
assert_eq!(db.catalog().node_types.len(), 2);
|
||||
assert!(dir.path().join("_schema.ir.json").exists());
|
||||
assert!(dir.path().join("__schema_state.json").exists());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_open_rejects_legacy_repo_with_public_branch() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
db.branch_create("feature").await.unwrap();
|
||||
|
||||
fs::remove_file(dir.path().join("_schema.ir.json")).unwrap();
|
||||
fs::remove_file(dir.path().join("__schema_state.json")).unwrap();
|
||||
|
||||
let err = match Omnigraph::open(uri).await {
|
||||
Ok(_) => panic!("expected legacy repo with public branch to fail schema bootstrap"),
|
||||
Err(err) => err,
|
||||
};
|
||||
let message = err.to_string();
|
||||
assert!(message.contains("public branches block schema evolution entirely"));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_long_lived_handle_rejects_schema_source_drift() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let drifted = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
fs::write(dir.path().join("_schema.pg"), drifted).unwrap();
|
||||
|
||||
let err = match db.snapshot_of(ReadTarget::branch("main")).await {
|
||||
Ok(_) => panic!("expected schema source drift to be rejected"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("current _schema.pg no longer matches the accepted compiled schema")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_long_lived_handle_rejects_schema_ir_drift() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
fs::write(dir.path().join("_schema.ir.json"), "{not valid json").unwrap();
|
||||
|
||||
let err = match db.snapshot_of(ReadTarget::branch("main")).await {
|
||||
Ok(_) => panic!("expected schema IR drift to be rejected"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("accepted compiled schema contract in _schema.ir.json is invalid")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_long_lived_handle_rejects_ir_and_source_updates_without_state_update() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let drifted = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
let drifted_ir = read_schema_ir_from_source(&drifted).unwrap();
|
||||
let drifted_ir_json = omnigraph_compiler::schema_ir_pretty_json(&drifted_ir).unwrap();
|
||||
fs::write(dir.path().join("_schema.pg"), drifted).unwrap();
|
||||
fs::write(dir.path().join("_schema.ir.json"), drifted_ir_json).unwrap();
|
||||
|
||||
let err = match db.snapshot_of(ReadTarget::branch("main")).await {
|
||||
Ok(_) => panic!("expected schema state mismatch to be rejected"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("accepted compiled schema does not match the recorded schema state")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_comment_only_schema_edit_keeps_schema_state_valid() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let commented = format!("// comment-only drift\n{}", TEST_SCHEMA);
|
||||
fs::write(dir.path().join("_schema.pg"), commented).unwrap();
|
||||
|
||||
let snapshot = db.snapshot_of(ReadTarget::branch("main")).await.unwrap();
|
||||
assert!(snapshot.entry("node:Person").is_some());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_plan_schema_reports_supported_additive_change() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let desired = TEST_SCHEMA.replace(
|
||||
" age: I32?\n}",
|
||||
" age: I32?\n nickname: String?\n}",
|
||||
);
|
||||
|
||||
let plan = db.plan_schema(&desired).await.unwrap();
|
||||
assert!(plan.supported);
|
||||
assert!(plan.steps.iter().any(|step| matches!(
|
||||
step,
|
||||
SchemaMigrationStep::AddProperty {
|
||||
type_kind: SchemaTypeKind::Node,
|
||||
type_name,
|
||||
property_name,
|
||||
..
|
||||
} if type_name == "Person" && property_name == "nickname"
|
||||
)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_plan_schema_rejects_when_schema_contract_has_drifted() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let drifted = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
fs::write(dir.path().join("_schema.pg"), drifted).unwrap();
|
||||
|
||||
let err = db.plan_schema(TEST_SCHEMA).await.unwrap_err();
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("current _schema.pg no longer matches the accepted compiled schema")
|
||||
);
|
||||
}
|
||||
|
||||
async fn table_rows_json(db: &Omnigraph, table_key: &str) -> Vec<Value> {
|
||||
let snapshot = db.snapshot();
|
||||
let ds = snapshot.open(table_key).await.unwrap();
|
||||
|
|
@ -1838,18 +1647,6 @@ edge WorksAt: Person -> Company
|
|||
.unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_apply_schema_noop_returns_not_applied() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let result = db.apply_schema(TEST_SCHEMA).await.unwrap();
|
||||
assert!(result.supported);
|
||||
assert!(!result.applied);
|
||||
assert!(result.steps.is_empty());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_apply_schema_adds_nullable_property_and_preserves_rows() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
|
|
@ -1925,24 +1722,6 @@ edge WorksAt: Person -> Company
|
|||
assert!(historical.entry("node:Human").is_none());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_apply_schema_rejects_when_non_main_branch_exists() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
db.branch_create("feature").await.unwrap();
|
||||
|
||||
let desired = TEST_SCHEMA.replace(
|
||||
" age: I32?\n}",
|
||||
" age: I32?\n nickname: String?\n}",
|
||||
);
|
||||
let err = db.apply_schema(&desired).await.unwrap_err();
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("schema apply requires a repo with only main")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_apply_schema_succeeds_after_load_creates_published_run_branch() {
|
||||
// Regression for MR-670: schema apply used to fail after any load or
|
||||
|
|
@ -2066,51 +1845,4 @@ edge WorksAt: Person -> Company
|
|||
let branches = db.branch_list().await.unwrap();
|
||||
assert_eq!(branches, vec!["main".to_string()]);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_apply_schema_unsupported_plan_does_not_advance_manifest() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
let before_version = db.snapshot().version();
|
||||
|
||||
let desired = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
let err = db.apply_schema(&desired).await.unwrap_err();
|
||||
assert!(err.to_string().contains("changing property type"));
|
||||
assert_eq!(db.snapshot().version(), before_version);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_open_nonexistent_fails() {
|
||||
let result = Omnigraph::open("/tmp/nonexistent_omnigraph_test_xyz").await;
|
||||
assert!(result.is_err());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_snapshot_version_is_pinned() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
// Take snapshot before any writes
|
||||
let snap1 = db.snapshot();
|
||||
let v1 = snap1.version();
|
||||
|
||||
// Load data — advances manifest version
|
||||
crate::loader::load_jsonl(
|
||||
&mut db,
|
||||
r#"{"type": "Person", "data": {"name": "Alice", "age": 30}}"#,
|
||||
crate::loader::LoadMode::Overwrite,
|
||||
)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// Snapshot from handle sees new version
|
||||
let snap2 = db.snapshot();
|
||||
assert!(snap2.version() > v1);
|
||||
|
||||
// But the old snapshot is still pinned
|
||||
assert_eq!(snap1.version(), v1);
|
||||
}
|
||||
}
|
||||
|
|
|
|||
187
crates/omnigraph/tests/lifecycle.rs
Normal file
187
crates/omnigraph/tests/lifecycle.rs
Normal file
|
|
@ -0,0 +1,187 @@
|
|||
mod helpers;
|
||||
|
||||
use std::fs;
|
||||
|
||||
use omnigraph::db::{Omnigraph, ReadTarget};
|
||||
use omnigraph_compiler::{build_schema_ir, schema_ir_pretty_json};
|
||||
use omnigraph_compiler::schema::parser::parse_schema;
|
||||
|
||||
use helpers::*;
|
||||
|
||||
#[tokio::test]
|
||||
async fn init_creates_repo() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
assert!(dir.path().join("_schema.pg").exists());
|
||||
assert!(dir.path().join("_schema.ir.json").exists());
|
||||
assert!(dir.path().join("__schema_state.json").exists());
|
||||
|
||||
let snap = snapshot_main(&db).await.unwrap();
|
||||
assert!(snap.entry("node:Person").is_some());
|
||||
assert!(snap.entry("node:Company").is_some());
|
||||
assert!(snap.entry("edge:Knows").is_some());
|
||||
assert!(snap.entry("edge:WorksAt").is_some());
|
||||
|
||||
assert_eq!(db.catalog().node_types.len(), 2);
|
||||
assert_eq!(db.catalog().edge_types.len(), 2);
|
||||
assert_eq!(
|
||||
db.catalog().node_types["Person"].key_property(),
|
||||
Some("name")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn open_reads_existing_repo() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
|
||||
Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let db = Omnigraph::open(uri).await.unwrap();
|
||||
assert_eq!(db.catalog().node_types.len(), 2);
|
||||
assert_eq!(db.catalog().edge_types.len(), 2);
|
||||
let snap = snapshot_main(&db).await.unwrap();
|
||||
assert!(snap.entry("node:Person").is_some());
|
||||
assert!(snap.entry("edge:Knows").is_some());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn open_bootstraps_legacy_schema_state_for_main_only_repo() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
fs::remove_file(dir.path().join("_schema.ir.json")).unwrap();
|
||||
fs::remove_file(dir.path().join("__schema_state.json")).unwrap();
|
||||
|
||||
let db = Omnigraph::open(uri).await.unwrap();
|
||||
assert_eq!(db.catalog().node_types.len(), 2);
|
||||
assert!(dir.path().join("_schema.ir.json").exists());
|
||||
assert!(dir.path().join("__schema_state.json").exists());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn open_rejects_legacy_repo_with_public_branch() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
db.branch_create("feature").await.unwrap();
|
||||
|
||||
fs::remove_file(dir.path().join("_schema.ir.json")).unwrap();
|
||||
fs::remove_file(dir.path().join("__schema_state.json")).unwrap();
|
||||
|
||||
let err = match Omnigraph::open(uri).await {
|
||||
Ok(_) => panic!("expected legacy repo with public branch to fail schema bootstrap"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("public branches block schema evolution entirely")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn long_lived_handle_rejects_schema_source_drift() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let drifted = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
fs::write(dir.path().join("_schema.pg"), drifted).unwrap();
|
||||
|
||||
let err = match db.snapshot_of(ReadTarget::branch("main")).await {
|
||||
Ok(_) => panic!("expected schema source drift to be rejected"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("current _schema.pg no longer matches the accepted compiled schema")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn long_lived_handle_rejects_schema_ir_drift() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
fs::write(dir.path().join("_schema.ir.json"), "{not valid json").unwrap();
|
||||
|
||||
let err = match db.snapshot_of(ReadTarget::branch("main")).await {
|
||||
Ok(_) => panic!("expected schema IR drift to be rejected"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("accepted compiled schema contract in _schema.ir.json is invalid")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn long_lived_handle_rejects_ir_and_source_updates_without_state_update() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let drifted = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
let drifted_ast = parse_schema(&drifted).unwrap();
|
||||
let drifted_ir = build_schema_ir(&drifted_ast).unwrap();
|
||||
let drifted_ir_json = schema_ir_pretty_json(&drifted_ir).unwrap();
|
||||
fs::write(dir.path().join("_schema.pg"), drifted).unwrap();
|
||||
fs::write(dir.path().join("_schema.ir.json"), drifted_ir_json).unwrap();
|
||||
|
||||
let err = match db.snapshot_of(ReadTarget::branch("main")).await {
|
||||
Ok(_) => panic!("expected schema state mismatch to be rejected"),
|
||||
Err(err) => err,
|
||||
};
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("accepted compiled schema does not match the recorded schema state")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn comment_only_schema_edit_keeps_schema_state_valid() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let commented = format!("// comment-only drift\n{}", TEST_SCHEMA);
|
||||
fs::write(dir.path().join("_schema.pg"), commented).unwrap();
|
||||
|
||||
let snapshot = db.snapshot_of(ReadTarget::branch("main")).await.unwrap();
|
||||
assert!(snapshot.entry("node:Person").is_some());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn open_nonexistent_fails() {
|
||||
let result = Omnigraph::open("/tmp/nonexistent_omnigraph_test_xyz").await;
|
||||
assert!(result.is_err());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn snapshot_version_is_pinned() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let snap1 = snapshot_main(&db).await.unwrap();
|
||||
let v1 = snap1.version();
|
||||
|
||||
omnigraph::loader::load_jsonl(
|
||||
&mut db,
|
||||
r#"{"type": "Person", "data": {"name": "Alice", "age": 30}}"#,
|
||||
omnigraph::loader::LoadMode::Overwrite,
|
||||
)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
let snap2 = snapshot_main(&db).await.unwrap();
|
||||
assert!(snap2.version() > v1);
|
||||
|
||||
assert_eq!(snap1.version(), v1);
|
||||
}
|
||||
101
crates/omnigraph/tests/schema_apply.rs
Normal file
101
crates/omnigraph/tests/schema_apply.rs
Normal file
|
|
@ -0,0 +1,101 @@
|
|||
mod helpers;
|
||||
|
||||
use std::fs;
|
||||
|
||||
use omnigraph::db::{Omnigraph, ReadTarget};
|
||||
use omnigraph_compiler::{SchemaMigrationStep, SchemaTypeKind};
|
||||
|
||||
use helpers::*;
|
||||
|
||||
#[tokio::test]
|
||||
async fn plan_schema_reports_supported_additive_change() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let desired = TEST_SCHEMA.replace(
|
||||
" age: I32?\n}",
|
||||
" age: I32?\n nickname: String?\n}",
|
||||
);
|
||||
|
||||
let plan = db.plan_schema(&desired).await.unwrap();
|
||||
assert!(plan.supported);
|
||||
assert!(plan.steps.iter().any(|step| matches!(
|
||||
step,
|
||||
SchemaMigrationStep::AddProperty {
|
||||
type_kind: SchemaTypeKind::Node,
|
||||
type_name,
|
||||
property_name,
|
||||
..
|
||||
} if type_name == "Person" && property_name == "nickname"
|
||||
)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn plan_schema_rejects_when_schema_contract_has_drifted() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let drifted = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
fs::write(dir.path().join("_schema.pg"), drifted).unwrap();
|
||||
|
||||
let err = db.plan_schema(TEST_SCHEMA).await.unwrap_err();
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("current _schema.pg no longer matches the accepted compiled schema")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn apply_schema_noop_returns_not_applied() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
|
||||
let result = db.apply_schema(TEST_SCHEMA).await.unwrap();
|
||||
assert!(result.supported);
|
||||
assert!(!result.applied);
|
||||
assert!(result.steps.is_empty());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn apply_schema_rejects_when_non_main_branch_exists() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
db.branch_create("feature").await.unwrap();
|
||||
|
||||
let desired = TEST_SCHEMA.replace(
|
||||
" age: I32?\n}",
|
||||
" age: I32?\n nickname: String?\n}",
|
||||
);
|
||||
let err = db.apply_schema(&desired).await.unwrap_err();
|
||||
assert!(
|
||||
err.to_string()
|
||||
.contains("schema apply requires a repo with only main")
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn apply_schema_unsupported_plan_does_not_advance_manifest() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let uri = dir.path().to_str().unwrap();
|
||||
let mut db = Omnigraph::init(uri, TEST_SCHEMA).await.unwrap();
|
||||
let before_version = db
|
||||
.snapshot_of(ReadTarget::branch("main"))
|
||||
.await
|
||||
.unwrap()
|
||||
.version();
|
||||
|
||||
let desired = TEST_SCHEMA.replace("age: I32?", "age: I64?");
|
||||
let err = db.apply_schema(&desired).await.unwrap_err();
|
||||
assert!(err.to_string().contains("changing property type"));
|
||||
assert_eq!(
|
||||
db.snapshot_of(ReadTarget::branch("main"))
|
||||
.await
|
||||
.unwrap()
|
||||
.version(),
|
||||
before_version
|
||||
);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue