check if the file imports "android.test.ActivityInstrumentationTestCase2"

Author: shehan

src/main/java/ResultsWriter.java

@@ -44,7 +44,7 @@ private void outputTestMethodResults() throws IOException {
         for (FileEntity fileEntity: files) {
             try {
                 for (MethodEntity methodEntity: fileEntity.getMethods()) {
-                    dataLine = new String[9];
+                    dataLine = new String[10];
                     dataLine[0] = fileEntity.getAppName();
                     dataLine[1] = fileEntity.getFilePath();
                     dataLine[2] = fileEntity.getRelativeFilePath();
@@ -54,6 +54,7 @@ private void outputTestMethodResults() throws IOException {
                     dataLine[6] = methodEntity.isHasAnnotation()?"True":"False";
                     dataLine[7] = methodEntity.isHasTestInName()?"True":"False";
                     dataLine[8] = fileEntity.isHasTestInFileName()?"True":"False";
+                    dataLine[9] = fileEntity.getHas_androidtestActivityInstrumentationTestCase2()?"True":"False";
 
                     fileLines.add(dataLine);
                 }
@@ -71,13 +72,13 @@ private void outputTestFileResults() throws IOException {
         CSVWriter writer = new CSVWriter(new FileWriter(testFileCSV.toString()), ',');
         List<String[]> fileLines = new ArrayList<String[]>();
         String[] columnNames  = {"App","FilePath", "RelativeFilePath", "FileName", "TotalMethods","TotalTestMethods","AnnotationCount", "TestsWithoutAnnotationCount", "HasTestInFileName",
-                "junit.framework.Test", "junit.framework.TestCase", "org.junit.Test", "android.test.AndroidTestCase", "android.test.InstrumentationTestCase", "org.junit.Assert"};
+                "junitFrameworkTest", "junitFrameworkTestCase", "orgJunitTest", "androidTestAndroidTestCase", "androidTestInstrumentationTestCase", "orgJunitAssert", "androidTestActivityInstrumentationTestCase2"};
         fileLines.add(columnNames);
 
         String[] dataLine;
         for (FileEntity fileEntity: files) {
             try {
-                dataLine = new String[15];
+                dataLine = new String[16];
                 dataLine[0] = fileEntity.getAppName();
                 dataLine[1] = fileEntity.getFilePath();
                 dataLine[2] = fileEntity.getRelativeFilePath();
@@ -93,6 +94,7 @@ private void outputTestFileResults() throws IOException {
                 dataLine[12] = fileEntity.getHas_androidtestAndroidTestCase()?"True":"False";
                 dataLine[13] = fileEntity.getHas_androidtestInstrumentationTestCase()?"True":"False";
                 dataLine[14] = fileEntity.getHas_orgjunitAssert()?"True":"False";
+                dataLine[15] = fileEntity.getHas_androidtestActivityInstrumentationTestCase2()?"True":"False";
 
                 fileLines.add(dataLine);
             } catch (Exception e) {

src/main/java/TestFileDetector.java

@@ -26,6 +26,7 @@ public TestFileDetector(String absoluteFilePath) throws FileNotFoundException {
         imports.put("android.test.AndroidTestCase",false);
         imports.put("android.test.InstrumentationTestCase",false);
         imports.put("org.junit.Assert",false);
+        imports.put("android.test.ActivityInstrumentationTestCase2",false);
 
         parseFile(absoluteFilePath);
     }
@@ -94,6 +95,9 @@ public void visit(ImportDeclaration n, Void arg) {
                 imports.replace("org.junit.Assert",false,true);
             }
 
+            if (n.getNameAsString().contains("android.test.ActivityInstrumentationTestCase2")){
+                imports.replace("android.test.ActivityInstrumentationTestCase2",false,true);
+            }
 
             super.visit(n, arg);
         }

src/main/java/entity/FileEntity.java

@@ -80,6 +80,11 @@ public boolean getHas_androidtestInstrumentationTestCase()
         return  imports.get("android.test.InstrumentationTestCase");
     }
 
+    public boolean getHas_androidtestActivityInstrumentationTestCase2()
+    {
+        return  imports.get("android.test.ActivityInstrumentationTestCase2");
+    }
+
     public boolean getHas_orgjunitAssert(){
         return imports.get("org.junit.Assert");
     }