test: cover dropColumnRestrictsWhenReferencedByAnotherTablesForeignKey
This commit is contained in:
dualfroz committed
1 parent
575cb89ab4
commit
d53bea481c
1 file changed
+156
@@ -157,3 +157,159 @@ class AlterTableTest {
|
||||
assertTrue(error.message!!.contains("PRIMARY KEY"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnRestrictsWhenReferencedByAnotherTablesForeignKey() {
|
||||
run("CREATE TABLE departments (id INTEGER PRIMARY KEY, code TEXT UNIQUE)")
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, dept_code TEXT REFERENCES departments(code))")
|
||||
val error = assertFailsWith<KestrelException> { run("ALTER TABLE departments DROP COLUMN code") }
|
||||
assertTrue(error.message!!.contains("FOREIGN KEY"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnCascadeRemovesTheReferencingForeignKey() {
|
||||
run("CREATE TABLE departments (id INTEGER PRIMARY KEY, code TEXT UNIQUE)")
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, dept_code TEXT REFERENCES departments(code))")
|
||||
run("INSERT INTO departments VALUES (1, 'eng')")
|
||||
run("ALTER TABLE departments DROP COLUMN code CASCADE")
|
||||
// The FK is gone, so an employee can now reference a nonexistent department code freely.
|
||||
run("INSERT INTO employees VALUES (1, 'ghost')")
|
||||
assertEquals(listOf(listOf(text("ghost"))), select("SELECT dept_code FROM employees").rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnRestrictsWhenUsedByItsOwnForeignKey() {
|
||||
run("CREATE TABLE departments (id INTEGER PRIMARY KEY)")
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, dept_id INTEGER REFERENCES departments(id))")
|
||||
val error = assertFailsWith<KestrelException> { run("ALTER TABLE employees DROP COLUMN dept_id") }
|
||||
assertTrue(error.message!!.contains("FOREIGN KEY"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnCascadeRemovesItsOwnForeignKey() {
|
||||
run("CREATE TABLE departments (id INTEGER PRIMARY KEY)")
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, dept_id INTEGER REFERENCES departments(id))")
|
||||
run("ALTER TABLE employees DROP COLUMN dept_id CASCADE")
|
||||
assertEquals(listOf("id"), select("SELECT * FROM employees").columns)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnRestrictsWhenUsedByAUniqueConstraint() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, email TEXT UNIQUE)")
|
||||
val error = assertFailsWith<KestrelException> { run("ALTER TABLE t DROP COLUMN email") }
|
||||
assertTrue(error.message!!.contains("UNIQUE"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnCascadeRemovesTheUniqueConstraint() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, email TEXT UNIQUE)")
|
||||
run("INSERT INTO t VALUES (1, 'a@x.com')")
|
||||
run("ALTER TABLE t DROP COLUMN email CASCADE")
|
||||
run("INSERT INTO t VALUES (2)")
|
||||
assertEquals(2, select("SELECT * FROM t").rows.size)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnRestrictsWhenUsedByACheckConstraint() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0))")
|
||||
val error = assertFailsWith<KestrelException> { run("ALTER TABLE t DROP COLUMN age") }
|
||||
assertTrue(error.message!!.contains("CHECK"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnCascadeRemovesTheCheckConstraint() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0))")
|
||||
run("ALTER TABLE t DROP COLUMN age CASCADE")
|
||||
run("INSERT INTO t VALUES (1)")
|
||||
assertEquals(1, select("SELECT * FROM t").rows.size)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnRestrictsWhenUsedByAnIndex() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER)")
|
||||
run("CREATE INDEX idx_age ON t (age)")
|
||||
val error = assertFailsWith<KestrelException> { run("ALTER TABLE t DROP COLUMN age") }
|
||||
assertTrue(error.message!!.contains("idx_age"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropColumnCascadeRemovesTheIndex() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER)")
|
||||
run("CREATE INDEX idx_age ON t (age)")
|
||||
run("ALTER TABLE t DROP COLUMN age CASCADE")
|
||||
assertTrue(catalog.indexNames.isEmpty())
|
||||
}
|
||||
|
||||
// ---- RENAME COLUMN ----
|
||||
|
||||
@Test
|
||||
fun renameColumnRenamesSchemaAndPreservesData() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, bio TEXT)")
|
||||
run("INSERT INTO t VALUES (1, 'mathematician')")
|
||||
run("ALTER TABLE t RENAME COLUMN bio TO biography")
|
||||
val result = select("SELECT * FROM t")
|
||||
assertEquals(listOf("id", "biography"), result.columns)
|
||||
assertEquals(listOf(listOf(int(1), text("mathematician"))), result.rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnRejectsUnknownColumn() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY)")
|
||||
assertFailsWith<KestrelException> { run("ALTER TABLE t RENAME COLUMN missing TO whatever") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnRejectsWhenNewNameAlreadyExists() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, name TEXT)")
|
||||
val error = assertFailsWith<KestrelException> { run("ALTER TABLE t RENAME COLUMN name TO id") }
|
||||
assertTrue(error.message!!.contains("already exists"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnUpdatesCompositePrimaryKeyListing() {
|
||||
run("CREATE TABLE t (region TEXT, seq INTEGER, PRIMARY KEY (region, seq))")
|
||||
run("INSERT INTO t VALUES ('eu', 1)")
|
||||
run("ALTER TABLE t RENAME COLUMN region TO area")
|
||||
assertFailsWith<KestrelException> { run("INSERT INTO t VALUES ('eu', 1)") }
|
||||
run("INSERT INTO t VALUES ('eu', 2)")
|
||||
assertEquals(2, select("SELECT * FROM t").rows.size)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnUpdatesItsOwnForeignKeyColumns() {
|
||||
run("CREATE TABLE departments (id INTEGER PRIMARY KEY)")
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, dept_id INTEGER REFERENCES departments(id))")
|
||||
run("INSERT INTO departments VALUES (1)")
|
||||
run("ALTER TABLE employees RENAME COLUMN dept_id TO department_id")
|
||||
assertFailsWith<KestrelException> { run("INSERT INTO employees VALUES (1, 99)") }
|
||||
run("INSERT INTO employees VALUES (1, 1)")
|
||||
assertEquals(1, select("SELECT * FROM employees").rows.size)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnPropagatesToOtherTablesForeignKeyReferencedColumns() {
|
||||
run("CREATE TABLE departments (id INTEGER PRIMARY KEY)")
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, dept_id INTEGER REFERENCES departments(id) ON DELETE CASCADE)")
|
||||
run("INSERT INTO departments VALUES (1)")
|
||||
run("INSERT INTO employees VALUES (1, 1)")
|
||||
run("ALTER TABLE departments RENAME COLUMN id TO dept_pk")
|
||||
run("DELETE FROM departments WHERE dept_pk = 1")
|
||||
assertTrue(select("SELECT * FROM employees").rows.isEmpty(), "cascade must still work through the renamed referenced column")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnUpdatesAUniqueConstraint() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, email TEXT UNIQUE)")
|
||||
run("INSERT INTO t VALUES (1, 'a@x.com')")
|
||||
run("ALTER TABLE t RENAME COLUMN email TO contact_email")
|
||||
assertFailsWith<KestrelException> { run("INSERT INTO t VALUES (2, 'a@x.com')") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun renameColumnUpdatesACheckConstraintAndStillEnforcesIt() {
|
||||
run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0))")
|
||||
run("ALTER TABLE t RENAME COLUMN age TO years")
|
||||
assertFailsWith<KestrelException> { run("INSERT INTO t VALUES (1, -5)") }
|
||||
run("INSERT INTO t VALUES (1, 30)")
|
||||
assertEquals(listOf(listOf(int(30))), select("SELECT years FROM t").rows)
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user