commonTest.com.squareup.sqldelight.logs.SqlDriverMigrationCallbackTest.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of runtime Show documentation
Multiplatform runtime library to support generated code
The newest version!
package com.squareup.sqldelight.logs

import app.cash.sqldelight.db.AfterVersion
import app.cash.sqldelight.db.QueryResult
import app.cash.sqldelight.db.SqlDriver
import app.cash.sqldelight.db.SqlSchema
import app.cash.sqldelight.db.migrateWithCallbacks
import kotlin.test.Test
import kotlin.test.assertFalse
import kotlin.test.assertTrue

class SqlDriverMigrationCallbackTest {

  @Test fun migrationCallbackInvokedOnCorrectVersion() {
    val schema = fakeSchema()

    var callbackInvoked = false

    schema.migrateWithCallbacks(
      driver = FakeSqlDriver(),
      oldVersion = 1,
      newVersion = 2,
      AfterVersion(1) { callbackInvoked = true },
    )

    assertTrue(callbackInvoked)
  }

  @Test fun migrationCallbacks() {
    val schema = fakeSchema()

    var callback1Invoked = false
    var callback2Invoked = false
    var callback3Invoked = false

    schema.migrateWithCallbacks(
      driver = FakeSqlDriver(),
      oldVersion = 0,
      newVersion = 4,
      AfterVersion(1) { callback1Invoked = true },
      AfterVersion(2) { callback2Invoked = true },
      AfterVersion(3) { callback3Invoked = true },
    )

    assertTrue(callback1Invoked)
    assertTrue(callback2Invoked)
    assertTrue(callback3Invoked)
  }

  @Test fun migrationCallbackLessThanOldVersionNotCalled() {
    val schema = fakeSchema()

    var callback1Invoked = false
    var callback2Invoked = false
    var callback3Invoked = false

    schema.migrateWithCallbacks(
      driver = FakeSqlDriver(),
      oldVersion = 2,
      newVersion = 4,
      AfterVersion(1) { callback1Invoked = true },
      AfterVersion(2) { callback2Invoked = true },
      AfterVersion(3) { callback3Invoked = true },
    )

    assertFalse(callback1Invoked)
    assertTrue(callback2Invoked)
    assertTrue(callback3Invoked)
  }

  @Test fun migrationCallbackGreaterThanNewVersionNotCalled() {
    val schema = fakeSchema()

    var callback1Invoked = false
    var callback2Invoked = false
    var callback3Invoked = false
    var callback4Invoked = false

    schema.migrateWithCallbacks(
      driver = FakeSqlDriver(),
      oldVersion = 0,
      newVersion = 4,
      AfterVersion(1) { callback1Invoked = true },
      AfterVersion(2) { callback2Invoked = true },
      AfterVersion(3) { callback3Invoked = true },
      AfterVersion(4) { callback4Invoked = true },
    )

    assertTrue(callback1Invoked)
    assertTrue(callback2Invoked)
    assertTrue(callback3Invoked)
    assertFalse(callback4Invoked)
  }

  private fun fakeSchema() = object : SqlSchema {
    override val version: Int = 1
    override fun create(driver: SqlDriver) = QueryResult.Unit
    override fun migrate(driver: SqlDriver, oldVersion: Int, newVersion: Int) = QueryResult.Unit
  }
}