Implement tests for generic and remote repls

2016-09-22 11:36:42 +02:00
parent 43037373d7
commit 52f291f372
2 changed files with 233 additions and 63 deletions
@@ -0,0 +1,114 @@
+/*
+ * Copyright 2010-2016 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.cli.jvm.repl
+
+import com.intellij.openapi.Disposable
+import com.intellij.openapi.util.Disposer
+import junit.framework.TestCase
+import org.jetbrains.kotlin.cli.common.messages.MessageRenderer
+import org.jetbrains.kotlin.cli.common.messages.PrintingMessageCollector
+import org.jetbrains.kotlin.cli.common.repl.*
+import org.jetbrains.kotlin.cli.jvm.config.addJvmClasspathRoots
+import org.jetbrains.kotlin.cli.jvm.config.jvmClasspathRoots
+import org.jetbrains.kotlin.config.CommonConfigurationKeys
+import org.jetbrains.kotlin.config.CompilerConfiguration
+import org.jetbrains.kotlin.integration.KotlinIntegrationTestBase
+import org.jetbrains.kotlin.script.KotlinScriptDefinition
+import org.jetbrains.kotlin.script.KotlinScriptDefinitionFromAnnotatedTemplate
+import org.jetbrains.kotlin.utils.PathUtil
+import org.junit.Test
+import java.io.File
+import java.net.URLClassLoader
+
+class GenericReplTest : TestCase() {
+
+    @Test
+    fun testReplBasics() {
+
+        val disposable = Disposer.newDisposable()
+
+        val repl = TestRepl(disposable,
+                            listOf(File(KotlinIntegrationTestBase.getCompilerLib(), "kotlin-runtime.jar")),
+                            "kotlin.script.StandardScriptTemplate")
+
+        val res1 = repl.replCompiler?.check(ReplCodeLine(0, "val x ="), emptyList())
+        TestCase.assertTrue("Unexpected check results: $res1", res1 is ReplCheckResult.Incomplete)
+
+        val codeLine0 = ReplCodeLine(0, "1 + 2")
+        val res2 = repl.replCompiler?.compile(codeLine0, emptyList())
+        val res2c = res2 as? ReplCompileResult.CompiledClasses
+        TestCase.assertNotNull("Unexpected compile result: $res2", res2c)
+
+        val res21 = repl.compiledEvaluator.eval(codeLine0, emptyList(), res2c!!.classes, res2c.hasResult, res2c.newClasspath)
+        val res21e = res21 as? ReplEvalResult.ValueResult
+        TestCase.assertNotNull("Unexpected eval result: $res21", res21e)
+        TestCase.assertEquals(3, res21e!!.value)
+
+        val codeLine1 = ReplCodeLine(1, "val x = 5")
+        val res3 = repl.replCompiler?.compile(codeLine1, listOf(codeLine0))
+        val res3c = res3 as? ReplCompileResult.CompiledClasses
+        TestCase.assertNotNull("Unexpected compile result: $res3", res3c)
+
+        val res31 = repl.compiledEvaluator.eval(codeLine1, listOf(codeLine0), res3c!!.classes, res3c.hasResult, res3c.newClasspath)
+        val res31e = res31 as? ReplEvalResult.UnitResult
+        TestCase.assertNotNull("Unexpected eval result: $res31", res31e)
+
+        val codeLine2 = ReplCodeLine(2, "x + 2")
+        val res4x = repl.replCompiler?.compile(codeLine2, listOf(codeLine1))
+        TestCase.assertNotNull("Unexpected compile result: $res4x", res4x as? ReplCompileResult.HistoryMismatch)
+
+        val res4 = repl.replCompiler?.compile(codeLine2, listOf(codeLine0, codeLine1))
+        val res4c = res4 as? ReplCompileResult.CompiledClasses
+        TestCase.assertNotNull("Unexpected compile result: $res4", res4c)
+
+        val res41 = repl.compiledEvaluator.eval(codeLine2, emptyList(), res4c!!.classes, res4c.hasResult, res4c.newClasspath)
+        val res41e = res41 as? ReplEvalResult.ValueResult
+        TestCase.assertNotNull("Unexpected eval result: $res41", res41e)
+        TestCase.assertEquals(7, res41e!!.value)
+
+        Disposer.dispose(disposable)
+    }
+}
+
+internal class TestRepl(
+        disposable: Disposable,
+        templateClasspath: List<File>,
+        templateClassName: String
+) {
+    private val configuration = CompilerConfiguration().apply {
+        addJvmClasspathRoots(PathUtil.getJdkClassesRoots())
+        addJvmClasspathRoots(PathUtil.getKotlinPathsForCompiler().let { listOf(it.runtimePath, it.reflectPath) })
+        addJvmClasspathRoots(templateClasspath)
+        put(CommonConfigurationKeys.MODULE_NAME, "kotlin-script")
+    }
+    private fun makeScriptDefinition(templateClasspath: List<File>, templateClassName: String): KotlinScriptDefinition {
+        val classloader = URLClassLoader(templateClasspath.map { it.toURI().toURL() }.toTypedArray(), this.javaClass.classLoader)
+        val cls = classloader.loadClass(templateClassName)
+        return KotlinScriptDefinitionFromAnnotatedTemplate(cls.kotlin, null, null, emptyMap())
+    }
+
+    private val scriptDef = makeScriptDefinition(templateClasspath, templateClassName)
+
+    val replCompiler : GenericReplCompiler? by lazy {
+        GenericReplCompiler(disposable, scriptDef, configuration, PrintingMessageCollector(System.out, MessageRenderer.WITHOUT_PATHS, false))
+    }
+
+    val compiledEvaluator : GenericReplCompiledEvaluator by lazy {
+        GenericReplCompiledEvaluator(configuration.jvmClasspathRoots, null, arrayOf(emptyArray<String>()))
+    }
+}
+