io.github.sergkhram.tasks.taskActions.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ares-exec-plugin Show documentation
Android report synchro plugin + execution
There is a newer version: 1.2.10-RELEASE
package io.github.sergkhram.tasks

import com.malinskiy.marathon.FixedQuotaRetryStrategyConfiguration
import com.malinskiy.marathon.MarathonRunTask
import com.malinskiy.marathon.android.configuration.AllureConfiguration
import com.malinskiy.marathon.device.DeviceFeature
import io.github.sergkhram.*
import io.github.sergkhram.configuration.Configuration
import io.github.sergkhram.configuration.ConfigurationExtension
import io.github.sergkhram.configuration.MarathonBlock
import io.github.sergkhram.configuration.ScreenRecordAttachment
import io.github.sergkhram.helpers.CustomException
import io.github.sergkhram.helpers.Defaults.NO_RETRY_STRATEGY_VALUE
import org.gradle.api.GradleException
import org.gradle.api.Project
import org.gradle.kotlin.dsl.get
import java.io.File

fun marathonAfterExecutionAction(project: Project) {
    val buildType = Configuration.buildType
    val marathonTaskName = "marathon${buildType.capitalize()}AndroidTest"
    val marathonTask = project.tasks.get(marathonTaskName) as MarathonRunTask
    logger.info("Getting retry count config value")
    val retryCount: Int = marathonTask.getRetryCountValue()
    logger.info("Retry count value is $retryCount")
    val reportDirectory = Configuration.getReportDirectory(project.projectDir.path)
    logger.info("ReportDirectory is $reportDirectory")
    logger.info("Getting all failed test names")
    val currentMarathonVideosDir = File("${reportDirectory}${ScreenRecordAttachment.directoryName}")
    val marathonVideoDir = marathonVideoDir(reportDirectory)
    val marathonGifDir = marathonGifDir(reportDirectory)
    if(marathonVideoDir.exists() || marathonGifDir.exists()) {
        if (!currentMarathonVideosDir.exists()) throw CustomException("There is no $currentMarathonVideosDir directory. Check the screenRecordType property")
        val failedTests = mutableMapOf()
        currentMarathonVideosDir.listFiles()!!.forEach { shardingStrategy ->
            shardingStrategy.listFiles()!!.forEach { deviceDir ->
                deviceDir.listFiles()!!.forEach { file ->
                    val testName = file.name.split("-").first()
                    if(failedTests.containsKey(testName)) {
                        failedTests.put(testName, failedTests[testName]!!.toInt() + 1)
                    } else {
                        failedTests.put(testName, 1)
                    }
                }
            }
        }
        logger.debug("Failed tests: $failedTests")
        failedTests.forEach { failedTest ->
            if(failedTest.value >= retryCount + 1) {
                failedTestsExceptionAction("marathon")
            }
        }
    }
}

fun failedTestsExceptionAction(executor: String) {
    val message = "At least one test failed during all attempts by $executor. See build/allure-results directory or generate report using allure gradle plugin"
    logger.info(message)
    throw GradleException(message)
}

fun MarathonRunTask.enableAllureForMarathon() {
    this.marathonExtension.get().apply {
        if(this.allureConfiguration==null || !this.allureConfiguration!!.enabled) {
            val allureResultsDir: String? = this.allureConfiguration?.relativeResultsDirectory
            this.allureConfiguration = AllureConfiguration().apply {
                this.enabled = true
                allureResultsDir?.let { resDir -> this.relativeResultsDirectory = resDir }
            }
        }
    }
}

fun MarathonRunTask.getRetryCountValue(): Int {
    val retryStrategy = this.marathonExtension.get().retryStrategy
    return if(retryStrategy?.fixedQuota == null) {
        NO_RETRY_STRATEGY_VALUE
    } else {
        retryStrategy.fixedQuota!!.retryPerTestQuota ?: FixedQuotaRetryStrategyConfiguration().retryPerTestQuota
    }
}

fun ConfigurationExtension.setScreenRecordTypeFromMarathon(marathonTask: MarathonRunTask) {
    val marathonVideoConfiguration = marathonTask.marathonExtension.get().screenRecordConfiguration?.preferableRecorderType?.name
        ?: DeviceFeature.VIDEO.name
    if(this.marathonBlock == null) {
        this.marathonBlock = MarathonBlock(screenRecordType = marathonVideoConfiguration)
    } else {
        this.marathonBlock!!.screenRecordType = marathonVideoConfiguration
    }
}