test: cover rollbackUndoesCreateSequence and rollbackUndoesDropSequence

src/test/kotlin/kestrel/SequenceTest.kt

@@ -160,3 +160,145 @@ class SequenceTest {
         assertEquals(listOf(listOf(int(2))), select("SELECT id FROM t").rows)
     }
 
+    @Test
+    fun rollbackUndoesCreateSequence() {
+        run("BEGIN")
+        run("CREATE SEQUENCE seq")
+        run("ROLLBACK")
+        assertFailsWith<KestrelException> { run("DROP SEQUENCE seq") }
+    }
+
+    @Test
+    fun rollbackUndoesDropSequence() {
+        run("CREATE SEQUENCE seq")
+        run("BEGIN")
+        run("DROP SEQUENCE seq")
+        run("ROLLBACK")
+        run("CREATE TABLE t (id INTEGER)")
+        run("INSERT INTO t VALUES (NEXTVAL('seq'))")
+        assertEquals(listOf(listOf(int(1))), select("SELECT id FROM t").rows)
+    }
+
+    @Test
+    fun rollbackUndoesAlterSequenceButNotTheNextvalAdvanceMadeBeforeIt() {
+        run("CREATE SEQUENCE seq")
+        run("CREATE TABLE t (id INTEGER)")
+        run("INSERT INTO t VALUES (NEXTVAL('seq'))")
+        run("BEGIN")
+        run("ALTER SEQUENCE seq INCREMENT BY 100")
+        run("ROLLBACK")
+        run("INSERT INTO t VALUES (NEXTVAL('seq'))")
+        assertEquals(listOf(listOf(int(1)), listOf(int(2))), select("SELECT id FROM t ORDER BY id").rows)
+    }
+
+    @Test
+    fun nextvalAsColumnDefaultActsLikeAnAutoIncrement() {
+        run("CREATE SEQUENCE order_ids")
+        run("CREATE TABLE orders (id INTEGER PRIMARY KEY DEFAULT NEXTVAL('order_ids'), item TEXT)")
+        run("INSERT INTO orders (item) VALUES ('a'), ('b'), ('c')")
+        assertEquals(
+            listOf(listOf(int(1), Value.TextValue("a")), listOf(int(2), Value.TextValue("b")), listOf(int(3), Value.TextValue("c"))),
+            select("SELECT id, item FROM orders ORDER BY id").rows
+        )
+    }
+
+    @Test
+    fun createTableWithNextvalDefaultRequiresTheSequenceToAlreadyExist() {
+        assertFailsWith<KestrelException> {
+            run("CREATE TABLE orders (id INTEGER DEFAULT NEXTVAL('nope'))")
+        }
+    }
+
+    @Test
+    fun createTableDoesNotConsumeASequenceValueJustToValidateTheDefault() {
+        run("CREATE SEQUENCE seq")
+        run("CREATE TABLE orders (id INTEGER DEFAULT NEXTVAL('seq'))")
+        run("INSERT INTO orders (id) VALUES (DEFAULT)")
+        // If CREATE TABLE's own validation had already called NEXTVAL once, this would be 2.
+        assertEquals(listOf(listOf(int(1))), select("SELECT id FROM orders").rows)
+    }
+
+    @Test
+    fun nextvalDefaultOnANonIntegerColumnFails() {
+        run("CREATE SEQUENCE seq")
+        assertFailsWith<KestrelException> {
+            run("CREATE TABLE orders (id TEXT DEFAULT NEXTVAL('seq'))")
+        }
+    }
+
+    @Test
+    fun currvalAsColumnDefaultIsRejected() {
+        run("CREATE SEQUENCE seq")
+        assertFailsWith<KestrelException> {
+            run("CREATE TABLE orders (id INTEGER DEFAULT CURRVAL('seq'))")
+        }
+    }
+
+    @Test
+    fun addColumnWithNextvalDefaultIsRejected() {
+        run("CREATE SEQUENCE seq")
+        run("CREATE TABLE t (id INTEGER)")
+        assertFailsWith<KestrelException> {
+            run("ALTER TABLE t ADD COLUMN seq_id INTEGER DEFAULT NEXTVAL('seq')")
+        }
+    }
+
+    @Test
+    fun nextvalIsUsableInUpdateSet() {
+        run("CREATE SEQUENCE seq")
+        run("CREATE TABLE t (id INTEGER)")
+        run("INSERT INTO t VALUES (NULL)")
+        run("UPDATE t SET id = NEXTVAL('seq')")
+        assertEquals(listOf(listOf(int(1))), select("SELECT id FROM t").rows)
+    }
+
+    @Test
+    fun nextvalInAWhereClauseIsRejected() {
+        run("CREATE SEQUENCE seq")
+        run("CREATE TABLE t (id INTEGER)")
+        run("INSERT INTO t VALUES (1)")
+        assertFailsWith<KestrelException> { run("UPDATE t SET id = 2 WHERE id = NEXTVAL('seq')") }
+    }
+
+    @Test
+    fun nextvalInACheckConstraintIsRejectedAtCreateTableTime() {
+        run("CREATE SEQUENCE seq")
+        assertFailsWith<KestrelException> {
+            run("CREATE TABLE t (id INTEGER CHECK (id = NEXTVAL('seq')))")
+        }
+    }
+
+    @Test
+    fun nextvalInATriggerWhenClauseIsRejected() {
+        run("CREATE SEQUENCE seq")
+        run("CREATE TABLE t (id INTEGER)")
+        assertFailsWith<KestrelException> {
+            run("CREATE TRIGGER trg BEFORE INSERT ON t FOR EACH ROW WHEN (NEXTVAL('seq') = 1) BEGIN DELETE FROM t; END")
+        }
+    }
+
+    @Test
+    fun createSequenceMinvalueExceedingMaxvalueFails() {
+        assertFailsWith<KestrelException> { run("CREATE SEQUENCE seq MINVALUE 10 MAXVALUE 5") }
+    }
+
+    @Test
+    fun createSequenceIncrementByZeroFails() {
+        assertFailsWith<KestrelException> { run("CREATE SEQUENCE seq INCREMENT BY 0") }
+    }
+
+    @Test
+    fun createSequenceStartOutsideBoundsFails() {
+        assertFailsWith<KestrelException> { run("CREATE SEQUENCE seq MINVALUE 10 MAXVALUE 20 START WITH 5") }
+    }
+
+    @Test
+    fun noMinvalueAndNoMaxvalueOnAlterRecomputeTheDefaultBound() {
+        run("CREATE SEQUENCE seq MINVALUE 1 MAXVALUE 3")
+        run("ALTER SEQUENCE seq NO MAXVALUE")
+        run("CREATE TABLE t (id INTEGER)")
+        // Was capped at 3 before the ALTER; NO MAXVALUE lifts it back to the ascending default.
+        run("INSERT INTO t VALUES (NEXTVAL('seq')), (NEXTVAL('seq')), (NEXTVAL('seq')), (NEXTVAL('seq'))")
+        assertEquals(4, select("SELECT id FROM t").rows.size)
+    }
+}