Add State for RunTestAndroidCorellium

corellium/adapter/src/main/kotlin/flank/corellium/adapter/ParseApk.kt

@@ -0,0 +1,28 @@
+package flank.corellium.adapter
+
+import com.linkedin.dex.parser.DexParser
+import com.linkedin.dex.parser.TestMethod
+import flank.corellium.api.Apk
+import net.dongliu.apk.parser.ApkFile
+import org.xml.sax.InputSource
+import java.io.StringReader
+import javax.xml.parsers.DocumentBuilderFactory
+
+object ParseApkTestCases : Apk.ParseTestCases, (String) -> List<String> by { path ->
+    DexParser.findTestMethods(path).map(TestMethod::testName)
+}
+
+object ParseApkPackageName : Apk.ParsePackageName, (String) -> String by { path ->
+    ApkFile(path).apkMeta.packageName
+}
+
+object ParseApkInfo : Apk.ParseInfo, (String) -> Apk.Info by { path ->
+    Apk.Info(
+        packageName = ApkFile(path).apkMeta.packageName,
+        testRunner = DocumentBuilderFactory.newInstance()
+            .newDocumentBuilder()
+            .parse(InputSource(StringReader(ApkFile(path).manifestXml)))
+            .getElementsByTagName("instrumentation").item(0).attributes
+            .getNamedItem("android:name").nodeValue
+    )
+}

corellium/adapter/src/test/kotlin/flank/corellium/adapter/ParseApkTest.kt

@@ -0,0 +1,41 @@
+package flank.corellium.adapter
+
+import flank.corellium.api.Apk
+import org.junit.Assert.assertEquals
+import org.junit.Test
+
+private const val TEST_APK_PATH = "../../test_artifacts/master/apk/app-single-success-debug-androidTest.apk"
+
+class ParseApkTest {
+
+    @Test
+    fun parseTestCases() {
+        assertEquals(
+            listOf(
+                "com.example.test_app.InstrumentedTest#ignoredTestWithIgnore",
+                "com.example.test_app.InstrumentedTest#ignoredTestWithSuppress",
+                "com.example.test_app.InstrumentedTest#test",
+            ),
+            ParseApkTestCases(TEST_APK_PATH)
+        )
+    }
+
+    @Test
+    fun parsePackageName() {
+        assertEquals(
+            "com.example.test_app.test",
+            ParseApkPackageName(TEST_APK_PATH)
+        )
+    }
+
+    @Test
+    fun parseInfo() {
+        assertEquals(
+            Apk.Info(
+                packageName = "com.example.test_app.test",
+                testRunner = "androidx.test.runner.AndroidJUnitRunner"
+            ),
+            ParseApkInfo(TEST_APK_PATH)
+        )
+    }
+}

corellium/api/src/main/kotlin/flank/corellium/api/TestApk.kt → corellium/api/src/main/kotlin/flank/corellium/api/Apk.kt

@@ -2,16 +2,13 @@ package flank.corellium.api
 
 /**
  * Structured representation of the parsed test apk file.
- * @property packageName Parsed apk package name.
-
  */
-data class TestApk(
-    val packageName: String,
-    val testCases: List<String>
-) {
-
+object Apk {
 
-    interface Parse : (LocalPath) -> TestApk
+    data class Info(
+        val packageName: String,
+        val testRunner: String,
+    )
 
     /**
      * @return The full list of test methods from parsed apk.
@@ -22,6 +19,11 @@ data class TestApk(
      * @return The package name for given apk.
      */
     interface ParsePackageName: (LocalPath) -> PackageName
+
+    /**
+     * @return The package name for given apk.
+     */
+    interface ParseInfo: (LocalPath) -> Info
 }
 
 /**

corellium/api/src/main/kotlin/flank/corellium/api/CorelliumApi.kt

@@ -8,6 +8,7 @@ class CorelliumApi(
     val invokeAndroidDevices: AndroidInstance.Invoke,
     val installAndroidApps: AndroidApps.Install,
     val executeTest: AndroidTestPlan.Execute,
-    val parseTestCases: TestApk.ParseTestCases,
-    val parsePackageName: TestApk.ParsePackageName,
+    val parseTestCases: Apk.ParseTestCases,
+    val parsePackageName: Apk.ParsePackageName,
+    val parseTestApkInfo: Apk.ParseInfo,
 )