Test enum values

kuanyingchou · Apr 11, 2024 · d8fa1e7 · d8fa1e7
1 parent cf3b401
commit d8fa1e7
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 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
@@ -29,8 +29,16 @@ class AnnotationArgumentProcessor : AbstractTestProcessor() {
     val visitor = ArgumentVisitor()
 
     override fun process(resolver: Resolver): List<KSAnnotated> {
-        resolver.getSymbolsWithAnnotation("Bar", true).forEach {
-            it.annotations.forEach { it.arguments.forEach { it.accept(visitor, Unit) } }
+        listOf("Bar", "BarJava").forEach { annotation ->
+            resolver.getSymbolsWithAnnotation(annotation, true).forEach {
+                println(">>> $it")
+                it.annotations.forEach { it.arguments.forEach { it.accept(visitor, Unit) } }
+            }
+        }
+        resolver.getClassDeclarationByName("TestLib")!!.let { cls ->
+            cls.annotations.forEach {
+                it.arguments.forEach { it.accept(visitor, Unit) }
+            }
         }
 
         val C = resolver.getClassDeclarationByName("C")
@@ -53,8 +61,14 @@ class AnnotationArgumentProcessor : AbstractTestProcessor() {
     inner class ArgumentVisitor : KSVisitorVoid() {
         override fun visitValueArgument(valueArgument: KSValueArgument, data: Unit) {
             if (valueArgument.value is KSType) {
+                println("adding KSType: ${valueArgument.value} : ${valueArgument.value!!::class}")
                 results.add((valueArgument.value as KSType).declaration.toString())
             } else {
+                println("adding non-KSType: ${valueArgument.value} : ${
+                    valueArgument.value?.let {
+                        it::class
+                    } ?: "null"
+                }")
                 results.add(valueArgument.value.toString())
             }
         }

diff --git a/test-utils/testData/api/annotationValue_kt.kt b/test-utils/testData/api/annotationValue_kt.kt
@@ -33,10 +33,28 @@
 // G
 // ONE
 // 31
+// 42
+// JavaEnum
+// 42
+// null
 // Throws
 // END
-// FILE: a.kt
+// MODULE: module1
+// FILE: JavaEnum.java
+enum JavaEnum { ONE, TWO, THREE }
+@interface BarJava {
+    int i();
+    JavaEnum argJavaEnum();
+}
+// FILE: TestLib.java
+@BarJava(
+    i = 42,
+    argJavaEnum = JavaEnum.ONE
+)
+class TestLib {}
 
+// MODULE: main(module1)
+// FILE: a.kt
 enum class RGB {
     R, G, B
 }
@@ -90,7 +108,9 @@ fun Fun() {
     )
     class Local
 }
-
-// FILE: JavaEnum.java
-
-enum JavaEnum { ONE, TWO, THREE }
+// FILE: Test.java
+@BarJava(
+    i = 42,
+    argJavaEnum = JavaEnum.ONE
+)
+class Test {}