Generating test files under different package

thsaravana · May 20, 2020 · 136bebf · 136bebf
1 parent 4c5e1ab
commit 136bebf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/build.gradle b/build.gradle
@@ -4,7 +4,7 @@ plugins {
 }
 
 group 'com.madrapps'
-version '1.0'
+version '1.1'
 
 repositories {
     mavenCentral()

diff --git a/src/main/kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt b/src/main/kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt
@@ -29,7 +29,7 @@ class DaggerProcessor : AbstractProcessor() {
         roundEnv.getElementsAnnotatedWith(GenerateTest::class.java).forEach { element ->
             if (element is Symbol.ClassSymbol) {
                 val ann: GenerateTest = element.getAnnotation(GenerateTest::class.java)
-                val packageName = element.owner.qualifiedName.toString()
+                val packageName = element.owner.qualifiedName.toString() + "." + ann.directory
                 val className = element.simpleName.toString()
                 val injectFile = File(testDateFile, ann.directory)
 
@@ -41,6 +41,8 @@ class DaggerProcessor : AbstractProcessor() {
                             """
 package $packageName
 
+import com.madrapps.dagger.$className
+
 class $testClassName : $className() {
 ${getTestMethods(folder)}
 }