rdbms-playground/tests/sql_create_table.rs

//! Sub-phase 4a integration tests for advanced-mode SQL
//! `CREATE TABLE` (ADR-0035 §1/§4).
//!
//! Worker round-trip: a `Command::SqlCreateTable` executes
//! **structurally** through the existing `do_create_table` machinery,
//! so an advanced-mode-created table is a first-class playground
//! object (metadata + the ten-type vocabulary). Covers:
//! - Created tables appear in `list_tables` and `describe_table`
//!   reports the playground `user_type` per column.
//! - A `serial` sole-PK autoincrements even in a multi-column table
//!   (the §6.4 inline-`PRIMARY KEY` extension).
//! - `IF NOT EXISTS` on an existing table is a no-op (`Skipped`); the
//!   plain form errors when the table exists (§4).
//! - One SQL `CREATE TABLE` is exactly one undo step (ADR-0006).
//!
//! Parsing (text → `Command::SqlCreateTable`) is covered by the
//! `builder_tests` in `src/dsl/grammar/sql_create_table.rs`; these
//! tests drive the worker directly, mirroring `tests/sql_insert.rs`.

use rdbms_playground::db::{CreateOutcome, Database};
use rdbms_playground::dsl::{ColumnSpec, Type, Value};
use rdbms_playground::persistence::Persistence;
use rdbms_playground::project;

fn rt() -> tokio::runtime::Runtime {
    tokio::runtime::Builder::new_current_thread()
        .enable_all()
        .build()
        .expect("tokio rt")
}

fn open(undo: bool) -> (project::Project, Database, tempfile::TempDir) {
    let dir = tempfile::tempdir().expect("create tempdir");
    let project =
        project::open_or_create(None, Some(dir.path())).expect("open or create project");
    let persistence = Persistence::new(project.path().to_path_buf());
    let db = Database::open_with_persistence_and_undo(project.db_path(), persistence, undo)
        .expect("open db with persistence");
    (project, db, dir)
}

#[test]
fn created_table_appears_with_playground_types() {
    let (_p, db, _d) = open(false);
    let r = rt();
    let out = r
        .block_on(db.sql_create_table(
            "Widget".to_string(),
            vec![
                ColumnSpec::new("id", Type::Int),
                ColumnSpec::new("name", Type::Text),
            ],
            vec!["id".to_string()],
            vec![], // no composite UNIQUE
            false,
            Some("create table Widget (id int primary key, name text)".to_string()),
        ))
        .expect("create should succeed");
    assert!(matches!(out, CreateOutcome::Created(_)));

    let tables = r.block_on(db.list_tables()).expect("list");
    assert!(tables.contains(&"Widget".to_string()));

    let desc = r
        .block_on(db.describe_table("Widget".to_string(), None))
        .expect("describe");
    let types: Vec<(String, Option<Type>)> = desc
        .columns
        .iter()
        .map(|c| (c.name.clone(), c.user_type))
        .collect();
    assert_eq!(
        types,
        vec![
            ("id".to_string(), Some(Type::Int)),
            ("name".to_string(), Some(Type::Text)),
        ]
    );
}

#[test]
fn integer_primary_key_is_plain_int() {
    // ADR-0035 §3: INTEGER PRIMARY KEY maps to plain `int`, not
    // `serial`. The structural object reports `int`.
    let (_p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![ColumnSpec::new("id", Type::Int)],
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id integer primary key)".to_string()),
    ))
    .expect("create");
    let desc = r
        .block_on(db.describe_table("T".to_string(), None))
        .expect("describe");
    assert_eq!(desc.columns[0].user_type, Some(Type::Int));
}

#[test]
fn serial_pk_autoincrements_in_multi_column_table() {
    // §6.4: a `serial` sole-PK in a *multi-column* table must inline
    // `PRIMARY KEY` so it keeps autoincrement (rowid-alias) semantics
    // — the case simple mode never produces in one statement.
    let (_p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![
            ColumnSpec::new("id", Type::Serial),
            ColumnSpec::new("name", Type::Text),
        ],
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id serial primary key, name text)".to_string()),
    ))
    .expect("create");

    // Form B inserts (no column list): the serial id is auto-filled.
    for name in ["a", "b"] {
        r.block_on(db.insert(
            "T".to_string(),
            None,
            vec![Value::Text(name.to_string())],
            Some(format!("insert into T (name) values ('{name}')")),
        ))
        .expect("insert");
    }

    let data = r
        .block_on(db.query_data("T".to_string(), None, None, None))
        .expect("query");
    let id_idx = data
        .columns
        .iter()
        .position(|c| c == "id")
        .expect("id column");
    let mut ids: Vec<Option<String>> = data.rows.iter().map(|row| row[id_idx].clone()).collect();
    ids.sort();
    assert_eq!(
        ids,
        vec![Some("1".to_string()), Some("2".to_string())],
        "serial PK autoincremented 1, 2"
    );
}

#[test]
fn if_not_exists_is_a_noop_when_table_exists() {
    let (_p, db, _d) = open(false);
    let r = rt();
    let specs = || vec![ColumnSpec::new("id", Type::Int)];
    r.block_on(db.sql_create_table(
        "T".to_string(),
        specs(),
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id int)".to_string()),
    ))
    .expect("first create");

    let out = r
        .block_on(db.sql_create_table(
            "T".to_string(),
            specs(),
            vec!["id".to_string()],
            vec![], // no composite UNIQUE
            true, // IF NOT EXISTS
            Some("create table if not exists T (id int)".to_string()),
        ))
        .expect("second create should succeed as a no-op");
    assert!(
        matches!(out, CreateOutcome::Skipped(_)),
        "IF NOT EXISTS on an existing table is a no-op"
    );

    let tables = r.block_on(db.list_tables()).expect("list");
    assert_eq!(tables.iter().filter(|t| t.as_str() == "T").count(), 1);
}

#[test]
fn table_without_primary_key_is_allowed() {
    // Advanced mode allows a PK-less table (standard SQL; the
    // "trust the user like SQL" posture, ADR-0035 §7) — unlike simple
    // mode, which requires/defaults a PK. User-confirmed 2026-05-25.
    let (_p, db, _d) = open(false);
    let r = rt();
    let out = r
        .block_on(db.sql_create_table(
            "Notes".to_string(),
            vec![ColumnSpec::new("body", Type::Text)],
            vec![], // no primary key
            vec![], // no composite UNIQUE
            false,
            Some("create table Notes (body text)".to_string()),
        ))
        .expect("a PK-less table should create");
    assert!(matches!(out, CreateOutcome::Created(_)));
    // And it is usable: a row inserts and reads back.
    r.block_on(db.insert(
        "Notes".to_string(),
        None,
        vec![Value::Text("hello".to_string())],
        Some("insert into Notes (body) values ('hello')".to_string()),
    ))
    .expect("insert into PK-less table");
    let data = r
        .block_on(db.query_data("Notes".to_string(), None, None, None))
        .expect("query");
    assert_eq!(data.rows.len(), 1);
}

/// A column carrying a raw-SQL `CHECK` (ADR-0035 §4a.2).
fn col_check(name: &str, ty: Type, check_sql: &str) -> ColumnSpec {
    let mut c = ColumnSpec::new(name, ty);
    c.check_sql = Some(check_sql.to_string());
    c
}

/// A column carrying a raw-SQL `DEFAULT` (ADR-0035 §4a.2).
fn col_default(name: &str, ty: Type, default_sql: &str) -> ColumnSpec {
    let mut c = ColumnSpec::new(name, ty);
    c.default_sql = Some(default_sql.to_string());
    c
}

#[test]
fn check_constraint_is_enforced() {
    let (_p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![ColumnSpec::new("id", Type::Serial), col_check("price", Type::Real, "price >= 0")],
        vec!["id".to_string()],
        vec![],
        false,
        Some("create table T (id serial primary key, price real check (price >= 0))".to_string()),
    ))
    .expect("create");
    // A satisfying row inserts; a violating one is rejected by the CHECK.
    r.block_on(db.insert(
        "T".to_string(),
        Some(vec!["price".to_string()]),
        vec![Value::Number("10".to_string())],
        Some("insert".to_string()),
    ))
    .expect("price 10 satisfies the check");
    let bad = r.block_on(db.insert(
        "T".to_string(),
        Some(vec!["price".to_string()]),
        vec![Value::Number("-5".to_string())],
        Some("insert".to_string()),
    ));
    assert!(bad.is_err(), "CHECK (price >= 0) rejects -5");
}

#[test]
fn default_is_applied_when_column_omitted() {
    let (_p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![
            ColumnSpec::new("id", Type::Serial),
            ColumnSpec::new("label", Type::Text),
            col_default("n", Type::Int, "7"),
        ],
        vec!["id".to_string()],
        vec![],
        false,
        Some("create table T (id serial primary key, label text, n int default 7)".to_string()),
    ))
    .expect("create");
    // Insert only `label`; `id` auto-fills and `n` takes its default.
    r.block_on(db.insert(
        "T".to_string(),
        Some(vec!["label".to_string()]),
        vec![Value::Text("x".to_string())],
        Some("insert".to_string()),
    ))
    .expect("insert");
    let data = r
        .block_on(db.query_data("T".to_string(), None, None, None))
        .expect("query");
    let n_idx = data.columns.iter().position(|c| c == "n").expect("n column");
    assert_eq!(data.rows[0][n_idx].as_deref(), Some("7"), "DEFAULT 7 applied");
}

#[test]
fn composite_unique_is_enforced() {
    let (_p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![ColumnSpec::new("a", Type::Int), ColumnSpec::new("b", Type::Int)],
        vec![],
        vec![vec!["a".to_string(), "b".to_string()]],
        false,
        Some("create table T (a int, b int, unique (a, b))".to_string()),
    ))
    .expect("create");
    let ins = |a: &str, b: &str| {
        db.insert(
            "T".to_string(),
            None,
            vec![Value::Number(a.to_string()), Value::Number(b.to_string())],
            Some("insert".to_string()),
        )
    };
    r.block_on(ins("1", "2")).expect("first (1,2)");
    assert!(r.block_on(ins("1", "2")).is_err(), "UNIQUE(a,b) rejects duplicate (1,2)");
    r.block_on(ins("1", "3")).expect("distinct (1,3) is allowed");
}

#[test]
fn check_default_and_composite_unique_survive_rebuild() {
    // The part-D round-trip: CHECK (metadata), DEFAULT (PRAGMA), and
    // composite UNIQUE (TableSchema + PRAGMA index_list origin 'u')
    // must all be reconstructed from project.yaml on rebuild.
    let (p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![
            ColumnSpec::new("a", Type::Int),
            ColumnSpec::new("b", Type::Int),
            col_check("price", Type::Real, "price >= 0"),
            col_default("n", Type::Int, "7"),
        ],
        vec![],
        vec![vec!["a".to_string(), "b".to_string()]],
        false,
        Some(
            "create table T (a int, b int, price real check (price >= 0), \
             n int default 7, unique (a, b))"
                .to_string(),
        ),
    ))
    .expect("create");

    r.block_on(db.rebuild_from_text(p.path().to_path_buf(), None))
        .expect("rebuild");

    let ins = |a: &str, b: &str, price: &str| {
        db.insert(
            "T".to_string(),
            Some(vec!["a".to_string(), "b".to_string(), "price".to_string()]),
            vec![
                Value::Number(a.to_string()),
                Value::Number(b.to_string()),
                Value::Number(price.to_string()),
            ],
            Some("insert".to_string()),
        )
    };
    // CHECK survived: a negative price is rejected.
    assert!(r.block_on(ins("1", "1", "-1")).is_err(), "CHECK survived rebuild");
    // A valid row inserts; DEFAULT n=7 survived.
    r.block_on(ins("1", "1", "5")).expect("valid row");
    let data = r
        .block_on(db.query_data("T".to_string(), None, None, None))
        .expect("query");
    let n_idx = data.columns.iter().position(|c| c == "n").expect("n column");
    assert_eq!(data.rows[0][n_idx].as_deref(), Some("7"), "DEFAULT survived rebuild");
    // Composite UNIQUE survived: (1,1) again is rejected.
    assert!(r.block_on(ins("1", "1", "5")).is_err(), "composite UNIQUE survived rebuild");
}

#[test]
fn if_not_exists_noop_is_journalled() {
    // A successful no-op is still a submission and belongs in the
    // complete journal (ADR-0034) — like read-only `show table`, and
    // unlike a *failed* duplicate-create (journalled `err`).
    let (p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![ColumnSpec::new("id", Type::Int)],
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id int)".to_string()),
    ))
    .expect("first create");
    let noop = "create table if not exists T (id int)";
    let out = r
        .block_on(db.sql_create_table(
            "T".to_string(),
            vec![ColumnSpec::new("id", Type::Int)],
            vec!["id".to_string()],
            vec![], // no composite UNIQUE
            true,
            Some(noop.to_string()),
        ))
        .expect("no-op");
    assert!(matches!(out, CreateOutcome::Skipped(_)));
    let log = std::fs::read_to_string(p.path().join("history.log")).expect("read history.log");
    assert!(log.contains(noop), "the no-op skip should be journalled; log:\n{log}");
}

#[test]
fn plain_create_errors_when_table_exists() {
    let (_p, db, _d) = open(false);
    let r = rt();
    let specs = || vec![ColumnSpec::new("id", Type::Int)];
    r.block_on(db.sql_create_table(
        "T".to_string(),
        specs(),
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id int)".to_string()),
    ))
    .expect("first create");

    let err = r.block_on(db.sql_create_table(
        "T".to_string(),
        specs(),
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false, // no IF NOT EXISTS
        Some("create table T (id int)".to_string()),
    ));
    assert!(err.is_err(), "re-creating an existing table without IF NOT EXISTS errors");
}

#[test]
fn sql_create_table_is_one_undo_step() {
    let (_p, db, _d) = open(true); // undo enabled
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![ColumnSpec::new("id", Type::Int)],
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id int)".to_string()),
    ))
    .expect("create");
    assert!(r.block_on(db.list_tables()).unwrap().contains(&"T".to_string()));

    let undone = r.block_on(db.undo()).expect("undo call");
    assert!(undone.is_some(), "the CREATE TABLE recorded one undo step");
    assert!(
        !r.block_on(db.list_tables()).unwrap().contains(&"T".to_string()),
        "table is gone after a single undo"
    );
}

/// Sorted `id` column values of table `T`.
fn ids(db: &Database, r: &tokio::runtime::Runtime) -> Vec<Option<String>> {
    let d = r
        .block_on(db.query_data("T".to_string(), None, None, None))
        .expect("query");
    let idx = d.columns.iter().position(|c| c == "id").expect("id column");
    let mut v: Vec<Option<String>> = d.rows.iter().map(|row| row[idx].clone()).collect();
    v.sort();
    v
}

fn insert_row(db: &Database, r: &tokio::runtime::Runtime, name: &str) {
    r.block_on(db.insert(
        "T".to_string(),
        None,
        vec![Value::Text(name.to_string())],
        Some(format!("insert into T (name) values ('{name}')")),
    ))
    .expect("insert");
}

/// `serial` PK as the **first** column must keep autoincrement across a
/// rebuild: the structural create and the `schema_to_ddl` rebuild both
/// inline `PRIMARY KEY` on a first-column single PK, so the DDL is
/// identical and the sequence continues (id 3 after rebuild).
#[test]
fn serial_pk_first_column_autoincrements_after_rebuild() {
    let (p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![
            ColumnSpec::new("id", Type::Serial),
            ColumnSpec::new("name", Type::Text),
        ],
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (id serial primary key, name text)".to_string()),
    ))
    .expect("create");
    insert_row(&db, &r, "a");
    insert_row(&db, &r, "b");
    r.block_on(db.rebuild_from_text(p.path().to_path_buf(), None))
        .expect("rebuild");
    insert_row(&db, &r, "c");
    assert_eq!(
        ids(&db, &r),
        vec![Some("1".to_string()), Some("2".to_string()), Some("3".to_string())]
    );
}

/// `serial` PK as a **non-first** column must also keep autoincrement
/// across a rebuild. Here the rebuild emits a *table-level* PK (the PK
/// is not column 0), proving autoincrement does not rely on the
/// rowid-alias / inline-PK form — the insert path computes the next
/// value itself (ADR-0035 §6.4). Guards against silent round-trip loss.
#[test]
fn serial_pk_non_first_column_autoincrements_after_rebuild() {
    let (p, db, _d) = open(false);
    let r = rt();
    r.block_on(db.sql_create_table(
        "T".to_string(),
        vec![
            ColumnSpec::new("name", Type::Text),
            ColumnSpec::new("id", Type::Serial),
        ],
        vec!["id".to_string()],
        vec![], // no composite UNIQUE
        false,
        Some("create table T (name text, id serial primary key)".to_string()),
    ))
    .expect("create");
    insert_row(&db, &r, "a");
    insert_row(&db, &r, "b");
    assert_eq!(ids(&db, &r), vec![Some("1".to_string()), Some("2".to_string())]);
    r.block_on(db.rebuild_from_text(p.path().to_path_buf(), None))
        .expect("rebuild");
    insert_row(&db, &r, "c");
    assert_eq!(
        ids(&db, &r),
        vec![Some("1".to_string()), Some("2".to_string()), Some("3".to_string())],
        "serial keeps autoincrement after a rebuild even as a non-first column"
    );
}