test: cover compositeForeignKeyAcceptsAFullMatch
This commit is contained in:
dualfroz committed
1 parent
9f5328b55e
commit
eb56dc4b87
1 file changed
+158
@@ -156,3 +156,161 @@ class CompositeKeyTest {
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyAcceptsAFullMatch() {
|
||||
seedOrdersAndLineItems()
|
||||
val result = exec("INSERT INTO line_items VALUES ('eu', 1, 'SKU1', 3)")
|
||||
assertEquals(InsertResult(1), result)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyMatchSimpleSkipsCheckWhenAnyLocalColumnIsNull() {
|
||||
seedOrdersAndLineItems()
|
||||
val result = exec("INSERT INTO line_items VALUES ('eu', NULL, 'SKU1', 3)")
|
||||
assertEquals(InsertResult(1), result)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyOnDeleteCascadeRemovesMatchingChildren() {
|
||||
seedOrdersAndLineItems()
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1', 3), ('us', 1, 'SKU2', 4)")
|
||||
run("DELETE FROM orders WHERE region = 'eu'")
|
||||
val remaining = select("SELECT sku FROM line_items")
|
||||
assertEquals(listOf(listOf(Value.TextValue("SKU2"))), remaining.rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyOnUpdateCascadePropagatesBothColumnsTogether() {
|
||||
seedOrdersAndLineItems()
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1', 3)")
|
||||
run("UPDATE orders SET order_no = 7 WHERE region = 'eu'")
|
||||
val updated = select("SELECT region, order_no FROM line_items WHERE sku = 'SKU1'")
|
||||
assertEquals(listOf(listOf(Value.TextValue("eu"), int(7))), updated.rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyOnDeleteRestrictBlocksByDefaultWithoutCascade() {
|
||||
run("CREATE TABLE orders (region TEXT, order_no INTEGER, PRIMARY KEY (region, order_no))")
|
||||
run(
|
||||
"CREATE TABLE line_items (region TEXT, order_no INTEGER, sku TEXT, " +
|
||||
"FOREIGN KEY (region, order_no) REFERENCES orders (region, order_no))"
|
||||
)
|
||||
run("INSERT INTO orders VALUES ('eu', 1)")
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1')")
|
||||
assertFailsWith<KestrelException> { run("DELETE FROM orders WHERE region = 'eu'") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyOnDeleteSetNullNullsEveryLocalColumn() {
|
||||
run("CREATE TABLE orders (region TEXT, order_no INTEGER, PRIMARY KEY (region, order_no))")
|
||||
run(
|
||||
"CREATE TABLE line_items (region TEXT, order_no INTEGER, sku TEXT, " +
|
||||
"FOREIGN KEY (region, order_no) REFERENCES orders (region, order_no) ON DELETE SET NULL)"
|
||||
)
|
||||
run("INSERT INTO orders VALUES ('eu', 1)")
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1')")
|
||||
run("DELETE FROM orders WHERE region = 'eu'")
|
||||
val row = select("SELECT region, order_no FROM line_items WHERE sku = 'SKU1'").rows.single()
|
||||
assertEquals(listOf(Value.NullValue, Value.NullValue), row)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackAfterCompositeCascadeDeleteRestoresBothTables() {
|
||||
seedOrdersAndLineItems()
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1', 3)")
|
||||
run("BEGIN")
|
||||
run("DELETE FROM orders WHERE region = 'eu'")
|
||||
run("ROLLBACK")
|
||||
assertEquals(2, select("SELECT * FROM orders").rows.size)
|
||||
assertEquals(1, select("SELECT * FROM line_items").rows.size)
|
||||
}
|
||||
|
||||
// ---- composite FOREIGN KEY referencing a composite UNIQUE (not the PRIMARY KEY) ----
|
||||
|
||||
@Test
|
||||
fun compositeForeignKeyMayReferenceACompositeUniqueConstraintInsteadOfThePrimaryKey() {
|
||||
run(
|
||||
"CREATE TABLE products (id INTEGER PRIMARY KEY, brand TEXT, model TEXT, UNIQUE (brand, model))"
|
||||
)
|
||||
run(
|
||||
"CREATE TABLE reviews (id INTEGER PRIMARY KEY, brand TEXT, model TEXT, stars INTEGER, " +
|
||||
"FOREIGN KEY (brand, model) REFERENCES products (brand, model))"
|
||||
)
|
||||
run("INSERT INTO products VALUES (1, 'acme', 'x100')")
|
||||
val result = exec("INSERT INTO reviews VALUES (1, 'acme', 'x100', 5)")
|
||||
assertEquals(InsertResult(1), result)
|
||||
assertFailsWith<KestrelException> {
|
||||
run("INSERT INTO reviews VALUES (2, 'acme', 'missing', 4)")
|
||||
}
|
||||
}
|
||||
|
||||
// ---- self-referencing composite FOREIGN KEY ----
|
||||
|
||||
@Test
|
||||
fun selfReferencingCompositeForeignKeySeedableInASingleMultiRowInsert() {
|
||||
run(
|
||||
"CREATE TABLE edges (" +
|
||||
"tenant TEXT, node INTEGER, parent_tenant TEXT, parent_node INTEGER, " +
|
||||
"PRIMARY KEY (tenant, node), " +
|
||||
"FOREIGN KEY (parent_tenant, parent_node) REFERENCES edges (tenant, node))"
|
||||
)
|
||||
// The root's parent columns are NULL (no parent); child references the root, all seeded
|
||||
// together in one INSERT so the child must see its not-yet-committed sibling.
|
||||
run(
|
||||
"INSERT INTO edges VALUES " +
|
||||
"('t1', 1, NULL, NULL), ('t1', 2, 't1', 1), ('t1', 3, 't1', 2)"
|
||||
)
|
||||
assertEquals(3, select("SELECT * FROM edges").rows.size)
|
||||
}
|
||||
|
||||
// ---- SqlWriter round-trip ----
|
||||
|
||||
@Test
|
||||
fun createTableWithCompositePrimaryKeyRoundTripsThroughSqlWriter() {
|
||||
val statement = parseOne(
|
||||
"CREATE TABLE line_items (order_id INTEGER, sku TEXT, qty INTEGER, PRIMARY KEY (order_id, sku))"
|
||||
)
|
||||
run(SqlWriter.write(statement))
|
||||
run("INSERT INTO line_items VALUES (1, 'A', 3)")
|
||||
assertFailsWith<KestrelException> { run("INSERT INTO line_items VALUES (1, 'A', 9)") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun createTableWithCompositeForeignKeyRoundTripsThroughSqlWriter() {
|
||||
run(SqlWriter.write(parseOne("CREATE TABLE orders (region TEXT, order_no INTEGER, PRIMARY KEY (region, order_no))")))
|
||||
run(
|
||||
SqlWriter.write(
|
||||
parseOne(
|
||||
"CREATE TABLE line_items (region TEXT, order_no INTEGER, sku TEXT, " +
|
||||
"FOREIGN KEY (region, order_no) REFERENCES orders (region, order_no) ON DELETE CASCADE)"
|
||||
)
|
||||
)
|
||||
)
|
||||
run("INSERT INTO orders VALUES ('eu', 1)")
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1')")
|
||||
run("DELETE FROM orders WHERE region = 'eu'")
|
||||
assertTrue(select("SELECT * FROM line_items").rows.isEmpty())
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dumpAndReplayPreservesCompositePrimaryKeyAndForeignKey() {
|
||||
seedOrdersAndLineItems()
|
||||
run("INSERT INTO line_items VALUES ('eu', 1, 'SKU1', 3)")
|
||||
val dump = SqlWriter.dump(catalog)
|
||||
|
||||
val freshCatalog = Catalog()
|
||||
val freshExecutor = Executor(freshCatalog)
|
||||
Parser(Lexer(dump).tokenize()).parseProgram().forEach { freshExecutor.execute(it) }
|
||||
|
||||
assertFailsWith<KestrelException> {
|
||||
Parser(Lexer("INSERT INTO orders VALUES ('eu', 1)").tokenize()).parseProgram()
|
||||
.forEach { freshExecutor.execute(it) }
|
||||
}
|
||||
Parser(Lexer("DELETE FROM orders WHERE region = 'eu'").tokenize()).parseProgram()
|
||||
.forEach { freshExecutor.execute(it) }
|
||||
val remaining = freshExecutor.execute(
|
||||
Parser(Lexer("SELECT * FROM line_items").tokenize()).parseProgram().single()
|
||||
) as SelectResult
|
||||
assertTrue(remaining.rows.isEmpty())
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user