test: cover rowsFrameShorthandIsWrittenBackOutInCanonicalBetweenForm
This commit is contained in:
dualfroz committed
1 parent
b5f8b44728
commit
310a2f10dc
1 file changed
+159
@@ -802,3 +802,162 @@ class SqlWriterTest {
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rowsFrameShorthandIsWrittenBackOutInCanonicalBetweenForm() {
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, salary INTEGER)")
|
||||
run("INSERT INTO employees VALUES (1, 400), (2, 300)")
|
||||
val statement = parseOne(
|
||||
"CREATE VIEW shorthand AS SELECT id, SUM(salary) OVER (ORDER BY id ROWS 1 PRECEDING) AS s FROM employees"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("ROWS BETWEEN 1 PRECEDING AND CURRENT ROW"), rendered)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun unboundedFrameBoundsRoundTripThroughSqlWriter() {
|
||||
run("CREATE TABLE employees (id INTEGER PRIMARY KEY, salary INTEGER)")
|
||||
run("INSERT INTO employees VALUES (1, 400), (2, 300)")
|
||||
val statement = parseOne(
|
||||
"CREATE VIEW wholePartition AS SELECT id, " +
|
||||
"SUM(salary) OVER (ORDER BY id ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING) AS s FROM employees"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING"), rendered)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun createViewOverAnExplicitRangeFrameRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE sales (id INTEGER PRIMARY KEY, amount INTEGER)")
|
||||
run("INSERT INTO sales VALUES (1, 10), (2, 20), (3, 20), (4, 30), (5, 50)")
|
||||
val statement = parseOne(
|
||||
"CREATE VIEW valueTotals AS SELECT id, " +
|
||||
"SUM(amount) OVER (ORDER BY amount RANGE BETWEEN 10 PRECEDING AND 10 FOLLOWING) AS s FROM sales"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("OVER (ORDER BY amount RANGE BETWEEN 10 PRECEDING AND 10 FOLLOWING)"), rendered)
|
||||
run(rendered)
|
||||
val result = select("SELECT id, s FROM valueTotals ORDER BY id ASC")
|
||||
assertEquals(
|
||||
listOf(50L, 80L, 80L, 70L, 50L),
|
||||
result.rows.map { (it[1] as Value.IntegerValue).value }
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun createViewOverAnExcludeCurrentRowFrameRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE sales (id INTEGER PRIMARY KEY, amount INTEGER)")
|
||||
run("INSERT INTO sales VALUES (1, 10), (2, 20), (3, 20), (4, 30), (5, 50)")
|
||||
val statement = parseOne(
|
||||
"CREATE VIEW excludingSelf AS SELECT id, SUM(amount) OVER (" +
|
||||
"ORDER BY amount ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING EXCLUDE CURRENT ROW" +
|
||||
") AS s FROM sales"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(
|
||||
rendered.contains("ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING EXCLUDE CURRENT ROW"),
|
||||
rendered
|
||||
)
|
||||
run(rendered)
|
||||
val result = select("SELECT id, s FROM excludingSelf ORDER BY id ASC")
|
||||
assertEquals(
|
||||
listOf(120L, 110L, 110L, 100L, 80L),
|
||||
result.rows.map { (it[1] as Value.IntegerValue).value }
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun createViewOverAnExplicitGroupsFrameRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE sales (id INTEGER PRIMARY KEY, amount INTEGER)")
|
||||
run("INSERT INTO sales VALUES (1, 10), (2, 20), (3, 20), (4, 30), (5, 50)")
|
||||
val statement = parseOne(
|
||||
"CREATE VIEW groupTotals AS SELECT id, " +
|
||||
"SUM(amount) OVER (ORDER BY amount GROUPS BETWEEN 1 PRECEDING AND 1 FOLLOWING) AS s FROM sales"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("OVER (ORDER BY amount GROUPS BETWEEN 1 PRECEDING AND 1 FOLLOWING)"), rendered)
|
||||
run(rendered)
|
||||
val result = select("SELECT id, s FROM groupTotals ORDER BY id ASC")
|
||||
assertEquals(
|
||||
listOf(50L, 80L, 80L, 120L, 80L),
|
||||
result.rows.map { (it[1] as Value.IntegerValue).value }
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun excludeNoOthersIsOmittedFromRenderedSqlSinceItIsTheDefault() {
|
||||
run("CREATE TABLE sales (id INTEGER PRIMARY KEY, amount INTEGER)")
|
||||
run("INSERT INTO sales VALUES (1, 10), (2, 20)")
|
||||
val statement = parseOne(
|
||||
"CREATE VIEW totals AS SELECT id, SUM(amount) OVER (" +
|
||||
"ORDER BY amount ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING EXCLUDE NO OTHERS" +
|
||||
") AS s FROM sales"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(!rendered.contains("EXCLUDE"), rendered)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun insertOnConflictDoNothingRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE accounts (id INTEGER PRIMARY KEY, balance INTEGER)")
|
||||
run("INSERT INTO accounts VALUES (1, 100)")
|
||||
val statement = parseOne("INSERT INTO accounts VALUES (1, 999) ON CONFLICT (id) DO NOTHING")
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("ON CONFLICT (id) DO NOTHING"), rendered)
|
||||
run(rendered)
|
||||
assertEquals(listOf(listOf(Value.IntegerValue(100))), select("SELECT balance FROM accounts WHERE id = 1").rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun insertOnConflictDoUpdateWithExcludedAndWhereRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE accounts (id INTEGER PRIMARY KEY, balance INTEGER)")
|
||||
run("INSERT INTO accounts VALUES (1, 100)")
|
||||
val statement = parseOne(
|
||||
"INSERT INTO accounts VALUES (1, 50) ON CONFLICT (id) DO UPDATE SET balance = accounts.balance + excluded.balance WHERE excluded.balance > 0"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("ON CONFLICT (id) DO UPDATE SET balance = (accounts.balance + excluded.balance) WHERE (excluded.balance > 0)"), rendered)
|
||||
run(rendered)
|
||||
assertEquals(listOf(listOf(Value.IntegerValue(150))), select("SELECT balance FROM accounts WHERE id = 1").rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun createTriggerRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE orders (id INTEGER PRIMARY KEY, amount INTEGER)")
|
||||
run("CREATE TABLE audit (order_id INTEGER)")
|
||||
val statement = parseOne(
|
||||
"CREATE TRIGGER log_insert AFTER INSERT ON orders FOR EACH ROW WHEN NEW.amount > 0 " +
|
||||
"BEGIN INSERT INTO audit VALUES (NEW.id); END"
|
||||
)
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("CREATE TRIGGER log_insert AFTER INSERT ON orders FOR EACH ROW WHEN"), rendered)
|
||||
assertTrue(rendered.contains("INSERT INTO audit VALUES (NEW.id)"), rendered)
|
||||
run(rendered)
|
||||
run("INSERT INTO orders VALUES (1, 50)")
|
||||
assertEquals(listOf(listOf(Value.IntegerValue(1))), select("SELECT order_id FROM audit").rows)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun dropTriggerRoundTripsThroughSqlWriter() {
|
||||
run("CREATE TABLE t (id INTEGER)")
|
||||
run("CREATE TRIGGER log AFTER INSERT ON t FOR EACH ROW BEGIN DELETE FROM t WHERE 1 = 0; END")
|
||||
val rendered = SqlWriter.write(parseOne("DROP TRIGGER log"))
|
||||
assertEquals("DROP TRIGGER log;", rendered)
|
||||
run(rendered)
|
||||
assertFailsWith<KestrelException> { run("DROP TRIGGER log") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun createSequenceRoundTripsThroughSqlWriter() {
|
||||
val statement = parseOne("CREATE SEQUENCE seq START WITH 5 INCREMENT BY 2 MAXVALUE 100 CYCLE")
|
||||
val rendered = SqlWriter.write(statement)
|
||||
assertTrue(rendered.contains("CREATE SEQUENCE seq"), rendered)
|
||||
assertTrue(rendered.contains("INCREMENT BY 2"), rendered)
|
||||
assertTrue(rendered.contains("MAXVALUE 100"), rendered)
|
||||
assertTrue(rendered.contains("START WITH 5"), rendered)
|
||||
assertTrue(rendered.contains("CYCLE"), rendered)
|
||||
run(rendered)
|
||||
run("CREATE TABLE t (id INTEGER)")
|
||||
run("INSERT INTO t VALUES (NEXTVAL('seq'))")
|
||||
assertEquals(listOf(listOf(Value.IntegerValue(5))), select("SELECT id FROM t").rows)
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user