lora-database 0.16.1

//! CREATE INDEX and SHOW INDEXES integration tests.

use std::collections::BTreeMap;

mod test_helpers;
use test_helpers::TestDb;

use lora_database::LoraValue;
use serde_json::Value as JsonValue;

fn rows_for_index_named<'a>(rows: &'a [JsonValue], name: &str) -> Option<&'a JsonValue> {
    rows.iter()
        .find(|r| r.get("name").and_then(|v| v.as_str()) == Some(name))
}

#[test]
fn create_named_range_node_index() {
    let db = TestDb::new();
    let rows = db.run("CREATE INDEX node_range_index_name FOR (n:Person) ON (n.surname)");
    assert!(rows.is_empty(), "CREATE INDEX returns no rows on success");

    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "node_range_index_name")
        .expect("expected index in SHOW INDEXES");
    assert_eq!(entry["type"], JsonValue::String("RANGE".into()));
    assert_eq!(entry["entityType"], JsonValue::String("NODE".into()));
    assert_eq!(
        entry["labelsOrTypes"],
        JsonValue::Array(vec![JsonValue::String("Person".into())])
    );
    assert_eq!(
        entry["properties"],
        JsonValue::Array(vec![JsonValue::String("surname".into())])
    );
    assert_eq!(entry["state"], JsonValue::String("ONLINE".into()));
}

#[test]
fn schema_command_routing_ignores_long_leading_whitespace() {
    let db = TestDb::new();
    let query = format!(
        "{}CREATE INDEX spaced_index FOR (n:Person) ON (n.surname)",
        " ".repeat(128)
    );
    db.run(&query);

    let listed = db.run("SHOW INDEXES");
    assert!(rows_for_index_named(&listed, "spaced_index").is_some());
}

#[test]
fn create_default_kind_is_range() {
    let db = TestDb::new();
    db.run("CREATE INDEX default_kind FOR (n:Person) ON (n.surname)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "default_kind").unwrap();
    assert_eq!(entry["type"], JsonValue::String("RANGE".into()));
}

#[test]
fn create_relationship_range_index() {
    let db = TestDb::new();
    db.run("CREATE INDEX rel_range_index_name FOR ()-[r:KNOWS]-() ON (r.since)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "rel_range_index_name").unwrap();
    assert_eq!(entry["type"], JsonValue::String("RANGE".into()));
    assert_eq!(
        entry["entityType"],
        JsonValue::String("RELATIONSHIP".into())
    );
    assert_eq!(
        entry["labelsOrTypes"],
        JsonValue::Array(vec![JsonValue::String("KNOWS".into())])
    );
}

#[test]
fn create_composite_node_range_index() {
    let db = TestDb::new();
    db.run("CREATE INDEX composite_range_node_index_name FOR (n:Person) ON (n.age, n.country)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "composite_range_node_index_name").unwrap();
    assert_eq!(
        entry["properties"],
        JsonValue::Array(vec![
            JsonValue::String("age".into()),
            JsonValue::String("country".into())
        ])
    );
}

#[test]
fn create_composite_relationship_range_index() {
    let db = TestDb::new();
    db.run(
        "CREATE INDEX composite_range_rel_index_name FOR ()-[r:PURCHASED]-() ON (r.date, r.amount)",
    );
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "composite_range_rel_index_name").unwrap();
    assert_eq!(
        entry["properties"],
        JsonValue::Array(vec![
            JsonValue::String("date".into()),
            JsonValue::String("amount".into())
        ])
    );
}

#[test]
fn create_text_index() {
    let db = TestDb::new();
    db.run("CREATE TEXT INDEX node_text_index_nickname FOR (n:Person) ON (n.nickname)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "node_text_index_nickname").unwrap();
    assert_eq!(entry["type"], JsonValue::String("TEXT".into()));
}

#[test]
fn create_relationship_text_index() {
    let db = TestDb::new();
    db.run("CREATE TEXT INDEX rel_text_index_name FOR ()-[r:KNOWS]-() ON (r.interest)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "rel_text_index_name").unwrap();
    assert_eq!(entry["type"], JsonValue::String("TEXT".into()));
    assert_eq!(
        entry["entityType"],
        JsonValue::String("RELATIONSHIP".into())
    );
}

#[test]
fn create_point_index() {
    let db = TestDb::new();
    db.run("CREATE POINT INDEX node_point_index_name FOR (n:Person) ON (n.sublocation)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "node_point_index_name").unwrap();
    assert_eq!(entry["type"], JsonValue::String("POINT".into()));
}

#[test]
fn create_point_index_with_options() {
    let db = TestDb::new();
    db.run(
        "CREATE POINT INDEX point_index_with_config FOR (n:Label) ON (n.prop2) \
         OPTIONS { indexConfig: { `spatial.cartesian.min`: [-100.0, -100.0], \
         `spatial.cartesian.max`: [100.0, 100.0] } }",
    );
    let listed = db.run("SHOW INDEXES");
    assert!(rows_for_index_named(&listed, "point_index_with_config").is_some());
}

#[test]
fn create_lookup_index_node() {
    let db = TestDb::new();
    db.run("CREATE LOOKUP INDEX node_label_lookup_index FOR (n) ON EACH node.labels(n)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "node_label_lookup_index").unwrap();
    assert_eq!(entry["type"], JsonValue::String("LOOKUP".into()));
    assert_eq!(entry["entityType"], JsonValue::String("NODE".into()));
    assert_eq!(entry["properties"], JsonValue::Array(Vec::new()));
}

#[test]
fn create_lookup_index_relationship() {
    let db = TestDb::new();
    db.run("CREATE LOOKUP INDEX rel_type_lookup_index FOR ()-[r]-() ON EACH edge.type(r)");
    let listed = db.run("SHOW INDEXES");
    let entry = rows_for_index_named(&listed, "rel_type_lookup_index").unwrap();
    assert_eq!(entry["type"], JsonValue::String("LOOKUP".into()));
    assert_eq!(
        entry["entityType"],
        JsonValue::String("RELATIONSHIP".into())
    );
}

#[test]
fn create_index_using_parameter_for_name() {
    let db = TestDb::new();
    let mut params: BTreeMap<String, LoraValue> = BTreeMap::new();
    params.insert("name".into(), LoraValue::String("range_index_param".into()));
    let _ = db.run_with_params("CREATE INDEX $name FOR (n:Person) ON (n.firstname)", params);
    let listed = db.run("SHOW INDEXES");
    assert!(rows_for_index_named(&listed, "range_index_param").is_some());
}

#[test]
fn duplicate_index_name_errors() {
    let db = TestDb::new();
    db.run("CREATE INDEX dupe FOR (n:Person) ON (n.surname)");
    let err = db
        .exec("CREATE INDEX dupe FOR (n:Movie) ON (n.title)")
        .expect_err("duplicate name should fail");
    let msg = err.to_string();
    assert!(
        msg.contains("same name") || msg.contains("dupe"),
        "expected duplicated name error, got: {msg}"
    );
}

#[test]
fn equivalent_index_errors() {
    let db = TestDb::new();
    db.run("CREATE INDEX original FOR (n:Person) ON (n.surname)");
    let err = db
        .exec("CREATE INDEX another_name FOR (n:Person) ON (n.surname)")
        .expect_err("equivalent index should fail");
    let msg = err.to_string();
    assert!(
        msg.contains("equivalent") || msg.contains("Person"),
        "expected equivalent index error, got: {msg}"
    );
}

#[test]
fn if_not_exists_is_idempotent_on_name_match() {
    let db = TestDb::new();
    db.run("CREATE INDEX once FOR (n:Person) ON (n.surname)");
    let rows = db.run("CREATE INDEX once IF NOT EXISTS FOR (n:Movie) ON (n.title)");
    assert!(rows.is_empty(), "IF NOT EXISTS should not return rows");
    let listed = db.run("SHOW INDEXES");
    assert_eq!(listed.len(), 1, "no second index should have been created");
}

#[test]
fn if_not_exists_is_idempotent_on_schema_match() {
    let db = TestDb::new();
    db.run("CREATE INDEX one FOR (n:Person) ON (n.surname)");
    let rows = db.run("CREATE INDEX two IF NOT EXISTS FOR (n:Person) ON (n.surname)");
    assert!(rows.is_empty());
    let listed = db.run("SHOW INDEXES");
    assert_eq!(
        listed.len(),
        1,
        "schema-equivalent re-create should be a no-op"
    );
}

#[test]
fn auto_generated_name_when_omitted() {
    let db = TestDb::new();
    db.run("CREATE INDEX FOR (n:Person) ON (n.surname)");
    let listed = db.run("SHOW INDEXES");
    assert_eq!(listed.len(), 1);
    let name = listed[0]["name"].as_str().unwrap();
    assert!(
        name.starts_with("index_"),
        "expected auto-generated name to start with 'index_', got '{name}'"
    );
}

#[test]
fn show_indexes_empty_when_none_created() {
    let db = TestDb::new();
    let listed = db.run("SHOW INDEXES");
    assert!(listed.is_empty());
}

#[test]
fn drop_index_removes_named_entry() {
    let db = TestDb::new();
    db.run("CREATE INDEX gone FOR (n:Person) ON (n.surname)");
    let listed = db.run("SHOW INDEXES");
    assert_eq!(listed.len(), 1);
    let _ = db.run("DROP INDEX gone");
    let listed = db.run("SHOW INDEXES");
    assert!(listed.is_empty());
}

#[test]
fn drop_missing_index_errors() {
    let db = TestDb::new();
    let err = db
        .exec("DROP INDEX nope")
        .expect_err("dropping a missing index should fail");
    let msg = err.to_string();
    assert!(msg.contains("42N51"), "expected 42N51 status, got: {msg}");
    assert!(msg.contains("nope"));
    assert_eq!(err.code(), lora_database::LoraErrorCode::NotFound);
}

#[test]
fn drop_missing_index_if_exists_is_noop() {
    let db = TestDb::new();
    let rows = db.run("DROP INDEX nope IF EXISTS");
    assert!(rows.is_empty());
}

#[test]
fn create_index_error_carries_gql_status() {
    let db = TestDb::new();
    db.run("CREATE INDEX dupe FOR (n:Person) ON (n.surname)");
    let err = db
        .exec("CREATE INDEX dupe FOR (n:Movie) ON (n.title)")
        .expect_err("duplicate name should fail");
    assert!(err.to_string().contains("22N71"));
}

// ---------- SHOW INDEXES type filter ----------

fn seed_mixed_indexes(db: &TestDb) {
    db.run("CREATE RANGE INDEX r1 FOR (n:Person) ON (n.surname)");
    db.run("CREATE TEXT INDEX t1 FOR (n:Person) ON (n.name)");
    db.run("CREATE POINT INDEX p1 FOR (n:Place) ON (n.location)");
    db.run("CREATE LOOKUP INDEX l1 FOR (n) ON EACH node.labels(n)");
}

fn index_names(rows: &[JsonValue]) -> Vec<String> {
    rows.iter()
        .filter_map(|r| r.get("name").and_then(|v| v.as_str()).map(String::from))
        .collect()
}

#[test]
fn show_range_indexes_filters_by_type() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW RANGE INDEXES");
    let names = index_names(&rows);
    assert_eq!(names, vec!["r1"], "got {names:?}");
}

#[test]
fn show_text_indexes_filters_by_type() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let names = index_names(&db.run("SHOW TEXT INDEXES"));
    assert_eq!(names, vec!["t1"], "got {names:?}");
}

#[test]
fn show_point_indexes_filters_by_type() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let names = index_names(&db.run("SHOW POINT INDEXES"));
    assert_eq!(names, vec!["p1"], "got {names:?}");
}

#[test]
fn show_lookup_indexes_filters_by_type() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let names = index_names(&db.run("SHOW LOOKUP INDEXES"));
    assert_eq!(names, vec!["l1"], "got {names:?}");
}

#[test]
fn show_all_indexes_returns_every_kind() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW ALL INDEXES");
    assert_eq!(rows.len(), 4);
}

#[test]
fn show_vector_indexes_is_empty_when_unsupported() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    assert!(db.run("SHOW VECTOR INDEXES").is_empty());
}

#[test]
fn show_fulltext_indexes_is_empty_when_unsupported() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    assert!(db.run("SHOW FULLTEXT INDEXES").is_empty());
}

#[test]
fn create_fulltext_index_now_supported() {
    let db = TestDb::new();
    db.run("CREATE FULLTEXT INDEX names FOR (n:Person) ON EACH [n.name]");
    let listed = db.run("SHOW INDEXES");
    let entry = listed
        .iter()
        .find(|r| r.get("name").and_then(|v| v.as_str()) == Some("names"))
        .expect("fulltext index listed");
    assert_eq!(entry["type"], JsonValue::String("FULLTEXT".into()));
}

#[test]
fn show_index_filter_also_accepts_singular_keyword() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let names = index_names(&db.run("SHOW RANGE INDEX"));
    assert_eq!(names, vec!["r1"]);
}

// ---------- SHOW INDEXES YIELD / WHERE / RETURN ----------

#[test]
fn show_indexes_yield_columns_only() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW INDEXES YIELD name, type");
    assert!(!rows.is_empty());
    for row in &rows {
        let obj = row.as_object().expect("row is object");
        // YIELD restricts the output to just the listed columns.
        assert_eq!(obj.len(), 2, "row has unexpected columns: {obj:?}");
        assert!(obj.contains_key("name"));
        assert!(obj.contains_key("type"));
    }
}

#[test]
fn show_indexes_yield_with_alias() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW INDEXES YIELD name AS idx_name");
    for row in &rows {
        let obj = row.as_object().unwrap();
        assert!(obj.contains_key("idx_name"));
        assert!(!obj.contains_key("name"));
    }
}

#[test]
fn show_indexes_yield_star_is_pass_through() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let pre = db.run("SHOW INDEXES");
    let post = db.run("SHOW INDEXES YIELD *");
    assert_eq!(pre.len(), post.len());
    // Same column shape on both sides.
    let pre_keys: std::collections::BTreeSet<String> =
        pre[0].as_object().unwrap().keys().cloned().collect();
    let post_keys: std::collections::BTreeSet<String> =
        post[0].as_object().unwrap().keys().cloned().collect();
    assert_eq!(pre_keys, post_keys);
}

#[test]
fn show_indexes_where_filters_yielded_rows() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW INDEXES YIELD name, type WHERE type = 'RANGE'");
    assert_eq!(index_names(&rows), vec!["r1"]);
}

#[test]
fn show_indexes_yield_order_by_limit() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW INDEXES YIELD name ORDER BY name LIMIT 2");
    let names = index_names(&rows);
    assert_eq!(names, vec!["l1", "p1"], "got {names:?}");
}

#[test]
fn show_indexes_yield_skip() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run("SHOW INDEXES YIELD name ORDER BY name SKIP 2");
    let names = index_names(&rows);
    assert_eq!(names, vec!["r1", "t1"], "got {names:?}");
}

#[test]
fn show_indexes_yield_where_return() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let rows = db.run(
        "SHOW INDEXES YIELD name, type, entityType \
         WHERE entityType = 'NODE' \
         RETURN name ORDER BY name LIMIT 2",
    );
    let names = index_names(&rows);
    assert_eq!(names, vec!["l1", "p1"]);
    // RETURN projected only `name`.
    for row in &rows {
        let obj = row.as_object().unwrap();
        assert_eq!(obj.keys().collect::<Vec<_>>(), vec!["name"]);
    }
}

#[test]
fn show_indexes_filter_combined_with_yield_pipeline() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let names = index_names(
        &db.run("SHOW RANGE INDEXES YIELD name, type WHERE type = 'RANGE' RETURN name"),
    );
    assert_eq!(names, vec!["r1"]);
}

#[test]
fn show_indexes_yield_parameter_in_limit() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let mut params = BTreeMap::new();
    params.insert("lim".into(), LoraValue::Int(2));
    let rows = db.run_with_params("SHOW INDEXES YIELD name ORDER BY name LIMIT $lim", params);
    assert_eq!(rows.len(), 2);
}

#[test]
fn show_indexes_return_without_yield_fails_to_parse() {
    let db = TestDb::new();
    let err = db.run_err("SHOW INDEXES RETURN name");
    assert!(
        err.to_lowercase().contains("parse") || err.to_lowercase().contains("syntax"),
        "expected parse error, got: {err}"
    );
}

#[test]
fn show_indexes_yield_starts_with_predicate() {
    let db = TestDb::new();
    seed_mixed_indexes(&db);
    let names =
        index_names(&db.run("SHOW INDEXES YIELD name WHERE name STARTS WITH 'r' RETURN name"));
    assert_eq!(names, vec!["r1"]);
}

// ---------- SHOW CONSTRAINTS pipeline ----------

#[test]
fn show_constraints_yield_where_return() {
    let db = TestDb::new();
    db.run("CREATE CONSTRAINT u FOR (n:Book) REQUIRE n.isbn IS UNIQUE");
    db.run("CREATE CONSTRAINT e FOR (a:Author) REQUIRE a.name IS NOT NULL");
    db.run("CREATE CONSTRAINT t FOR (m:Movie) REQUIRE m.title IS :: STRING");
    let rows = db.run(
        "SHOW CONSTRAINTS YIELD name, type \
         WHERE type = 'NODE_PROPERTY_UNIQUENESS' \
         RETURN name",
    );
    let names: Vec<String> = rows
        .iter()
        .filter_map(|r| r.get("name").and_then(|v| v.as_str()).map(String::from))
        .collect();
    assert_eq!(names, vec!["u"]);
}

#[test]
fn show_constraints_yield_order_by_skip_limit() {
    let db = TestDb::new();
    db.run("CREATE CONSTRAINT c1 FOR (n:A) REQUIRE n.x IS UNIQUE");
    db.run("CREATE CONSTRAINT c2 FOR (n:B) REQUIRE n.y IS UNIQUE");
    db.run("CREATE CONSTRAINT c3 FOR (n:C) REQUIRE n.z IS UNIQUE");
    let rows = db.run("SHOW CONSTRAINTS YIELD name ORDER BY name SKIP 1 LIMIT 1");
    let names: Vec<String> = rows
        .iter()
        .filter_map(|r| r.get("name").and_then(|v| v.as_str()).map(String::from))
        .collect();
    assert_eq!(names, vec!["c2"]);
}