if (testCoverage.testType !in
                    listOf(TestType.ALL_VERSIONS_CROSS_VERSION, TestType.QUICK_FEEDBACK_CROSS_VERSION, TestType.SOAK)
                ) {
                    result[testCoverage] = splitBucketsByTestClassesForBuildProject(testCoverage, buildProjectClassTimes)
                }
            }
        }
        return result
    }

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

                        TestCoverage(35, TestType.SOAK, Os.MACOS, JvmCategory.MAX_LTS_VERSION, 1, arch = Arch.AARCH64),
                        TestCoverage(
                            10,
                            TestType.ALL_VERSIONS_CROSS_VERSION,
                            Os.LINUX,
                            JvmCategory.MIN_VERSION,

        val testTasks = getTestTaskName(testCoverage, subprojects)

        val assembledExtraParameters =
            mutableListOf(
                stage.getBuildScanCustomValueParam(testCoverage),
                functionalTestExtraParameters(
                    listOf(FUNCTIONAL_TEST_TAG),
                    testCoverage.os,
                    testCoverage.arch,
                    testCoverage.testJvmVersion.major.toString(),

        applyDefaultSettings()

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

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

class FlakyTestQuarantine(
    model: CIBuildModel,
    stage: Stage,
    testCoverage: TestCoverage,
) : OsAwareBaseGradleBuildType(os = testCoverage.os, stage = stage, init = {
        val os = testCoverage.os
        val arch = testCoverage.arch
        id("${model.projectId}_FlakyQuarantine_${testCoverage.asId(model)}")
        name = "Flaky Test Quarantine - ${testCoverage.asName()}"
        description = "Run all flaky tests skipped multiple times"

        .slice(this.stageName.ordinal until StageName.READY_FOR_RELEASE.ordinal + 1)
        .map { it.uuid }

fun Stage.getBuildScanCustomValueParam(testCoverage: TestCoverage? = null): String {
    val customValues =
        if (testCoverage != null) {
            listOf(testCoverage.asBuildScanCustomValue()) + getBuildScanCustomValues()
        } else {
            getBuildScanCustomValues()
        }

        model: CIBuildModel,
        stage: Stage,
        testCoverage: TestCoverage,
        bucketIndex: Int,
    ): FunctionalTest =
        FunctionalTest(
            model,
            testCoverage.getBucketUuid(model, bucketIndex),
            getName(testCoverage),
            getDescription(testCoverage),
            testCoverage,
            stage,
            parallelizationMethod,

import model.Stage
import model.TestCoverage

const val DEFAULT_FUNCTIONAL_TEST_BUCKET_SIZE = 50
const val DEFAULT_LINUX_FUNCTIONAL_TEST_BUCKET_SIZE = 20
const val DEFAULT_MACOS_FUNCTIONAL_TEST_BUCKET_SIZE = 20

class FunctionalTestProject(
    val model: CIBuildModel,
    functionalTestBucketProvider: FunctionalTestBucketProvider,
    val testCoverage: TestCoverage,
    val stage: Stage,
) : Project({

        topLevelFunctionalTests.forEach(this::buildType)

        val functionalTestProjects =
            allCoverage.map { testCoverage ->
                FunctionalTestProject(
                    model,
                    functionalTestBucketProvider,
                    testCoverage,
                    stage,
                )
            }

        functionalTestProjects.forEach { functionalTestProject ->