subProject(flakyTestQuarantineProject)
                buildType(flakyTestQuarantineProjectTrigger)
            }
        }

        stage.performanceTestPartialTriggers.forEach { trigger ->
            buildType(
                PartialTrigger(
                    trigger.triggerName,
                    trigger.triggerId,
                    model,

        )

    @BeforeEach
    fun setUp() {
        val stepsCapturer = slot<BuildSteps.() -> Unit>()
        every { buildType.steps } returns steps
        every { buildType.stage } returns buildModel.stages[2]
        every {
            buildType.steps(capture(stepsCapturer))
        } answers {
            stepsCapturer.captured(steps)
            mockk()
        }
    }

    @Test

        cleanupRule(historyDays = 14, artifactsDays = 7)

        buildType(SanityCheck)
        buildType(PublishNightlySnapshot(branch))
        buildType(PublishNightlySnapshotFromQuickFeedback(branch))
        buildType(PublishNightlySnapshotFromQuickFeedbackStepCheckReady(branch))
        buildType(PublishNightlySnapshotFromQuickFeedbackStepUpload(branch))
        buildType(PublishNightlySnapshotFromQuickFeedbackStepPromote(branch))

            }
        skipConditionally(buildType)
    }
}

fun BuildStep.onlyRunOnGitHubMergeQueueBranch() {
    conditions {
        matches("teamcity.build.branch", "gh-readonly-queue/.*")
    }
}

fun BuildStep.skipConditionally(buildType: BuildType? = null) {
    // we need to run CompileALl unconditionally because of artifact dependency
    if (buildType !is CompileAll) {
        conditions {

    buildType.killProcessStep(KILL_PROCESSES_STARTED_BY_GRADLE, os, arch, executionMode = ExecutionMode.ALWAYS)

    buildType.steps {
        extraSteps()
        checkCleanM2AndAndroidUserHome(os, buildType)
    }
    buildType.addEc2PostBuild(os)

    applyDefaultDependencies(model, buildType)
}

fun applyDefaultDependencies(
    model: CIBuildModel,
    buildType: BuildType,
) {

object UtilProject : Project({
    id("Util")
    name = "Util"

    buildType(RerunFlakyTest(Os.LINUX))
    buildType(RerunFlakyTest(Os.WINDOWS))
    buildType(RerunFlakyTest(Os.MACOS, Arch.AMD64))
    buildType(RerunFlakyTest(Os.MACOS, Arch.AARCH64))
    buildType(WarmupEc2Agent)
    buildType(DownloadGitRepoToEc2Agent)
    buildType(UpdateWrapper)

    buildType(PublishKotlinDslPlugin)

    params {

package configurations

import common.applyDefaultSettings
import jetbrains.buildServer.configs.kotlin.BuildStep
import jetbrains.buildServer.configs.kotlin.BuildType
import jetbrains.buildServer.configs.kotlin.buildSteps.script

object DownloadGitRepoToEc2Agent : BuildType({
    val id = "Util_DownloadGitRepoToEc2Agent"
    name = "Download Git Repo to EC2 Agent"
    description = "Do nothing but downloading gradle/gradle repo to EC2 agents"

fun <T : BaseGradleBuildType> Dependencies.snapshotDependencies(
    buildTypes: Iterable<T>,
    snapshotConfig: SnapshotDependency.(T) -> Unit = {},
) {
    buildTypes.forEach { buildType ->
        dependency(buildType.id!!) {
            snapshot {
                if (!buildType.failStage) {
                    onDependencyFailure = FailureAction.IGNORE

 */

package common

import jetbrains.buildServer.configs.kotlin.BuildStep
import jetbrains.buildServer.configs.kotlin.BuildSteps
import jetbrains.buildServer.configs.kotlin.BuildType
import jetbrains.buildServer.configs.kotlin.buildSteps.script

fun BuildType.applyPerformanceTestSettings(
    os: Os = Os.LINUX,
    arch: Arch = Arch.AMD64,
    timeout: Int = 30,
) {
    applyDefaultSettings(os = os, arch = arch, timeout = timeout)

import common.setArtifactRules
import common.substDirOnWindows
import jetbrains.buildServer.configs.kotlin.BuildType
import jetbrains.buildServer.configs.kotlin.ParameterDisplay
import jetbrains.buildServer.configs.kotlin.ParametrizedWithType

abstract class AdHocPerformanceScenario(
    os: Os,
    arch: Arch = Arch.AMD64,
) : BuildType({
        val id = "Util_Performance_AdHocPerformanceScenario${os.asName()}${arch.asName()}"