1 files changed, 8 insertions, 2 deletions
diff --git a/platform/android/scripts/generate-test-code.js b/platform/android/scripts/generate-test-code.js
index e27de7e40b..64b38e5ace 100644
--- a/platform/android/scripts/generate-test-code.js
+++ b/platform/android/scripts/generate-test-code.js
@@ -36,8 +36,14 @@ for(const subPackage of subPackages) {
     }
 
     for (const activity of activities) {
-      // strip .java from input file
-      const activityName = activity.slice(0, -5);
+      var activityName;
+      if (activity.slice(-5) === '.java') {
+        // .java file
+        activityName = activity.slice(0, -5);
+      } else {
+        // .kt file
+        activityName = activity.slice(0, -3);
+      }
 
       // create path for test file
       const filePath = testBasePath+"/"+subPackage+"/"+activityName+'Test.java';