test(sql): cover DiskCatalogTest and tempFile
This commit is contained in:
dualfroz committed
1 parent
910c0761fa
commit
65424efff3
1 file changed
+159
@@ -0,0 +1,159 @@
|
||||
package kestrel.store.sql
|
||||
|
||||
import kestrel.catalog.Column
|
||||
import kestrel.catalog.SequenceDefinition
|
||||
import kestrel.catalog.UniqueConstraint
|
||||
import kestrel.storage.ColumnType
|
||||
import kestrel.storage.Row
|
||||
import kestrel.storage.Value
|
||||
import java.io.File
|
||||
import java.nio.file.Files
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertFailsWith
|
||||
import kotlin.test.assertTrue
|
||||
|
||||
class DiskCatalogTest {
|
||||
private fun tempFile(prefix: String): File = Files.createTempFile(prefix, ".db").toFile().also { it.deleteOnExit() }
|
||||
|
||||
private val usersColumns = listOf(
|
||||
Column("id", ColumnType.Integer, notNull = true, primaryKey = true),
|
||||
Column("name", ColumnType.Text, notNull = true, primaryKey = false)
|
||||
)
|
||||
|
||||
@Test
|
||||
fun creatingTheSameTableTwiceFails() {
|
||||
val catalog = DiskCatalog.open(tempFile("kestrel-cat-data"), tempFile("kestrel-cat-wal"))
|
||||
catalog.createTable("users", usersColumns)
|
||||
assertFailsWith<DiskCatalogException> { catalog.createTable("users", usersColumns) }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun droppedTableIsNoLongerListedOrQueryable() {
|
||||
val catalog = DiskCatalog.open(tempFile("kestrel-cat-data"), tempFile("kestrel-cat-wal"))
|
||||
catalog.createTable("users", usersColumns)
|
||||
catalog.dropTable("users")
|
||||
|
||||
assertTrue(catalog.tableNames.isEmpty())
|
||||
assertFailsWith<DiskCatalogException> { catalog.table("users") }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun reopeningAFreshProcessSeesTablesAndRowsThroughNewObjectsEntirely() {
|
||||
val dataFile = tempFile("kestrel-cat-data")
|
||||
val walFile = tempFile("kestrel-cat-wal")
|
||||
|
||||
val catalog1 = DiskCatalog.open(dataFile, walFile)
|
||||
val table1 = catalog1.createTable("users", usersColumns)
|
||||
val txn1 = catalog1.beginTransaction()
|
||||
table1.insert(txn1, Row(listOf(Value.IntegerValue(1), Value.TextValue("alice"))))
|
||||
table1.insert(txn1, Row(listOf(Value.IntegerValue(2), Value.TextValue("bob"))))
|
||||
catalog1.commit(txn1)
|
||||
catalog1.close()
|
||||
|
||||
val catalog2 = DiskCatalog.open(dataFile, walFile)
|
||||
assertEquals(listOf("users"), catalog2.tableNames)
|
||||
val table2 = catalog2.table("users")
|
||||
val reader = catalog2.beginTransaction()
|
||||
assertEquals(setOf(1L, 2L), table2.scan(reader).map { (it.second.values[0] as Value.IntegerValue).value }.toSet())
|
||||
val (_, aliceRow) = table2.findByPrimaryKey(reader, Value.IntegerValue(1))!!
|
||||
assertEquals("alice", (aliceRow.values[1] as Value.TextValue).value)
|
||||
catalog2.commit(reader)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun aCommittedWriteThatNeverFlushedTheBufferPoolSurvivesAReopen() {
|
||||
val dataFile = tempFile("kestrel-cat-data")
|
||||
val walFile = tempFile("kestrel-cat-wal")
|
||||
|
||||
val catalog1 = DiskCatalog.open(dataFile, walFile)
|
||||
val table1 = catalog1.createTable("users", usersColumns)
|
||||
val txn = catalog1.beginTransaction()
|
||||
table1.insert(txn, Row(listOf(Value.IntegerValue(1), Value.TextValue("alice"))))
|
||||
catalog1.commit(txn)
|
||||
// No catalog1.close(): the dirty pages may still only live in the buffer pool.
|
||||
|
||||
val catalog2 = DiskCatalog.open(dataFile, walFile)
|
||||
val table2 = catalog2.table("users")
|
||||
val reader = catalog2.beginTransaction()
|
||||
assertEquals("alice", table2.findByPrimaryKey(reader, Value.IntegerValue(1))!!.second.values[1].let { (it as Value.TextValue).value })
|
||||
catalog2.commit(reader)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun aPrimaryKeyIndexThatSplitsSurvivesAReopenWithTheCorrectRoot() {
|
||||
val dataFile = tempFile("kestrel-cat-data")
|
||||
val walFile = tempFile("kestrel-cat-wal")
|
||||
|
||||
val catalog1 = DiskCatalog.open(dataFile, walFile)
|
||||
val table1 = catalog1.createTable("users", usersColumns)
|
||||
val txn = catalog1.beginTransaction()
|
||||
for (id in 1..300L) table1.insert(txn, Row(listOf(Value.IntegerValue(id), Value.TextValue("user$id"))))
|
||||
catalog1.commit(txn)
|
||||
catalog1.close()
|
||||
|
||||
val catalog2 = DiskCatalog.open(dataFile, walFile)
|
||||
val table2 = catalog2.table("users")
|
||||
val reader = catalog2.beginTransaction()
|
||||
assertEquals(300, table2.scan(reader).size)
|
||||
for (id in listOf(1L, 150L, 300L)) {
|
||||
assertEquals("user$id", table2.findByPrimaryKey(reader, Value.IntegerValue(id))!!.second.values[1].let { (it as Value.TextValue).value })
|
||||
}
|
||||
val range = table2.rangeByPrimaryKey(reader, Value.IntegerValue(100), true, Value.IntegerValue(105), true)
|
||||
assertEquals((100..105L).toList(), range.map { (it.second.values[0] as Value.IntegerValue).value })
|
||||
catalog2.commit(reader)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun aSecondaryIndexCreatedBeforeCloseSurvivesAReopen() {
|
||||
val dataFile = tempFile("kestrel-cat-data")
|
||||
val walFile = tempFile("kestrel-cat-wal")
|
||||
val columns = usersColumns + Column("dept", ColumnType.Text, notNull = false, primaryKey = false)
|
||||
|
||||
val catalog1 = DiskCatalog.open(dataFile, walFile)
|
||||
val table1 = catalog1.createTable("users", columns)
|
||||
catalog1.createIndex("users", "idx_dept", "dept")
|
||||
val txn = catalog1.beginTransaction()
|
||||
table1.insert(txn, Row(listOf(Value.IntegerValue(1), Value.TextValue("alice"), Value.TextValue("eng"))))
|
||||
table1.insert(txn, Row(listOf(Value.IntegerValue(2), Value.TextValue("bob"), Value.TextValue("eng"))))
|
||||
catalog1.commit(txn)
|
||||
catalog1.close()
|
||||
|
||||
val catalog2 = DiskCatalog.open(dataFile, walFile)
|
||||
val table2 = catalog2.table("users")
|
||||
val reader = catalog2.beginTransaction()
|
||||
val engineers = table2.findBySecondary(reader, "idx_dept", Value.TextValue("eng"))
|
||||
assertEquals(setOf("alice", "bob"), engineers.map { (it.second.values[1] as Value.TextValue).value }.toSet())
|
||||
catalog2.commit(reader)
|
||||
}
|
||||
|
||||
private val orderLineColumns = listOf(
|
||||
Column("order_id", ColumnType.Integer, notNull = true, primaryKey = false),
|
||||
Column("line_no", ColumnType.Integer, notNull = true, primaryKey = false),
|
||||
Column("sku", ColumnType.Text, notNull = false, primaryKey = false)
|
||||
)
|
||||
|
||||
@Test
|
||||
fun aTableCannotDeclareBothAnInlinePrimaryKeyColumnAndACompositePrimaryKey() {
|
||||
val catalog = DiskCatalog.open(tempFile("kestrel-cat-data"), tempFile("kestrel-cat-wal"))
|
||||
assertFailsWith<DiskCatalogException> {
|
||||
catalog.createTable("users", usersColumns, compositePrimaryKey = listOf("name"))
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun aCompositePrimaryKeyReferencingAnUnknownColumnIsRejected() {
|
||||
val catalog = DiskCatalog.open(tempFile("kestrel-cat-data"), tempFile("kestrel-cat-wal"))
|
||||
assertFailsWith<DiskCatalogException> {
|
||||
catalog.createTable("order_lines", orderLineColumns, compositePrimaryKey = listOf("order_id", "missing"))
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun aCompositePrimaryKeyListingTheSameColumnTwiceIsRejected() {
|
||||
val catalog = DiskCatalog.open(tempFile("kestrel-cat-data"), tempFile("kestrel-cat-wal"))
|
||||
assertFailsWith<DiskCatalogException> {
|
||||
catalog.createTable("order_lines", orderLineColumns, compositePrimaryKey = listOf("order_id", "order_id"))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user