From dc7171e9538be1f9ae110f21e169b89a418ec0f9 Mon Sep 17 00:00:00 2001
From: Michael Nedzelsky <Michael.Nedzelsky@jetbrains.com>
Date: Wed, 18 Nov 2015 22:34:38 +0300
Subject: [PATCH] improve check for access to internal from tests for projects
 compiled via gradle

---
 .../org/jetbrains/kotlin/gradle/tasks/Tasks.kt    | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt b/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt
index 3c20ec15e45..4bfdad7d2cb 100644
--- a/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt
+++ b/libraries/tools/kotlin-gradle-plugin-core/src/main/kotlin/org/jetbrains/kotlin/gradle/tasks/Tasks.kt
@@ -100,6 +100,12 @@ public open class KotlinCompile() : AbstractKotlinCompile<K2JVMCompilerArguments
     // Should be SourceDirectorySet or File
     val srcDirsSources = HashSet<Any>()
 
+    private val testsMap = mapOf(
+            "compileTestKotlin" to  "compileKotlin",
+            "compileDebugUnitTestKotlin" to  "compileDebugKotlin",
+            "compileReleaseUnitTestKotlin" to  "compileReleaseKotlin"
+            )
+
     override fun populateTargetSpecificArgs(args: K2JVMCompilerArguments) {
         // show kotlin compiler where to look for java source files
         args.freeArgs = (args.freeArgs + getJavaSourceRoots().map { it.getAbsolutePath() }).toSet().toList()
@@ -149,13 +155,10 @@ public open class KotlinCompile() : AbstractKotlinCompile<K2JVMCompilerArguments
             }
         }
 
-        if (this.name == "compileTestKotlin") {
-            addFriendPathForTestTask("compileKotlin")
+        testsMap.get(this.name)?.let {
+            addFriendPathForTestTask(it)
         }
-        if (this.name == "compileDebugUnitTestKotlin") {
-            addFriendPathForTestTask("compileDebugKotlin")
-        }
-
+9
         getLogger().kotlinDebug("args.moduleName = ${args.moduleName}")
     }