blob: bf86c1b119e47077c7c2b9399f11ca65b197bb8b (
plain)
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
|
package utils
import org.jetbrains.dokka.model.DClass
import org.jetbrains.dokka.model.Documentable
import org.jetbrains.dokka.model.doc.*
import org.jetbrains.dokka.testApi.testRunner.AbstractCoreTest
@DslMarker
annotation class TestDSL
@TestDSL
abstract class ModelDSL : AbstractCoreTest() {
operator fun Documentable?.div(name: String): Documentable? =
this?.children?.find { it.name == name }
inline fun <reified T : Documentable> Documentable?.cast(): T =
(this as? T).assertNotNull()
}
@TestDSL
interface AssertDSL {
infix fun Any?.equals(other: Any?) = this.assertEqual(other)
infix fun Collection<Any>?.allEquals(other: Any?) =
this?.also { c -> c.forEach { it equals other } } ?: run { assert(false) { "Collection is empty" } }
infix fun <T> Collection<T>?.counts(n: Int) = this.orEmpty().assertCount(n)
infix fun <T> T?.notNull(name: String): T = this.assertNotNull(name)
fun <T> Collection<T>.assertCount(n: Int, prefix: String = "") =
assert(count() == n) { "${prefix}Expected $n, got ${count()}" }
fun <T> T?.assertEqual(expected: T, prefix: String = "") = assert(this == expected) {
"${prefix}Expected $expected, got $this"
}
}
inline fun <reified T : Any> Any?.assertIsInstance(name: String): T =
this.let { it as? T } ?: throw AssertionError("$name should not be null")
fun List<DocumentationNode>.commentsToString(): String =
this.flatMap { it.children }.joinToString(separator = "\n") { it.root.docTagSummary() }
fun TagWrapper.text(): String = when (val t = this) {
is NamedTagWrapper -> "${t.name}: [${t.root.text()}]"
else -> t.root.text()
}
fun DocTag.text(): String = when (val t = this) {
is Text -> t.body
is Code -> t.children.joinToString("\n") { it.text() }
is P -> t.children.joinToString(separator = "\n") { it.text() }
else -> t.toString()
}
fun <T : Documentable> T?.comments(): String = docs().map { it.text() }
.joinToString(separator = "\n") { it }
fun <T> T?.assertNotNull(name: String = ""): T = this ?: throw AssertionError("$name should not be null")
fun <T : Documentable> T?.docs() = this?.documentation.orEmpty().values.flatMap { it.children }
val DClass.supers
get() = supertypes.flatMap{it.component2()}
|
