diff options
Diffstat (limited to 'test-utils/src/main/kotlin/com/google/devtools')
-rw-r--r-- | test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt index 3fd6634a..68e263c6 100644 --- a/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt +++ b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotationArgumentProcessor.kt @@ -30,16 +30,15 @@ class AnnotationArgumentProcessor : AbstractTestProcessor() { override fun process(resolver: Resolver): List<KSAnnotated> { resolver.getSymbolsWithAnnotation("Bar", true).forEach { - val annotation = it.annotations.single() - annotation.arguments.forEach { it.accept(visitor, Unit) } + it.annotations.forEach { it.arguments.forEach { it.accept(visitor, Unit) } } } - val C = resolver.getClassDeclarationByName("C")!! - C.annotations.first().arguments.forEach { results.add(it.value.toString()) } - val ThrowsClass = resolver.getClassDeclarationByName("ThrowsClass")!! - ThrowsClass.declarations.filter { + val C = resolver.getClassDeclarationByName("C") + C?.annotations?.first()?.arguments?.forEach { results.add(it.value.toString()) } + val ThrowsClass = resolver.getClassDeclarationByName("ThrowsClass") + ThrowsClass?.declarations?.filter { it.simpleName.asString() == "throwsException" - }.forEach { + }?.forEach { it.annotations.single().annotationType.resolve().declaration.let { results.add(it.toString()) } |