Test code generation in correct folder structure

arrow-kt · Jun 6, 2018 · a3f51d8 · a3f51d8
1 parent 14ded90
commit a3f51d8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/modules/core/arrow-annotations-processor/src/test/java/arrow/ap/tests/APTest.kt b/modules/core/arrow-annotations-processor/src/test/java/arrow/ap/tests/APTest.kt
@@ -11,7 +11,8 @@ import java.io.File
 import java.nio.file.Paths
 
 abstract class APTest(
-  val pckg: String
+  private val pckg: String,
+  private val enforcePackage: Boolean = true
 ) : StringSpec() {
 
   fun testProcessor(
@@ -56,10 +57,11 @@ abstract class APTest(
           assertThat(compilation)
             .succeeded()
 
-          generationDir.listFiles().size shouldBe 1
+          val targetDir = if (enforcePackage) File("${generationDir.absolutePath}/${pckg.replace(".", "/")}") else generationDir
+          targetDir.listFiles().size shouldBe 1
 
           val expected = File(expectedDir, dest).readText()
-          val actual = generationDir.listFiles()[0].readText()
+          val actual = targetDir.listFiles()[0].readText()
 
           actual shouldBe expected
 

diff --git a/modules/core/arrow-annotations-processor/src/test/java/arrow/ap/tests/RenzuTest.kt b/modules/core/arrow-annotations-processor/src/test/java/arrow/ap/tests/RenzuTest.kt
@@ -3,7 +3,7 @@ package arrow.ap.tests
 import arrow.renzu.RenzuProcessor
 import java.io.File
 
-class RenzuTest : APTest("arrow.ap.objects.renzu") {
+class RenzuTest : APTest("arrow.ap.objects.renzu", false) {
 
   init {
     testProcessor(AnnotationProcessor(