/*
 * Copyright 2016-2022 JetBrains s.r.o.
 * Use of this source code is governed by the Apache 2.0 License that can be found in the LICENSE.txt file.
 */

package kotlinx.atomicfu.plugin.gradle.internal

import org.gradle.testkit.runner.BuildResult
import org.gradle.testkit.runner.TaskOutcome
import kotlin.test.assertEquals

/**
 * Helper `fun` for asserting a [TaskOutcome] to be equal to [TaskOutcome.SUCCESS]
 */
internal fun BuildResult.assertTaskSuccess(task: String) {
    assertTaskOutcome(TaskOutcome.SUCCESS, task)
}

/**
 * Helper `fun` for asserting a [TaskOutcome] to be equal to [TaskOutcome.FAILED]
 */
internal fun BuildResult.assertTaskFailure(task: String) {
    assertTaskOutcome(TaskOutcome.FAILED, task)
}

internal fun BuildResult.assertTaskUpToDate(task: String) {
    assertTaskOutcome(TaskOutcome.UP_TO_DATE, task)
}

private fun BuildResult.assertTaskOutcome(taskOutcome: TaskOutcome, taskName: String) {
    assertEquals(
        taskOutcome,
        task(taskName)?.outcome,
        "Task $taskName does not have ${taskOutcome.name} outcome"
    )
}