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"

        fun BuildSteps.gradleStep(tasks: List<String>) {
            gradleWrapper {
                name = "GRADLE_RUNNER"
                gradleParams =

 */

package common

interface Jvm {
    val version: JvmVersion
    val vendor: JvmVendor
}

data class DefaultJvm(
    override val version: JvmVersion,
    override val vendor: JvmVendor,
) : Jvm

object BuildToolBuildJvm : Jvm {
    override val version: JvmVersion
        get() = JvmVersion.JAVA_17
    override val vendor: JvmVendor
        get() = JvmVendor.OPENJDK
}

 */

package common

enum class Arch(
    val suffix: String,
    val nameOnLinuxWindows: String,
    val nameOnMac: String,
) {
    AMD64("64bit", "amd64", "x86_64"),
    AARCH64("aarch64", "aarch64", "aarch64"),
    ;

    fun asName() = name.lowercase().toCapitalized()
}

enum class Os(
    val agentRequirement: String,
    val androidHome: String,
    val jprofilerHome: String,

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!!

    object None : ParallelizationMethod()

     4. Use dogfood-second to run a Gradle build which runs tests.

     */
        val dogfoodTimestamp1 = "19800101010101+0000"
        val dogfoodTimestamp2 = "19800202020202+0000"
        val buildScanTagForType = buildScanTagParam("Gradleception")
        val buildScanTagForGroovyMajor = buildScanTagParam("Groovy$bundleGroovyMajor")
        val buildScanTags =

}

data class PerformanceTestPartialTrigger(
    val triggerName: String,
    val triggerId: String,
    val dependencies: List<PerformanceTestCoverage>,
)

data class PerformanceTestCoverage(
    private val uuid: Int,
    override val type: PerformanceTestType,
    override val os: Os,
    override val arch: Arch = Arch.AMD64,
    val numberOfBuckets: Int = 40,
    private val oldUuid: String? = null,

import model.Stage

open class BaseGradleBuildType(
    open val stage: Stage? = null,
    open val failStage: Boolean = true,
    init: BaseGradleBuildType.() -> Unit = {},
) : BuildType() {
    init {
        this.init()
    }
}

open class OsAwareBaseGradleBuildType(
    val os: Os,
    override val stage: Stage? = null,
    override val failStage: Boolean = true,
    init: BaseGradleBuildType.() -> Unit = {},

    arch: Arch = Arch.AMD64,
) : BuildType({
        val id = "Util_RerunFlakyTest${os.asName()}${arch.asName()}"
        name = "Rerun Flaky Test - ${os.asName()} ${arch.asName()}"
        description = "Allows you to rerun a selected flaky test 10 times"
        id(id)
        val testJvmVendorParameter = "testJavaVendor"
        val testJvmVersionParameter = "testJavaVersion"
        val testTaskParameterName = "testTask"

class StatisticBasedFunctionalTestBucketProvider(
    val model: CIBuildModel,
    testBucketsJson: File,
) : FunctionalTestBucketProvider {
    private val objectMapper = ObjectMapper().registerKotlinModule()
    private val buckets: Map<TestCoverage, List<BuildTypeBucket>> by lazy {
        val uuidToTestCoverage = model.stages.flatMap { it.functionalTests }.associateBy { it.uuid }

        init = {
            id("${performanceTestProject.spec.asConfigurationId(model)}_Trigger")
            val performanceTestSpec = performanceTestProject.spec
            name = performanceTestProject.name + " (Trigger)"

            val os = Os.LINUX
            val type = performanceTestSpec.type

            applyDefaultSettings(os)
            params {
                text(