Handle edge cases of KtAnnotated.annotations

where candidates can be empty.
google · Jul 25, 2024 · de0ed97 · de0ed97
1 parent ebb5ac8
commit de0ed97
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 6 deletions.
diff --git a/kotlin-analysis-api/src/main/kotlin/com/google/devtools/ksp/impl/symbol/kotlin/util.kt b/kotlin-analysis-api/src/main/kotlin/com/google/devtools/ksp/impl/symbol/kotlin/util.kt
@@ -323,6 +323,8 @@ internal fun KtAnnotated.annotations(
     parent: KSNode? = null,
     candidates: List<KaAnnotation> = kaAnnotated.annotations
 ): Sequence<KSAnnotation> {
+    if (candidates.isEmpty())
+        return emptySequence()
     return annotationEntries.filter { !it.isUseSiteTargetAnnotation() }.asSequence().map { annotationEntry ->
         KSAnnotationImpl.getCached(annotationEntry, parent) {
             candidates.single { it.psi == annotationEntry }

diff --git a/kotlin-analysis-api/src/test/kotlin/com/google/devtools/ksp/test/KSPAATest.kt b/kotlin-analysis-api/src/test/kotlin/com/google/devtools/ksp/test/KSPAATest.kt
@@ -97,13 +97,13 @@ class KSPAATest : AbstractKSPAATest() {
     @TestMetadata("annotationValue_java.kt")
     @Test
     fun testAnnotationValue_java() {
-        runTest("../kotlin-analysis-api/testData/annotationValue/java.kt")
+        runTest("../kotlin-analysis-api/testData/annotationValue/annotationValue_java.kt")
     }
 
     @TestMetadata("annotationValue_kt.kt")
     @Test
     fun testAnnotationValue_kt() {
-        runTest("../kotlin-analysis-api/testData/annotationValue/kotlin.kt")
+        runTest("../kotlin-analysis-api/testData/annotationValue/annotationValue_kt.kt")
     }
 
     @TestMetadata("annotationWithArrayValue.kt")

diff --git a/...ysis-api/testData/annotationValue/java.kt → ...a/annotationValue/annotationValue_java.kt b/...ysis-api/testData/annotationValue/java.kt → ...a/annotationValue/annotationValue_java.kt
diff --git a/...is-api/testData/annotationValue/kotlin.kt → ...ata/annotationValue/annotationValue_kt.kt b/...is-api/testData/annotationValue/kotlin.kt → ...ata/annotationValue/annotationValue_kt.kt
@@ -106,6 +106,8 @@ fun Fun() {
     class Local
 }
 
+data class DataClass
+
 // FILE: JavaEnum.java
 
 enum JavaEnum { ONE, TWO, THREE }
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
@@ -19,10 +19,7 @@ package com.google.devtools.ksp.processor
 
 import com.google.devtools.ksp.getClassDeclarationByName
 import com.google.devtools.ksp.processing.Resolver
-import com.google.devtools.ksp.symbol.KSAnnotated
-import com.google.devtools.ksp.symbol.KSType
-import com.google.devtools.ksp.symbol.KSValueArgument
-import com.google.devtools.ksp.symbol.KSVisitorVoid
+import com.google.devtools.ksp.symbol.*
 
 class AnnotationArgumentProcessor : AbstractTestProcessor() {
     val results = mutableListOf<String>()
@@ -37,6 +34,14 @@ class AnnotationArgumentProcessor : AbstractTestProcessor() {
             }
         }
 
+        resolver.getClassDeclarationByName("DataClass")?.let { cls ->
+            cls.declarations.filterIsInstance<KSFunctionDeclaration>().single {
+                it.simpleName.asString() == "copy"
+            }.annotations.forEach {
+                it.arguments
+            }
+        }
+
         resolver.getSymbolsWithAnnotation("Bar", true).forEach {
             it.annotations.forEach { it.arguments.forEach { it.accept(visitor, Unit) } }
         }