migratedb.v1.integrationtest.util.base.IntegrationTest.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of migratedb-integration-tests Show documentation
The newest version!
/*
 * Copyright 2022-2024 The MigrateDB contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package migratedb.v1.integrationtest.util.base

import migratedb.v1.integrationtest.database.DbSystem
import migratedb.v1.integrationtest.util.container.SharedResources
import migratedb.v1.integrationtest.util.container.SharedResources.Companion.resources
import migratedb.v1.integrationtest.util.dsl.Dsl
import migratedb.v1.testing.util.base.AbstractTest
import org.junit.jupiter.api.AfterEach
import org.junit.jupiter.api.BeforeEach
import org.junit.jupiter.api.TestInfo
import org.junit.jupiter.api.Timeout
import org.junit.jupiter.api.extension.BeforeAllCallback
import org.junit.jupiter.api.extension.ExtendWith
import org.junit.jupiter.api.extension.ExtensionContext
import org.junit.jupiter.api.extension.ExtensionContext.Namespace
import java.util.concurrent.TimeUnit
import java.util.concurrent.locks.ReentrantLock
import kotlin.concurrent.withLock

@ExtendWith(IntegrationTest.Extension::class)
@Timeout(60, unit = TimeUnit.MINUTES)
abstract class IntegrationTest : AbstractTest() {

    class Extension : BeforeAllCallback {
        companion object {
            private val currentTestPerThread = ThreadLocal.withInitial { null }
            private val namespace = Namespace.create(Extension::class.java)
            private val lock = ReentrantLock()
            private var sharedResources: SharedResources? = null

            fun sharedResources() = lock.withLock {
                sharedResources ?: throw IllegalStateException("Not initialized - beforeAll hasn't been invoked")
            }

            fun setCurrentTestInfo(testInfo: TestInfo) {
                currentTestPerThread.set(testInfo)
            }

            fun unsetCurrentTestInfo() {
                currentTestPerThread.remove()
            }

            fun getCurrentTestInfo(): TestInfo? = currentTestPerThread.get()
        }

        override fun beforeAll(context: ExtensionContext) = lock.withLock {
            if (sharedResources == null) {
                sharedResources = context.root.getStore(namespace).resources()
            }
        }
    }

    companion object {
        fun currentTestInfo() = Extension.getCurrentTestInfo()
    }

    @BeforeEach
    internal fun setTestInfo(testInfo: TestInfo) {
        Extension.setCurrentTestInfo(testInfo)
    }

    @AfterEach
    internal fun unsetTestInfo(testInfo: TestInfo) {
        Extension.unsetCurrentTestInfo()
    }

    fun withDsl(dbSystem: DbSystem, block: (Dsl).() -> (Unit)) = Dsl(dbSystem, Extension.sharedResources()).use(block)
}