From 136bebfb8d59d065b6242cc306f6c849c44a27d4 Mon Sep 17 00:00:00 2001
From: thsaravana <th.saravana@gmail.com>
Date: Wed, 20 May 2020 15:27:49 +0530
Subject: [PATCH] Generating test files under different package

---
 build.gradle                                                  | 2 +-
 .../kotlin/com/madrapps/daggerprocessor/DaggerProcessor.kt    | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/build.gradle b/build.gradle
index baf2819..90a7289 100644
--- 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
index c4728e2..131b62f 100644
--- 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)}
 }