MicroUtils/repos/exposed/src/jvmTest/kotlin/ExposedCRUDRepoTests.kt

package full

import CommonCRUDRepoTests
import com.benasher44.uuid.uuid4
import dev.inmo.micro_utils.repos.CRUDRepo
import dev.inmo.micro_utils.repos.create
import dev.inmo.micro_utils.repos.deleteById
import dev.inmo.micro_utils.repos.exposed.AbstractExposedCRUDRepo
import dev.inmo.micro_utils.repos.exposed.initTable
import korlibs.time.seconds
import kotlinx.coroutines.test.TestResult
import kotlinx.coroutines.test.runTest
import org.jetbrains.exposed.sql.Database
import org.jetbrains.exposed.sql.ISqlExpressionBuilder
import org.jetbrains.exposed.sql.Op
import org.jetbrains.exposed.sql.ResultRow
import org.jetbrains.exposed.sql.statements.InsertStatement
import org.jetbrains.exposed.sql.statements.UpdateBuilder
import java.io.File
import javax.xml.crypto.Data
import kotlin.test.*

class ExposedCRUDRepoTests : CommonCRUDRepoTests() {
    class Repo(override val database: Database) : AbstractExposedCRUDRepo<CommonCRUDRepoTests.Registered, String, CommonCRUDRepoTests.New>() {
        val idColumn = text("_id")
        val dataColumn = text("data")

        override val primaryKey: PrimaryKey = PrimaryKey(idColumn)

        override val ResultRow.asId: String
            get() = get(idColumn)
        override val ResultRow.asObject: CommonCRUDRepoTests.Registered
            get() = CommonCRUDRepoTests.Registered(
                asId,
                get(dataColumn)
            )
        override val selectById: ISqlExpressionBuilder.(String) -> Op<Boolean> = { idColumn.eq(it) }

        init {
            initTable()
        }

        override fun update(id: String?, value: CommonCRUDRepoTests.New, it: UpdateBuilder<Int>) {
            it[idColumn] = id ?: uuid4().toString()
            it[dataColumn] = value.data
        }

        override fun InsertStatement<Number>.asObject(value: CommonCRUDRepoTests.New): CommonCRUDRepoTests.Registered {
            return CommonCRUDRepoTests.Registered(
                get(idColumn),
                get(dataColumn)
            )
        }
    }
    val filename = filename()
    var database: Database? = null
    override val repoCreator: suspend () -> CRUDRepo<Registered, String, New> = { Repo(database!!) }
    @BeforeTest
    fun beforeTest() {
        database = createDatabase(filename)
    }
    @AfterTest
    fun afterTest() {
        database = null
        File(filename).delete()
    }

    @Test
    override fun creatingWorksProperly(): TestResult {
        return super.creatingWorksProperly()
    }

    @Test
    override fun removingWorksProperly(): TestResult {
        return super.removingWorksProperly()
    }
}