Ilya Chernikov
15f9ad9531
so ".kt" and ".java" files are not considered as scripts and quickly filtered out, and for the other files the the checks are implemented using sequences, mechanisms provided to supply script definitions lazily, and script discovery is implemented using this mechanisms.
80 lines
2.8 KiB
Kotlin
80 lines
2.8 KiB
Kotlin
/*
|
|
* Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
|
|
* that can be found in the license/LICENSE.txt file.
|
|
*/
|
|
|
|
package org.jetbrains.kotlin.scripts
|
|
|
|
import org.jetbrains.kotlin.cli.common.script.CliScriptDefinitionProvider
|
|
import org.jetbrains.kotlin.script.KotlinScriptDefinition
|
|
import org.jetbrains.kotlin.script.ScriptDefinitionsSource
|
|
import org.jetbrains.kotlin.test.testFramework.KtUsefulTestCase
|
|
import org.junit.Assert
|
|
import org.junit.Test
|
|
import java.util.concurrent.atomic.AtomicInteger
|
|
import kotlin.coroutines.experimental.buildSequence
|
|
import kotlin.script.templates.standard.ScriptTemplateWithArgs
|
|
|
|
class ScriptProviderTest : KtUsefulTestCase() {
|
|
|
|
@Test
|
|
fun testLazyScriptDefinitionsProvider() {
|
|
|
|
val genDefCounter = AtomicInteger()
|
|
val standardDef = FakeScriptDefinition()
|
|
val shadedDef = FakeScriptDefinition(".x.kts")
|
|
val provider = CliScriptDefinitionProvider().apply {
|
|
setScriptDefinitions(listOf(shadedDef, standardDef))
|
|
setScriptDefinitionsSources(listOf(TestScriptDefinitionSource(genDefCounter, ".y.kts", ".x.kts")))
|
|
}
|
|
|
|
Assert.assertEquals(0, genDefCounter.get())
|
|
|
|
provider.isScript("a.kt").let {
|
|
Assert.assertFalse(it)
|
|
Assert.assertEquals(0, genDefCounter.get())
|
|
}
|
|
|
|
provider.isScript("a.y.kts").let {
|
|
Assert.assertTrue(it)
|
|
Assert.assertEquals(1, genDefCounter.get())
|
|
}
|
|
|
|
provider.isScript("a.x.kts").let {
|
|
Assert.assertTrue(it)
|
|
Assert.assertEquals(2, genDefCounter.get())
|
|
Assert.assertEquals(0, shadedDef.matchCounter.get())
|
|
}
|
|
|
|
provider.isScript("a.z.kts").let {
|
|
Assert.assertTrue(it)
|
|
Assert.assertEquals(2, genDefCounter.get())
|
|
Assert.assertEquals(1, standardDef.matchCounter.get())
|
|
}
|
|
|
|
provider.isScript("a.ktx").let {
|
|
Assert.assertFalse(it)
|
|
Assert.assertEquals(2, genDefCounter.get())
|
|
}
|
|
}
|
|
}
|
|
|
|
private class FakeScriptDefinition(val suffix: String = ".kts") : KotlinScriptDefinition(ScriptTemplateWithArgs::class) {
|
|
val matchCounter = AtomicInteger()
|
|
override fun isScript(fileName: String): Boolean = fileName.endsWith(suffix).also {
|
|
if (it) matchCounter.incrementAndGet()
|
|
}
|
|
}
|
|
|
|
private class TestScriptDefinitionSource(val counter: AtomicInteger, val defGens: Iterable<() -> FakeScriptDefinition>) :
|
|
ScriptDefinitionsSource
|
|
{
|
|
constructor(counter: AtomicInteger, vararg suffixes: String) : this(counter, suffixes.map { { FakeScriptDefinition(it) } })
|
|
|
|
override val definitions: Sequence<KotlinScriptDefinition> = buildSequence {
|
|
for (gen in defGens) {
|
|
counter.incrementAndGet()
|
|
yield(gen())
|
|
}
|
|
}
|
|
} |