test: cover SavepointTest and run
This commit is contained in:
dualfroz committed
1 parent
a85575538e
commit
9e44181f8f
1 file changed
+209
@@ -0,0 +1,209 @@
|
||||
package kestrel
|
||||
|
||||
import kestrel.catalog.Catalog
|
||||
import kestrel.engine.Executor
|
||||
import kestrel.engine.SelectResult
|
||||
import kestrel.engine.StatementResult
|
||||
import kestrel.sql.lexer.Lexer
|
||||
import kestrel.sql.parser.Parser
|
||||
import kestrel.storage.Value
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertFailsWith
|
||||
|
||||
class SavepointTest {
|
||||
private val catalog = Catalog()
|
||||
private val executor = Executor(catalog)
|
||||
|
||||
private fun run(sql: String): List<StatementResult> =
|
||||
Parser(Lexer(sql).tokenize()).parseProgram().map { executor.execute(it) }
|
||||
|
||||
private fun select(sql: String): SelectResult = run(sql).single() as SelectResult
|
||||
|
||||
private fun rowCount(sql: String): Int = select(sql).rows.size
|
||||
|
||||
private fun seed() {
|
||||
run("CREATE TABLE accounts (id INTEGER PRIMARY KEY, balance INTEGER)")
|
||||
run("INSERT INTO accounts VALUES (1, 100), (2, 200)")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackToSavepointUndoesOnlyTheInsertsAfterIt() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("SAVEPOINT beforeFour")
|
||||
run("INSERT INTO accounts VALUES (4, 400)")
|
||||
assertEquals(4, rowCount("SELECT * FROM accounts"))
|
||||
run("ROLLBACK TO SAVEPOINT beforeFour")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
run("COMMIT")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackToSavepointUndoesUpdatesAndDeletes() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("UPDATE accounts SET balance = 0 WHERE id = 1")
|
||||
run("DELETE FROM accounts WHERE id = 2")
|
||||
assertEquals(Value.IntegerValue(0), select("SELECT balance FROM accounts WHERE id = 1").rows[0][0])
|
||||
run("ROLLBACK TO SAVEPOINT s1")
|
||||
assertEquals(Value.IntegerValue(100), select("SELECT balance FROM accounts WHERE id = 1").rows[0][0])
|
||||
assertEquals(2, rowCount("SELECT * FROM accounts"))
|
||||
run("ROLLBACK")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackToSavepointUndoesCreateTable() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("CREATE TABLE scratch (a INTEGER)")
|
||||
run("ROLLBACK TO SAVEPOINT s1")
|
||||
assertFailsWith<KestrelException> { run("SELECT * FROM scratch") }
|
||||
run("ROLLBACK")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun theSavepointItselfStaysUsableAfterRollingBackToIt() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("ROLLBACK TO SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (4, 400)")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
run("ROLLBACK TO SAVEPOINT s1")
|
||||
assertEquals(2, rowCount("SELECT * FROM accounts"))
|
||||
run("COMMIT")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun releaseSavepointForgetsItWithoutUndoingAnything() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("RELEASE SAVEPOINT s1")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
assertFailsWith<KestrelException> { run("ROLLBACK TO SAVEPOINT s1") }
|
||||
run("COMMIT")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun releaseSavepointAlsoForgetsLaterNestedSavepoints() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT spOuter")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("SAVEPOINT spInner")
|
||||
run("INSERT INTO accounts VALUES (4, 400)")
|
||||
run("RELEASE SAVEPOINT spOuter")
|
||||
assertFailsWith<KestrelException> { run("ROLLBACK TO SAVEPOINT spInner") }
|
||||
assertEquals(4, rowCount("SELECT * FROM accounts"))
|
||||
run("COMMIT")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun reusingASavepointNameShadowsTheOlderOneUntilItIsConsumed() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("SAVEPOINT s")
|
||||
run("INSERT INTO accounts VALUES (4, 400)")
|
||||
run("ROLLBACK TO SAVEPOINT s")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
run("ROLLBACK TO SAVEPOINT s")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
run("COMMIT")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackRestoresIndexStateEvenAcrossASavepoint() {
|
||||
seed()
|
||||
run("CREATE INDEX idxBalance ON accounts (balance)")
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (3, 100)")
|
||||
assertEquals(2, rowCount("SELECT * FROM accounts WHERE balance = 100"))
|
||||
run("ROLLBACK TO SAVEPOINT s1")
|
||||
assertEquals(1, rowCount("SELECT * FROM accounts WHERE balance = 100"))
|
||||
run("ROLLBACK")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun savepointOutsideATransactionIsRejected() {
|
||||
seed()
|
||||
assertFailsWith<KestrelException> { run("SAVEPOINT s1") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun releaseSavepointOutsideATransactionIsRejected() {
|
||||
seed()
|
||||
assertFailsWith<KestrelException> { run("RELEASE SAVEPOINT s1") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackToSavepointOutsideATransactionIsRejected() {
|
||||
seed()
|
||||
assertFailsWith<KestrelException> { run("ROLLBACK TO SAVEPOINT s1") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun rollbackToAnUnknownSavepointNameIsRejected() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
assertFailsWith<KestrelException> { run("ROLLBACK TO SAVEPOINT nope") }
|
||||
run("ROLLBACK")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun releaseAnUnknownSavepointNameIsRejected() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
assertFailsWith<KestrelException> { run("RELEASE SAVEPOINT nope") }
|
||||
run("ROLLBACK")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun fullRollbackClearsEveryOutstandingSavepoint() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("ROLLBACK")
|
||||
run("BEGIN")
|
||||
assertFailsWith<KestrelException> { run("ROLLBACK TO SAVEPOINT s1") }
|
||||
run("ROLLBACK")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun commitClearsEveryOutstandingSavepoint() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("COMMIT")
|
||||
run("BEGIN")
|
||||
assertFailsWith<KestrelException> { run("ROLLBACK TO SAVEPOINT s1") }
|
||||
run("ROLLBACK")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun releaseKeywordAloneWithoutSavepointKeywordIsAccepted() {
|
||||
seed()
|
||||
run("BEGIN")
|
||||
run("SAVEPOINT s1")
|
||||
run("INSERT INTO accounts VALUES (3, 300)")
|
||||
run("RELEASE s1")
|
||||
assertEquals(3, rowCount("SELECT * FROM accounts"))
|
||||
run("COMMIT")
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user