ProbabilityBasedFlakinessStrategy.kt

class ProbabilityBasedFlakinessStrategy(val minSuccessRate: Double,
                                        val maxCount: Int,
                                        val timeLimit: Instant) : FlakinessStrategy {

    override fun process(testShard: TestShard,
                         metricsProvider: MetricsProvider): TestShard {
        val tests = testShard.tests
        val output = mutableListOf<Test>()
        tests.forEach {
            val successRate = metricsProvider.successRate(it, timeLimit)

            if (successRate < minSuccessRate) {
                val maxFailRate = 1.0 - minSuccessRate
                var currentFailRate = 1.0 - successRate
                var counter = 0
                while (currentFailRate > maxFailRate && counter < maxCount) {
                    output.add(it)
                    currentFailRate *= currentFailRate
                    counter++
                }
            }
        }
        return testShard.copy(flakyTests = output)
    }
}