Kapt: Add tests with Filer.

JavaFileObject.getName() now returns a path as required. (cherry picked from commit c611f9b)
2016-08-31 19:40:51 +03:00
parent 824b778a7b
commit 743be477ec
9 changed files with 131 additions and 16 deletions
@@ -92,7 +92,7 @@ abstract class AbstractProcessorTest : AbstractBytecodeTextTest() {
            vararg supportedAnnotations: String
    ) = testAP(false, name, { set, roundEnv, env -> fail("Should not run") }, *supportedAnnotations)

-    private fun testAP(
+    protected fun testAP(
            shouldRun: Boolean,
            name: String,
            process: (Set<TypeElement>, RoundEnvironment, ProcessingEnvironment) -> Unit,
@@ -0,0 +1,79 @@
+/*
+ * Copyright 2010-2016 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.annotation.processing.test.processor
+
+import javax.lang.model.element.TypeElement
+import javax.tools.JavaFileObject
+
+class TestsWithFiler : AbstractProcessorTest() {
+    override val testDataDir = "plugins/annotation-processing/testData/withFiler"
+    
+    fun testSimple() = filerTest("Simple", 1) { files ->
+        assertEquals(listOf("generated/Inject2Test.java"), files)
+    }
+    
+    fun testMethodsFields() = filerTest("MethodsFields", 1) { files ->
+        assertEquals(listOf("generated/InjectmyField.java",
+                            "generated/InjectmyFunc.java"), files)
+    }
+    
+    fun testTwoRounds() = filerTest("TwoRounds", 2) { files ->
+        assertEquals(listOf("generated/Inject2InjectTest.java",
+                            "generated/InjectTest.java", 
+                            "generated/InjectmyFunc.java"), files)
+    }
+    
+    fun testOneRound() = filerTest("OneRound", 1) { files ->
+        assertEquals(listOf("generated/Inject2Test.java",
+                            "generated/Inject2myFunc.java",
+                            "generated/InjectmyFunc.java"), files)
+    }
+    
+    fun testZeroRounds() = filerTest("ZeroRounds", 0) { files ->
+        assertEquals(emptyList<String>(), files)
+    }
+    
+    private fun filerTest(
+            fileName: String,
+            roundCount: Int,
+            check: (List<String>) -> Unit) {
+        val filesCreated = mutableSetOf<JavaFileObject>()
+        var actualRoundCount = 0
+
+        testAP(roundCount > 0, fileName, { set, roundEnv, env ->
+            if (!roundEnv.processingOver()) actualRoundCount++
+            
+            for (anno in set) {
+                val annotated = roundEnv.getElementsAnnotatedWith(anno)
+                annotated.forEach { el ->
+                    val className = anno.simpleName.toString() + el.simpleName.toString()
+                    env.filer.createSourceFile("generated.$className").apply {
+                        filesCreated += this
+                        val inject2 = if (el is TypeElement && anno.simpleName.toString() == "Inject") "@Inject2\n" else ""
+                        openWriter().use {
+                            it.write("package generated;\n" + inject2 +
+                                     "public class $className {}")
+                        }
+                    }
+                }
+            }
+        }, "Inject", "Inject2")
+
+        assertEquals(roundCount, actualRoundCount)
+        check(filesCreated.map { it.name }.sorted())
+    }
+}