import common.Os
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import common.setArtifactRules
import model.CIBuildModel
import model.Stage

class CompileAll(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        id(buildTypeId(model))
        name = "Compile All"
        description = "Compiles all production/test source code and warms up the build cache"

        performanceTests =
            stage.performanceTests.map { createPerformanceTests(model, performanceTestBucketProvider, stage, it) } +
            stage.flameGraphs.map { createFlameGraphs(model, stage, it) }

        val (topLevelCoverage, allCoverage) = stage.functionalTests.partition { it.testType == TestType.SOAK }
        val topLevelFunctionalTests =
            topLevelCoverage

    stats = cache.stats();
    assertThat(stats.requestCount()).isEqualTo(3);
    assertThat(stats.hitCount()).isEqualTo(1);
    assertThat(stats.hitRate()).isEqualTo(1.0 / 3);
    assertThat(stats.missCount()).isEqualTo(2);
    assertThat(stats.missRate()).isEqualTo(2.0 / 3);
    assertThat(stats.loadCount()).isEqualTo(2);
    assertThat(stats.totalLoadTime()).isAtLeast(totalLoadTime);
    totalLoadTime = stats.totalLoadTime();

    },
    BuildLogicTest {
        override fun create(
            model: CIBuildModel,
            stage: Stage,
            flakyTestStrategy: FlakyTestStrategy,
        ): OsAwareBaseGradleBuildType = BuildLogicTest(model, stage)
    },
    BuildDistributions {
        override fun create(
            model: CIBuildModel,
            stage: Stage,
            flakyTestStrategy: FlakyTestStrategy,

    val testCoverage: TestCoverage,
    stage: Stage,
    parallelizationMethod: ParallelizationMethod = ParallelizationMethod.None,
    subprojects: List<String> = listOf(),
    extraParameters: String = "",
    extraBuildSteps: BuildSteps.() -> Unit = {},
    preBuildSteps: BuildSteps.() -> Unit = {},
) : OsAwareBaseGradleBuildType(os = testCoverage.os, stage = stage, init = {
        this.name = name

    }
}

class StageTrigger(
    model: CIBuildModel,
    stage: Stage,
    prevStage: Stage?,
    os: Os?,
    dependencies: List<BaseGradleBuildType>,
    generateTriggers: Boolean = true,
) : BaseGradleBuildType(init = {
        id(stageTriggerId(model, stage, os))
        uuid = stageTriggerUuid(model, stage, os)
        name = stage.stageName.stageName + " (Trigger)" + (os?.asName()?.toCapitalized()?.let { "($it)" } ?: "")

    override fun createFunctionalTestsFor(
        stage: Stage,
        testCoverage: TestCoverage,
    ): List<FunctionalTest> =
        when {
            testCoverage.testType == TestType.QUICK_FEEDBACK_CROSS_VERSION ->
                quickCrossVersionTestBucketProvider.createFunctionalTestsFor(
                    stage,
                    testCoverage,
                )

package configurations

import common.Os
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import model.CIBuildModel
import model.Stage

class SanityCheck(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        id(buildTypeId(model))
        name = "Sanity Check"
        description = "Static code analysis, checkstyle, release notes verification, etc."

import common.buildScanTagParam
import common.getBuildScanCustomValueParam
import common.setArtifactRules
import model.CIBuildModel
import model.Stage

class BuildDistributions(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = LINUX, stage = stage, init = {
        id("${model.projectId}_BuildDistributions")
        name = "Build Distributions"

    CONFIG_CACHE_DISABLED(false, "DocsTest", "Docs Test"),
}

class DocsTest(
    model: CIBuildModel,
    stage: Stage,
    os: Os,
    testJava: JvmCategory,
    docsTestType: DocsTestType,
) : OsAwareBaseGradleBuildType(os = os, stage = stage, init = {
        id("${model.projectId}_${docsTestType.docsTestName}_${os.asName()}")