1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/*
* Copyright 2016-2022 JetBrains s.r.o.
* Use of this source code is governed by the Apache 2.0 License that can be found in the LICENSE.txt file.
*/
package kotlinx.atomicfu.plugin.gradle.internal
import kotlinx.atomicfu.plugin.gradle.test.*
import org.gradle.testkit.runner.*
import java.io.*
internal fun BaseKotlinGradleTest.test(fn: BaseKotlinScope.() -> Unit): GradleRunner {
val baseKotlinScope = BaseKotlinScope()
fn(baseKotlinScope)
baseKotlinScope.files.forEach { scope ->
val fileWriteTo = rootProjectDir.resolve(scope.filePath)
.apply {
parentFile.mkdirs()
createNewFile()
}
scope.files.forEach {
val fileContent = readFileList(it)
fileWriteTo.appendText(fileContent)
}
}
return GradleRunner.create()
.withProjectDir(rootProjectDir)
.withArguments(baseKotlinScope.runner.arguments)
.withPluginClasspath()
.addPluginTestRuntimeClasspath()
}
/**
* same as [file][FileContainer.file], but prepends "src/${sourceSet}/kotlin" before given `classFileName`
*/
internal fun FileContainer.kotlin(classFileName: String, sourceSet: String = "main", fn: AppendableScope.() -> Unit) {
require(classFileName.endsWith(".kt")) {
"ClassFileName must end with '.kt'"
}
val fileName = "src/${sourceSet}/kotlin/$classFileName"
file(fileName, fn)
}
/**
* Shortcut for creating a `build.gradle.kts` by using [file][FileContainer.file]
*/
internal fun FileContainer.buildGradleKts(fn: AppendableScope.() -> Unit) {
val fileName = "build.gradle.kts"
file(fileName, fn)
}
/**
* Shortcut for creating a `settings.gradle.kts` by using [file][FileContainer.file]
*/
internal fun FileContainer.settingsGradleKts(fn: AppendableScope.() -> Unit) {
val fileName = "settings.gradle.kts"
file(fileName, fn)
}
/**
* Shortcut for creating a `gradle.properties` by using [file][FileContainer.file]
*/
internal fun FileContainer.gradleProperties(fn: AppendableScope.() -> Unit) {
val fileName = "gradle.properties"
file(fileName, fn)
}
/**
* Declares a directory with the given [dirName] inside the current container.
* All calls creating files within this scope will create the files nested in this directory.
*
* Note that it is valid to call this method multiple times at the same level with the same [dirName].
* Files declared within 2 independent calls to [dir] will be added to the same directory.
*/
internal fun FileContainer.dir(dirName: String, fn: DirectoryScope.() -> Unit) {
DirectoryScope(dirName, this).fn()
}
internal fun BaseKotlinScope.runner(fn: Runner.() -> Unit) {
val runner = Runner()
fn(runner)
this.runner = runner
}
internal fun AppendableScope.resolve(fileName: String) {
this.files.add(fileName)
}
internal interface FileContainer {
fun file(fileName: String, fn: AppendableScope.() -> Unit)
}
internal class BaseKotlinScope : FileContainer {
var files: MutableList<AppendableScope> = mutableListOf()
var runner: Runner = Runner()
override fun file(fileName: String, fn: AppendableScope.() -> Unit) {
val appendableScope = AppendableScope(fileName)
fn(appendableScope)
files.add(appendableScope)
}
}
internal class DirectoryScope(
val dirPath: String,
val parent: FileContainer
): FileContainer {
override fun file(fileName: String, fn: AppendableScope.() -> Unit) {
parent.file("$dirPath/$fileName", fn)
}
}
internal class AppendableScope(val filePath: String) {
val files: MutableList<String> = mutableListOf()
}
internal class Runner {
val arguments: MutableList<String> = mutableListOf()
}
internal fun readFileList(fileName: String): String =
getFile(fileName).readText()
internal fun getFile(fileName: String): File {
val resource = BaseKotlinGradleTest::class.java.classLoader.getResource(fileName)
?: throw IllegalStateException("Could not find resource '$fileName'")
return File(resource.toURI())
}
internal fun GradleRunner.addPluginTestRuntimeClasspath() = apply {
val pluginClasspath = getFile("plugin-classpath.txt").readLines().map { File(it) }
withPluginClasspath(pluginClasspath)
}
|
