test(sql): cover columnLevelCheckAcceptsSatisfyingRowsAndRejectsViola...
This commit is contained in:
dualfroz committed
1 parent
bac0b33caf
commit
085763bb52
1 file changed
+155
@@ -3012,3 +3012,158 @@ class DiskExecutorTest {
|
|||||||
assertFailsWith<KestrelException> { db.run("SELECT * FROM seniors") }
|
assertFailsWith<KestrelException> { db.run("SELECT * FROM seniors") }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun columnLevelCheckAcceptsSatisfyingRowsAndRejectsViolatingOnes() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0))")
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO t VALUES (1, 30)").single())
|
||||||
|
val error = assertFailsWith<KestrelException> { db.run("INSERT INTO t VALUES (2, -5)") }
|
||||||
|
assertTrue(error.message!!.contains("CHECK"))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkPassesWhenExpressionIsUnknownBecauseOfNull() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0))")
|
||||||
|
// three-valued logic: NULL > 0 is UNKNOWN, not FALSE, so the row is accepted
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO t (id) VALUES (1)").single())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun tableLevelCheckCanSpanMultipleColumns() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE ranges (id INTEGER PRIMARY KEY, low INTEGER, high INTEGER, CHECK (low <= high))")
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO ranges VALUES (1, 1, 10)").single())
|
||||||
|
assertFailsWith<KestrelException> { db.run("INSERT INTO ranges VALUES (2, 10, 1)") }
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun multipleCheckConstraintsAllMustBeSatisfied() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run(
|
||||||
|
"CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age >= 0), " +
|
||||||
|
"score INTEGER CHECK (score BETWEEN 0 AND 100))"
|
||||||
|
)
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO t VALUES (1, 30, 90)").single())
|
||||||
|
assertFailsWith<KestrelException> { db.run("INSERT INTO t VALUES (2, 30, 150)") }
|
||||||
|
assertFailsWith<KestrelException> { db.run("INSERT INTO t VALUES (3, -1, 90)") }
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkIsEnforcedOnUpdateToo() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0))")
|
||||||
|
db.run("INSERT INTO t VALUES (1, 30)")
|
||||||
|
assertFailsWith<KestrelException> { db.run("UPDATE t SET age = -1 WHERE id = 1") }
|
||||||
|
assertEquals(UpdateResult(1), db.run("UPDATE t SET age = 40 WHERE id = 1").single())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkRejectsAggregateFunctions() {
|
||||||
|
val db = newDatabase()
|
||||||
|
assertFailsWith<KestrelException> {
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, amount INTEGER CHECK (SUM(amount) > 0))")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkRejectsSubqueries() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE other (id INTEGER PRIMARY KEY)")
|
||||||
|
assertFailsWith<KestrelException> {
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY CHECK (id IN (SELECT id FROM other)))")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkRejectsUnknownColumn() {
|
||||||
|
val db = newDatabase()
|
||||||
|
assertFailsWith<KestrelException> {
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, CHECK (nope > 0))")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkCanReferenceStringAndCallExpressions() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, name TEXT CHECK (LENGTH(name) > 0))")
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO t VALUES (1, 'ada')").single())
|
||||||
|
assertFailsWith<KestrelException> { db.run("INSERT INTO t VALUES (2, '')") }
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun checkDefinitionSurvivesClosingAndReopeningTheCatalog() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.run("CREATE TABLE t (id INTEGER PRIMARY KEY, age INTEGER CHECK (age > 0), CHECK (id > 0))")
|
||||||
|
db.run("INSERT INTO t VALUES (1, 30)")
|
||||||
|
db.reopen()
|
||||||
|
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO t VALUES (2, 40)").single())
|
||||||
|
assertFailsWith<KestrelException> { db.run("INSERT INTO t VALUES (3, -1)") }
|
||||||
|
assertFailsWith<KestrelException> { db.run("UPDATE t SET age = -5 WHERE id = 1") }
|
||||||
|
|
||||||
|
db.reopen()
|
||||||
|
assertFailsWith<KestrelException> { db.run("INSERT INTO t VALUES (4, -1)") }
|
||||||
|
assertEquals(2, db.select("SELECT * FROM t").rows.size)
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun Database.seedDepartmentsAndEmployees() {
|
||||||
|
run("CREATE TABLE departments (id INTEGER PRIMARY KEY, name TEXT NOT NULL)")
|
||||||
|
run(
|
||||||
|
"CREATE TABLE employees (" +
|
||||||
|
"id INTEGER PRIMARY KEY, name TEXT NOT NULL, dept_id INTEGER REFERENCES departments(id))"
|
||||||
|
)
|
||||||
|
run("INSERT INTO departments VALUES (1, 'eng'), (2, 'ops')")
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun Database.seedCascadeDepartments() {
|
||||||
|
run("CREATE TABLE departments (id INTEGER PRIMARY KEY, name TEXT NOT NULL)")
|
||||||
|
run(
|
||||||
|
"CREATE TABLE employees (id INTEGER PRIMARY KEY, name TEXT NOT NULL, " +
|
||||||
|
"dept_id INTEGER REFERENCES departments(id) ON DELETE CASCADE ON UPDATE CASCADE)"
|
||||||
|
)
|
||||||
|
run("INSERT INTO departments VALUES (1, 'eng'), (2, 'ops')")
|
||||||
|
run("INSERT INTO employees VALUES (1, 'ada', 1), (2, 'linus', 1), (3, 'grace', 2)")
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun Database.seedSetNullDepartments(childNotNull: Boolean) {
|
||||||
|
run("CREATE TABLE departments (id INTEGER PRIMARY KEY, name TEXT NOT NULL)")
|
||||||
|
val childColumn = if (childNotNull) "dept_id INTEGER NOT NULL" else "dept_id INTEGER"
|
||||||
|
run(
|
||||||
|
"CREATE TABLE employees (id INTEGER PRIMARY KEY, name TEXT NOT NULL, " +
|
||||||
|
"$childColumn REFERENCES departments(id) ON DELETE SET NULL)"
|
||||||
|
)
|
||||||
|
run("INSERT INTO departments VALUES (1, 'eng')")
|
||||||
|
run("INSERT INTO employees VALUES (1, 'ada', 1)")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun insertRejectsForeignKeyViolation() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.seedDepartmentsAndEmployees()
|
||||||
|
val error = assertFailsWith<KestrelException> { db.run("INSERT INTO employees VALUES (1, 'ada', 99)") }
|
||||||
|
assertTrue(error.message!!.contains("FOREIGN KEY"))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun insertAllowsNullForeignKeyValue() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.seedDepartmentsAndEmployees()
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO employees VALUES (1, 'ada', NULL)").single())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun insertSucceedsWhenReferencedRowExists() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.seedDepartmentsAndEmployees()
|
||||||
|
assertEquals(InsertResult(1), db.run("INSERT INTO employees VALUES (1, 'ada', 1)").single())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun updateRejectsForeignKeyViolationOnChildColumn() {
|
||||||
|
val db = newDatabase()
|
||||||
|
db.seedDepartmentsAndEmployees()
|
||||||
|
db.run("INSERT INTO employees VALUES (1, 'ada', 1)")
|
||||||
|
assertFailsWith<KestrelException> { db.run("UPDATE employees SET dept_id = 99 WHERE id = 1") }
|
||||||
|
}
|
||||||
|
|
||||||
Reference in new issue
Block a user