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

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"

    override fun createPerformanceTestsFor(
        model: CIBuildModel,
        stage: Stage,
        performanceTestCoverage: PerformanceTestCoverage,
        bucketIndex: Int,
    ): PerformanceTest =
        createPerformanceTest(
            model,
            performanceTestCoverage,
            stage,
            bucketIndex,
            "Performance tests for $testProject",

    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

import common.toCapitalized
import model.CIBuildModel
import model.Stage

class SmokeTests(
    model: CIBuildModel,
    stage: Stage,
    testJava: JvmCategory,
    id: String,
    task: String = "smokeTest",
    splitNumber: Int = 1,
    flakyTestStrategy: FlakyTestStrategy,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {

            )
        }

        var prevStage: Stage? = null
        val previousPerformanceTestPasses: MutableList<PerformanceTestsPass> = mutableListOf()
        val previousCrossVersionTests: MutableList<BaseGradleBuildType> = mutableListOf()
        model.stages.forEach { stage ->
            if (isSecurityFork() && stage.stageName > StageName.READY_FOR_RELEASE) {
                return@forEach
            }

import common.getBuildScanCustomValueParam
import common.requiresNotEc2Agent
import model.CIBuildModel
import model.Stage

class SmokeIdeTests(
    model: CIBuildModel,
    stage: Stage,
    flakyTestStrategy: FlakyTestStrategy,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        val suffix = if (flakyTestStrategy == FlakyTestStrategy.ONLY)"_FlakyTestQuarantine" else ""

    model: CIBuildModel,
    performanceTestBucketProvider: PerformanceTestBucketProvider,
    stage: Stage,
    performanceTestCoverage: PerformanceTestCoverage,
) : PerformanceTestProject(
        model,
        performanceTestCoverage,
        performanceTestBucketProvider.createPerformanceTestsFor(stage, performanceTestCoverage),
    )

class ManuallySplitPerformanceTestProject(
    model: CIBuildModel,

// where X, Y, Z are all the stages including current stage
// For example, for the stage PullRequestFeedback, the custom value will be "PartOf=PullRequestFeedback,ReadyForNightly,ReadyForRelease"
private fun Stage.getBuildScanCustomValues(): List<String> =
    StageName
        .values()
        .slice(this.stageName.ordinal until StageName.READY_FOR_RELEASE.ordinal + 1)
        .map { it.uuid }