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

            .filter { it.os == os }
            .forEach {
                buildType(FlakyTestQuarantine(model, stage, it))
            }
    })

class FlakyTestQuarantine(
    model: CIBuildModel,
    stage: Stage,
    testCoverage: TestCoverage,
) : OsAwareBaseGradleBuildType(os = testCoverage.os, stage = stage, init = {
        val os = testCoverage.os
        val arch = testCoverage.arch

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

package configurations

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

class CheckLinks(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, failStage = false, init = {
        id("${model.projectId}_CheckLinks")
        name = "CheckLinks"
        description = "Check links in documentations"

        applyDefaults(
            model,

package configurations

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

class BuildLogicTest(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage = stage, init = {
        id(buildTypeId(model))
        name = "Build-logic test"
        description = "Run :build-logic:test"

        features {

package configurations

import common.Os
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
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"

import jetbrains.buildServer.configs.kotlin.buildSteps.script
import model.CIBuildModel
import model.Stage

class CheckTeamCityKotlinDSL(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(
        os = Os.LINUX,
        stage = stage,
        init = {
            id("${model.projectId}_CheckTeamCityKotlinDSL")
            name = "CheckTeamCityKotlinDSL"

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."

package configurations

import common.Os.LINUX
import common.buildScanTagParam
import common.getBuildScanCustomValueParam
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"

import common.killProcessStep
import common.skipConditionally
import jetbrains.buildServer.configs.kotlin.BuildSteps
import model.CIBuildModel
import model.Stage

class TestPerformanceTest(
    model: CIBuildModel,
    stage: Stage,
) : OsAwareBaseGradleBuildType(os = Os.LINUX, stage, init = {
        val os = Os.LINUX
        val buildTypeThis = this
        val testProject = "smallJavaMultiProject"