test(sql): cover DirectoryEntryCodecTest
This commit is contained in:
dualfroz committed
1 parent
17a9f92eef
commit
910c0761fa
1 file changed
+167
@@ -0,0 +1,167 @@
|
|||||||
|
package kestrel.store.sql
|
||||||
|
|
||||||
|
import kestrel.catalog.Column
|
||||||
|
import kestrel.catalog.UniqueConstraint
|
||||||
|
import kestrel.storage.ColumnType
|
||||||
|
import kotlin.test.Test
|
||||||
|
import kotlin.test.assertEquals
|
||||||
|
import kotlin.test.assertTrue
|
||||||
|
|
||||||
|
class DirectoryEntryCodecTest {
|
||||||
|
@Test
|
||||||
|
fun roundTripsATableWithNoIndexes() {
|
||||||
|
val entry = TableDirectoryEntry(
|
||||||
|
tableName = "widgets",
|
||||||
|
columns = listOf(Column("id", ColumnType.Integer, notNull = true, primaryKey = false)),
|
||||||
|
dataHeaderPageId = 3L,
|
||||||
|
primaryKeyIndexRootPageId = null,
|
||||||
|
secondaryIndexes = emptyList()
|
||||||
|
)
|
||||||
|
assertEquals(entry, DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(entry)))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun roundTripsATableWithAPrimaryKeyAndSecondaryIndexes() {
|
||||||
|
val entry = TableDirectoryEntry(
|
||||||
|
tableName = "orders",
|
||||||
|
columns = listOf(
|
||||||
|
Column("id", ColumnType.Integer, notNull = true, primaryKey = true),
|
||||||
|
Column("customer", ColumnType.Text, notNull = false, primaryKey = false),
|
||||||
|
Column("total", ColumnType.Real, notNull = true, primaryKey = false),
|
||||||
|
Column("shipped", ColumnType.Boolean, notNull = false, primaryKey = false)
|
||||||
|
),
|
||||||
|
dataHeaderPageId = 10L,
|
||||||
|
primaryKeyIndexRootPageId = 11L,
|
||||||
|
secondaryIndexes = listOf(
|
||||||
|
SecondaryIndexEntry("idx_customer", listOf("customer"), 12L),
|
||||||
|
SecondaryIndexEntry("idx_total", listOf("total"), 13L)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
assertEquals(entry, DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(entry)))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun roundTripsDecimalColumnsWithAndWithoutExplicitPrecisionAndDateColumns() {
|
||||||
|
val entry = TableDirectoryEntry(
|
||||||
|
tableName = "priced",
|
||||||
|
columns = listOf(
|
||||||
|
Column("id", ColumnType.Integer, notNull = true, primaryKey = true),
|
||||||
|
Column("bare", ColumnType.Decimal, notNull = false, primaryKey = false),
|
||||||
|
Column("scaled", ColumnType.Decimal, notNull = false, primaryKey = false, precision = 12, scale = 4),
|
||||||
|
Column("placed", ColumnType.Date, notNull = false, primaryKey = false)
|
||||||
|
),
|
||||||
|
dataHeaderPageId = 4L,
|
||||||
|
primaryKeyIndexRootPageId = 5L,
|
||||||
|
secondaryIndexes = emptyList()
|
||||||
|
)
|
||||||
|
val decoded = DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(entry))
|
||||||
|
assertEquals(entry, decoded)
|
||||||
|
assertEquals(null, decoded.columns[1].precision)
|
||||||
|
assertEquals(12, decoded.columns[2].precision)
|
||||||
|
assertEquals(4, decoded.columns[2].scale)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun roundTripsATableWithACompositePrimaryKeyAndEmptyOnesDoNotAddAnyBytes() {
|
||||||
|
val withComposite = TableDirectoryEntry(
|
||||||
|
tableName = "order_lines",
|
||||||
|
columns = 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)
|
||||||
|
),
|
||||||
|
dataHeaderPageId = 7L,
|
||||||
|
primaryKeyIndexRootPageId = 8L,
|
||||||
|
secondaryIndexes = emptyList(),
|
||||||
|
compositePrimaryKey = listOf("order_id", "line_no")
|
||||||
|
)
|
||||||
|
val decoded = DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(withComposite))
|
||||||
|
assertEquals(withComposite, decoded)
|
||||||
|
assertEquals(listOf("order_id", "line_no"), decoded.compositePrimaryKey)
|
||||||
|
|
||||||
|
val withoutComposite = withComposite.copy(compositePrimaryKey = emptyList())
|
||||||
|
assertEquals(withoutComposite, DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(withoutComposite)))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun roundTripsAUniqueSecondaryIndexFlagAndDeclarativeUniqueConstraints() {
|
||||||
|
val entry = TableDirectoryEntry(
|
||||||
|
tableName = "users",
|
||||||
|
columns = listOf(
|
||||||
|
Column("id", ColumnType.Integer, notNull = true, primaryKey = true),
|
||||||
|
Column("email", ColumnType.Text, notNull = false, primaryKey = false),
|
||||||
|
Column("event_id", ColumnType.Integer, notNull = false, primaryKey = false),
|
||||||
|
Column("user_id", ColumnType.Integer, notNull = false, primaryKey = false)
|
||||||
|
),
|
||||||
|
dataHeaderPageId = 20L,
|
||||||
|
primaryKeyIndexRootPageId = 21L,
|
||||||
|
secondaryIndexes = listOf(SecondaryIndexEntry("idx_email", listOf("email"), 22L, unique = true)),
|
||||||
|
uniqueConstraints = listOf(
|
||||||
|
UniqueConstraint("uq_email_named", listOf("email")),
|
||||||
|
UniqueConstraint(null, listOf("event_id", "user_id"))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
val decoded = DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(entry))
|
||||||
|
assertEquals(entry, decoded)
|
||||||
|
assertTrue(decoded.secondaryIndexes.single().unique)
|
||||||
|
assertEquals(listOf("event_id", "user_id"), decoded.uniqueConstraints[1].columns)
|
||||||
|
assertEquals(null, decoded.uniqueConstraints[1].name)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun aNonUniqueSecondaryIndexAndEmptyUniqueConstraintsDoNotAddSpuriousFlagsOnDecode() {
|
||||||
|
val entry = TableDirectoryEntry(
|
||||||
|
tableName = "widgets",
|
||||||
|
columns = listOf(Column("id", ColumnType.Integer, notNull = true, primaryKey = false)),
|
||||||
|
dataHeaderPageId = 1L,
|
||||||
|
primaryKeyIndexRootPageId = null,
|
||||||
|
secondaryIndexes = listOf(SecondaryIndexEntry("idx_id", listOf("id"), 2L))
|
||||||
|
)
|
||||||
|
val decoded = DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(entry))
|
||||||
|
assertEquals(false, decoded.secondaryIndexes.single().unique)
|
||||||
|
assertTrue(decoded.uniqueConstraints.isEmpty())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun roundTripsACompositeMultiColumnSecondaryIndex() {
|
||||||
|
val entry = TableDirectoryEntry(
|
||||||
|
tableName = "order_lines",
|
||||||
|
columns = listOf(
|
||||||
|
Column("order_id", ColumnType.Integer, notNull = true, primaryKey = false),
|
||||||
|
Column("sku", ColumnType.Text, notNull = false, primaryKey = false)
|
||||||
|
),
|
||||||
|
dataHeaderPageId = 5L,
|
||||||
|
primaryKeyIndexRootPageId = null,
|
||||||
|
secondaryIndexes = listOf(SecondaryIndexEntry("idx_order_sku", listOf("order_id", "sku"), 6L, unique = true))
|
||||||
|
)
|
||||||
|
val decoded = DirectoryEntryCodec.decode(DirectoryEntryCodec.encode(entry))
|
||||||
|
assertEquals(entry, decoded)
|
||||||
|
assertEquals(listOf("order_id", "sku"), decoded.secondaryIndexes.single().columnNames)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun decodingAnUnknownColumnTypeFails() {
|
||||||
|
val entry = TableDirectoryEntry("t", listOf(Column("c", ColumnType.Integer, false, false)), 0L, null, emptyList())
|
||||||
|
val bytes = DirectoryEntryCodec.encode(entry).copyOf()
|
||||||
|
// Corrupt the type-keyword length byte to force an unrecognised keyword.
|
||||||
|
var offset = 0
|
||||||
|
// skip table name: length-prefix varint then bytes
|
||||||
|
val nameLength = bytes[offset].toInt() and 0x7F
|
||||||
|
offset += 1 + nameLength
|
||||||
|
offset += 1 // column count varint (small)
|
||||||
|
val colNameLength = bytes[offset].toInt() and 0x7F
|
||||||
|
offset += 1 + colNameLength
|
||||||
|
val typeLength = bytes[offset].toInt() and 0x7F
|
||||||
|
for (i in 0 until typeLength) bytes[offset + 1 + i] = 'Z'.code.toByte()
|
||||||
|
assertThrowsCodecException { DirectoryEntryCodec.decode(bytes) }
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun assertThrowsCodecException(block: () -> Unit) {
|
||||||
|
try {
|
||||||
|
block()
|
||||||
|
throw AssertionError("expected decode to fail")
|
||||||
|
} catch (e: kestrel.store.codec.CodecException) {
|
||||||
|
// expected
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in new issue
Block a user