From 1bc369c63c0a9f923da372fd33e53c1df85ab313 Mon Sep 17 00:00:00 2001
From: Vyacheslav Gerasimov <vyacheslav.gerasimov@jetbrains.com>
Date: Thu, 10 Dec 2020 19:49:21 +0300
Subject: [PATCH] Build: Enable caching for test task with enabled test
 distribution

 #KTI-112
---
 build.gradle.kts                             | 2 +-
 buildSrc/src/main/kotlin/testDistribution.kt | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/build.gradle.kts b/build.gradle.kts
index abf16ff88c6..b3afa6c8093 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -467,7 +467,7 @@ allprojects {
     }
 
     tasks.withType<Test> {
-        outputs.doNotCacheIf("https://youtrack.jetbrains.com/issue/KT-37089") { true }
+        outputs.doNotCacheIf("https://youtrack.jetbrains.com/issue/KTI-112") { !isTestDistributionEnabled() }
     }
 
     normalization {
diff --git a/buildSrc/src/main/kotlin/testDistribution.kt b/buildSrc/src/main/kotlin/testDistribution.kt
index b773e1f27b5..863cdc45704 100644
--- a/buildSrc/src/main/kotlin/testDistribution.kt
+++ b/buildSrc/src/main/kotlin/testDistribution.kt
@@ -25,3 +25,6 @@ fun Test.configureTestDistribution(configure: TestDistributionExtension.() -> Un
         configure()
     }
 }
+
+fun Test.isTestDistributionEnabled(): Boolean =
+    extensions.findByType(TestDistributionExtension::class.java)?.enabled?.orNull ?: false