com.malinskiy.marathon.android.ApkParser.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vendor-android Show documentation
Android & iOS test runner
There is a newer version: 0.9.1
package com.malinskiy.marathon.android

import com.shazam.axmlparser.AXMLParser
import java.io.File
import java.io.IOException
import java.io.InputStream
import java.util.zip.ZipFile

class ApkParser {
    @Suppress("ComplexMethod",
            "ThrowsCount",
            "TooGenericExceptionThrown",
            "NestedBlockDepth"
    )
    fun parseInstrumentationInfo(apk: File): InstrumentationInfo {
        var apkInputStream: InputStream? = null
        try {
            val zip = ZipFile(apk)
            val entry = zip.getEntry("AndroidManifest.xml")
            apkInputStream = zip.getInputStream(entry)

            val parser = AXMLParser(apkInputStream)
            var eventType = parser.type

            var appPackage: String? = null
            var testPackage: String? = null
            var testRunnerClass: String? = null
            while (eventType != AXMLParser.END_DOCUMENT) {
                if (eventType == AXMLParser.START_TAG) {
                    val parserName = parser.name
                    val isManifest = "manifest" == parserName
                    val isInstrumentation = "instrumentation" == parserName
                    if (isManifest || isInstrumentation) {
                        for (i in 0 until parser.attributeCount) {
                            val parserAttributeName = parser.getAttributeName(i)
                            if (isManifest && "package" == parserAttributeName) {
                                testPackage = parser.getAttributeValueString(i)
                            } else if (isInstrumentation && "targetPackage" == parserAttributeName) {
                                appPackage = parser.getAttributeValueString(i)
                            } else if (isInstrumentation && "name" == parserAttributeName) {
                                testRunnerClass = parser.getAttributeValueString(i)
                            }
                        }
                    }
                }
                eventType = parser.next()
            }

            if (testRunnerClass == null) throw IllegalStateException("Could not find test runner class.")
            if (testPackage == null) throw IllegalStateException("Could not find test application package.")
            if (appPackage == null) throw IllegalStateException("Could not find application package.")

            // Support relative declaration of instrumentation test runner.
            if (testRunnerClass.startsWith(".")) {
                testRunnerClass = testPackage + testRunnerClass
            } else if (!testRunnerClass.contains(".")) {
                testRunnerClass = "$testPackage.$testRunnerClass"
            }

            return InstrumentationInfo(appPackage, testPackage, testRunnerClass)
        } catch (e: IOException) {
            throw RuntimeException("Unable to parse test app AndroidManifest.xml.", e)
        } finally {
            apkInputStream?.close()
        }
    }
}