/
RsMacroExpansionErrorTestBase.kt
64 lines (54 loc) · 2.13 KB
/
RsMacroExpansionErrorTestBase.kt
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
/*
* Use of this source code is governed by the MIT license that can be
* found in the LICENSE file.
*/
package org.rust.lang.core.macros
import org.intellij.lang.annotations.Language
import org.rust.RsTestBase
import org.rust.fileTreeFromText
import org.rust.lang.core.macros.errors.GetMacroExpansionError
import org.rust.lang.core.psi.ext.RsPossibleMacroCall
import org.rust.lang.core.psi.ext.descendantsOfType
import org.rust.lang.core.psi.ext.expansionResult
import org.rust.lang.core.psi.ext.isMacroCall
import org.rust.stdext.RsResult
abstract class RsMacroExpansionErrorTestBase : RsTestBase() {
protected inline fun <reified T : GetMacroExpansionError> checkError(
@Language("Rust") code: String
) {
checkError(code, T::class.java)
}
protected fun checkError(code: String, errorClass: Class<*>) {
InlineFile(code)
doCheck(errorClass)
}
protected inline fun <reified T : GetMacroExpansionError> checkErrorByTree(
@Language("Rust") code: String
) {
checkError(code, T::class.java)
}
protected fun checkErrorByTree(code: String, errorClass: Class<*>) {
fileTreeFromText(code).createAndOpenFileWithCaretMarker()
doCheck(errorClass)
}
private fun doCheck(errorClass: Class<*>) {
val markers = findElementsWithDataAndOffsetInEditor<RsPossibleMacroCall>()
val (macro, expectedErrorMessage) = if (markers.isEmpty()) {
myFixture.file
.descendantsOfType<RsPossibleMacroCall>()
.single { it.isMacroCall } to null
} else {
val (macro, message, _) = markers.single()
check(macro.isMacroCall)
macro to message
}
val err = when (val result = macro.expansionResult) {
is RsResult.Err -> result.err
is RsResult.Ok -> error("Expected a macro expansion error, got a successfully expanded macro")
}
check(errorClass.isInstance(err)) { "Expected error $errorClass, got $err" }
if (expectedErrorMessage != null) {
assertEquals(expectedErrorMessage, err.toUserViewableMessage())
}
}
}