stage.flameGraphs.map { createFlameGraphs(model, stage, it) }

        val (topLevelCoverage, allCoverage) = stage.functionalTests.partition { it.testType == TestType.SOAK }
        val topLevelFunctionalTests =
            topLevelCoverage
                .map { FunctionalTest(model, it.asConfigurationId(model), it.asName(), it.asName(), it, stage = stage) }
        topLevelFunctionalTests.forEach(this::buildType)

    fun hasTestsOf(testType: TestType) =
        (unitTests && testType.unitTests) ||
            (functionalTests && testType.functionalTests) ||
            (crossVersionTests && testType.crossVersionTests)
}

data class Stage(
    val stageName: StageName,
    val specificBuilds: List<SpecificBuild> = emptyList(),
    val functionalTests: List<TestCoverage> = emptyList(),
    val docsTests: List<DocsTestCoverage> = emptyList(),

        val functionalTests = !ignoredSubprojects.contains(name) && subproject["functionalTests"] as Boolean
        val crossVersionTests = !ignoredSubprojects.contains(name) && subproject["crossVersionTests"] as Boolean
        return GradleSubproject(name, path, unitTests, functionalTests, crossVersionTests)
    }

    val stage: Stage,
) : Project({
        this.id(testCoverage.asId(model))
        this.name = testCoverage.asName()
    }) {
    val functionalTests: List<BaseGradleBuildType> = functionalTestBucketProvider.createFunctionalTestsFor(stage, testCoverage)

    init {
        functionalTests.forEach(this::buildType)
    }

        val result = mutableMapOf<TestCoverage, List<SmallSubprojectBucket>>()
        for (stage in model.stages) {
            for (testCoverage in stage.functionalTests) {
                if (testCoverage.testType !in
                    listOf(TestType.ALL_VERSIONS_CROSS_VERSION, TestType.QUICK_FEEDBACK_CROSS_VERSION, TestType.SOAK)
                ) {

                        StageName.QUICK_FEEDBACK,
                        StageName.PULL_REQUEST_FEEDBACK,
                        StageName.READY_FOR_NIGHTLY,
                    )
            }.flatMap { it.functionalTests }
            .filter { it.os == os }
            .forEach {
                buildType(FlakyTestQuarantine(model, stage, it))
            }
    })

class FlakyTestQuarantine(
    model: CIBuildModel,

        name = functionalTestProject.name + " (Trigger)"
        type = Type.COMPOSITE

        applyDefaultSettings()

        dependencies {
            snapshotDependencies(functionalTestProject.functionalTests)
        }
    }) {
    val testCoverage: TestCoverage = functionalTestProject.testCoverage

    override val failStage: Boolean
        get() = testCoverage.failStage

        stage: Stage,
        testCoverage: TestCoverage,
        bucketIndex: Int,
    ): FunctionalTest {
        val parallelizationMethod =
            when (testCoverage.os) {
                Os.LINUX -> ParallelizationMethod.TestDistribution
                else -> ParallelizationMethod.None
            }

        return FunctionalTest(
            model,
            testCoverage.getBucketUuid(model, bucketIndex),

) {
    val triggers: List<BaseGradleBuildType>

    init {
        triggers = mutableListOf()
        val allDependencies =
            stageProject.specificBuildTypes + stageProject.performanceTests + stageProject.functionalTests + stageProject.docsTestTriggers
        triggers.add(StageTrigger(model, stage, prevStage, null, allDependencies))

        stageWithOsTriggers.getOrDefault(stage.stageName, emptyList()).forEach { targetOs ->

import jetbrains.buildServer.configs.kotlin.buildFeatures.perfmon
import model.CIBuildModel
import model.Stage
import model.StageName
import model.TestCoverage
import model.TestType

const val FUNCTIONAL_TEST_TAG = "FunctionalTest"

sealed class ParallelizationMethod {
    open val extraBuildParameters: String
        @JsonProperty(access = JsonProperty.Access.WRITE_ONLY)
        get() = ""

    val name: String = this::class.simpleName!!