1 files changed, 45 insertions, 0 deletions
diff --git a/testing/mozbase/mozversion/tests/test_apk.py b/testing/mozbase/mozversion/tests/test_apk.py
new file mode 100644
index 0000000000..7ca4d4e068
--- /dev/null
+++ b/testing/mozbase/mozversion/tests/test_apk.py
@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import zipfile
+
+import mozunit
+import pytest
+from mozversion import get_version
+
+"""test getting version information from an android .apk"""
+
+application_changeset = "a" * 40
+platform_changeset = "b" * 40
+
+
+@pytest.fixture(name="apk")
+def fixture_apk(tmpdir):
+    path = str(tmpdir.join("apk.zip"))
+    with zipfile.ZipFile(path, "w") as z:
+        z.writestr(
+            "application.ini", """[App]\nSourceStamp=%s\n""" % application_changeset
+        )
+        z.writestr("platform.ini", """[Build]\nSourceStamp=%s\n""" % platform_changeset)
+        z.writestr("AndroidManifest.xml", "")
+    return path
+
+
+def test_basic(apk):
+    v = get_version(apk)
+    assert v.get("application_changeset") == application_changeset
+    assert v.get("platform_changeset") == platform_changeset
+
+
+def test_with_package_name(apk):
+    with zipfile.ZipFile(apk, "a") as z:
+        z.writestr("package-name.txt", "org.mozilla.fennec")
+    v = get_version(apk)
+    assert v.get("package_name") == "org.mozilla.fennec"
+
+
+if __name__ == "__main__":
+    mozunit.main()