Updated AbstractJetInspectionTest to use runtime if necessary

idea/tests/org/jetbrains/jet/plugin/codeInsight/AbstractJetInspectionTest.kt

@@ -30,6 +30,11 @@ import com.intellij.testFramework.InspectionTestUtil
 import com.intellij.codeInspection.InspectionManager
 import com.intellij.testFramework.fixtures.impl.CodeInsightTestFixtureImpl
 import org.jetbrains.jet.JetTestCaseBuilder
+import org.jetbrains.jet.testing.ConfigLibraryUtil
+import com.intellij.testFramework.IdeaTestUtil
+import com.intellij.openapi.projectRoots.JavaSdk
+import com.intellij.openapi.projectRoots.Sdk
+import org.apache.commons.lang.SystemUtils;
 
 public abstract class AbstractJetInspectionTest: LightCodeInsightFixtureTestCase() {
     override fun getProjectDescriptor(): LightProjectDescriptor = JetLightProjectDescriptor.INSTANCE
@@ -51,6 +56,15 @@ public abstract class AbstractJetInspectionTest: LightCodeInsightFixtureTestCase
                     .listFiles { it.getName().endsWith(".kt") }!!
                     .map { configureByFile(it.getName())!!.getVirtualFile()!! }
 
+            val isWithRuntime = virtualFiles.any({ file ->
+                InTextDirectivesUtils.findStringWithPrefixes(String(file.contentsToByteArray(), "utf-8"), "// WITH_RUNTIME") != null
+             })
+
+            try {
+                if (isWithRuntime) {
+                    ConfigLibraryUtil.configureKotlinRuntime(myFixture!!.getModule(), getFullJavaJDK());
+                }
+
                 val scope = AnalysisScope(getProject(), virtualFiles)
                 scope.invalidate()
 
@@ -60,5 +74,17 @@ public abstract class AbstractJetInspectionTest: LightCodeInsightFixtureTestCase
                 InspectionTestUtil.runTool(toolWrapper, scope, globalContext, inspectionManager)
                 InspectionTestUtil.compareToolResults(globalContext, toolWrapper, false, inspectionsTestDir.getPath())
             }
+            finally {
+                if (isWithRuntime) {
+                    ConfigLibraryUtil.unConfigureKotlinRuntime(myFixture!!.getModule(), IdeaTestUtil.getMockJdk17());
+                }
+            }
+        }
+    }
+
+    protected fun getFullJavaJDK(): Sdk {
+        val javaPath = checkNotNull(SystemUtils.getJavaHome()?.getAbsolutePath(), "JDK Path must not be null")
+        val jdk = JavaSdk.getInstance()?.createJdk("JDK", javaPath);
+        return checkNotNull(jdk, "Could not obtain JDK instance")
     }
 }