org.jetbrains.spek.engine.FixturesAdapter.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spek-junit-platform-engine Show documentation

Spek TestEngine for the JUnit Platform

There is a newer version: 1.1.5

Show newest version

package org.jetbrains.spek.engine

import org.jetbrains.spek.extension.Extension
import org.jetbrains.spek.extension.GroupExtensionContext
import org.jetbrains.spek.extension.TestExtensionContext
import java.util.*

/**
 * Adapter for fixtures as a pseudo-extension.
 *
 * @author Ranie Jade Ramiso
 */
class FixturesAdapter: Extension {
    private val beforeEach: MutableMap Unit>> = WeakHashMap()
    private val afterEach: MutableMap Unit>> = WeakHashMap()

    fun beforeExecuteTest(test: TestExtensionContext) {
        invokeAllBeforeEach(test.parent)
    }

    fun afterExecuteTest(test: TestExtensionContext) {
        invokeAllAfterEach(test.parent)
    }

    fun beforeExecuteGroup(test: GroupExtensionContext) {
        invokeAllBeforeEach(test)
    }

    fun afterExecuteGroup(test: GroupExtensionContext) {
        invokeAllAfterEach(test)
    }

    fun registerBeforeEach(group: GroupExtensionContext, callback: () -> Unit) {
        beforeEach.getOrPut(group, { LinkedList() }).add(callback)
    }

    fun registerAfterEach(group: GroupExtensionContext, callback: () -> Unit) {
        afterEach.getOrPut(group, { LinkedList() }).add(callback)
    }

    private fun invokeAllBeforeEach(group: GroupExtensionContext) {
        if (group.parent != null) {
            invokeAllBeforeEach(group.parent!!)
        }
        beforeEach[group]?.forEach { it.invoke() }
    }

    private fun invokeAllAfterEach(group: GroupExtensionContext) {
        afterEach[group]?.forEach { it.invoke() }
        if (group.parent != null) {
            invokeAllAfterEach(group.parent!!)
        }
    }
}