5 files changed, 170 insertions, 0 deletions
diff --git a/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.dmg b/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.dmg
new file mode 100644
index 000000000..f7f36f631
--- /dev/null
+++ b/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.dmg
diff --git a/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.tar.bz2 b/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.tar.bz2
new file mode 100644
index 000000000..cb046a0e7
--- /dev/null
+++ b/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.tar.bz2
diff --git a/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.zip b/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.zip
new file mode 100644
index 000000000..7c3f61a5e
--- /dev/null
+++ b/testing/mozbase/mozinstall/tests/Installer-Stubs/firefox.zip
diff --git a/testing/mozbase/mozinstall/tests/manifest.ini b/testing/mozbase/mozinstall/tests/manifest.ini
new file mode 100644
index 000000000..528fdea7b
--- /dev/null
+++ b/testing/mozbase/mozinstall/tests/manifest.ini
@@ -0,0 +1 @@
+[test.py]
diff --git a/testing/mozbase/mozinstall/tests/test.py b/testing/mozbase/mozinstall/tests/test.py
new file mode 100644
index 000000000..b4c53bb42
--- /dev/null
+++ b/testing/mozbase/mozinstall/tests/test.py
@@ -0,0 +1,169 @@
+#!/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 mozinfo
+import mozinstall
+import mozfile
+import os
+import tempfile
+import unittest
+
+# Store file location at load time
+here = os.path.dirname(os.path.abspath(__file__))
+
+
+class TestMozInstall(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        """ Setting up stub installers """
+        cls.dmg = os.path.join(here, 'Installer-Stubs', 'firefox.dmg')
+        # XXX: We have removed firefox.exe since it is not valid for mozinstall 1.12 and higher
+        # Bug 1157352 - We should grab a firefox.exe from the build process or download it
+        cls.exe = os.path.join(here, 'Installer-Stubs', 'firefox.exe')
+        cls.zipfile = os.path.join(here, 'Installer-Stubs', 'firefox.zip')
+        cls.bz2 = os.path.join(here, 'Installer-Stubs', 'firefox.tar.bz2')
+
+    def setUp(self):
+        self.tempdir = tempfile.mkdtemp()
+
+    def tearDown(self):
+        mozfile.rmtree(self.tempdir)
+
+    @unittest.skipIf(mozinfo.isWin, "Bug 1157352 - We need a new firefox.exe "
+                     "for mozinstall 1.12 and higher.")
+    def test_get_binary(self):
+        """ Test mozinstall's get_binary method """
+
+        if mozinfo.isLinux:
+            installdir = mozinstall.install(self.bz2, self.tempdir)
+            binary = os.path.join(installdir, 'firefox')
+            self.assertEqual(binary, mozinstall.get_binary(installdir, 'firefox'))
+
+        elif mozinfo.isWin:
+            installdir_exe = mozinstall.install(self.exe,
+                                                os.path.join(self.tempdir, 'exe'))
+            binary_exe = os.path.join(installdir_exe, 'core', 'firefox.exe')
+            self.assertEqual(binary_exe, mozinstall.get_binary(installdir_exe,
+                                                               'firefox'))
+
+            installdir_zip = mozinstall.install(self.zipfile,
+                                                os.path.join(self.tempdir, 'zip'))
+            binary_zip = os.path.join(installdir_zip, 'firefox.exe')
+            self.assertEqual(binary_zip, mozinstall.get_binary(installdir_zip,
+                                                               'firefox'))
+
+        elif mozinfo.isMac:
+            installdir = mozinstall.install(self.dmg, self.tempdir)
+            binary = os.path.join(installdir, 'Contents', 'MacOS', 'firefox')
+            self.assertEqual(binary, mozinstall.get_binary(installdir, 'firefox'))
+
+    def test_get_binary_error(self):
+        """ Test an InvalidBinary error is raised """
+
+        tempdir_empty = tempfile.mkdtemp()
+        self.assertRaises(mozinstall.InvalidBinary, mozinstall.get_binary,
+                          tempdir_empty, 'firefox')
+        mozfile.rmtree(tempdir_empty)
+
+    @unittest.skipIf(mozinfo.isWin, "Bug 1157352 - We need a new firefox.exe "
+                     "for mozinstall 1.12 and higher.")
+    def test_is_installer(self):
+        """ Test we can identify a correct installer """
+
+        if mozinfo.isLinux:
+            self.assertTrue(mozinstall.is_installer(self.bz2))
+
+        if mozinfo.isWin:
+            # test zip installer
+            self.assertTrue(mozinstall.is_installer(self.zipfile))
+
+            # test exe installer
+            self.assertTrue(mozinstall.is_installer(self.exe))
+
+            try:
+                # test stub browser file
+                # without pefile on the system this test will fail
+                import pefile  # noqa
+                stub_exe = os.path.join(here, 'build_stub', 'firefox.exe')
+                self.assertFalse(mozinstall.is_installer(stub_exe))
+            except ImportError:
+                pass
+
+        if mozinfo.isMac:
+            self.assertTrue(mozinstall.is_installer(self.dmg))
+
+    def test_invalid_source_error(self):
+        """ Test InvalidSource error is raised with an incorrect installer """
+
+        if mozinfo.isLinux:
+            self.assertRaises(mozinstall.InvalidSource, mozinstall.install,
+                              self.dmg, 'firefox')
+
+        elif mozinfo.isWin:
+            self.assertRaises(mozinstall.InvalidSource, mozinstall.install,
+                              self.bz2, 'firefox')
+
+        elif mozinfo.isMac:
+            self.assertRaises(mozinstall.InvalidSource, mozinstall.install,
+                              self.bz2, 'firefox')
+
+    @unittest.skipIf(mozinfo.isWin, "Bug 1157352 - We need a new firefox.exe "
+                     "for mozinstall 1.12 and higher.")
+    def test_install(self):
+        """ Test mozinstall's install capability """
+
+        if mozinfo.isLinux:
+            installdir = mozinstall.install(self.bz2, self.tempdir)
+            self.assertEqual(os.path.join(self.tempdir, 'firefox'), installdir)
+
+        elif mozinfo.isWin:
+            installdir_exe = mozinstall.install(self.exe,
+                                                os.path.join(self.tempdir, 'exe'))
+            self.assertEqual(os.path.join(self.tempdir, 'exe', 'firefox'),
+                             installdir_exe)
+
+            installdir_zip = mozinstall.install(self.zipfile,
+                                                os.path.join(self.tempdir, 'zip'))
+            self.assertEqual(os.path.join(self.tempdir, 'zip', 'firefox'),
+                             installdir_zip)
+
+        elif mozinfo.isMac:
+            installdir = mozinstall.install(self.dmg, self.tempdir)
+            self.assertEqual(os.path.join(os.path.realpath(self.tempdir),
+                                          'FirefoxStub.app'), installdir)
+
+    @unittest.skipIf(mozinfo.isWin, "Bug 1157352 - We need a new firefox.exe "
+                     "for mozinstall 1.12 and higher.")
+    def test_uninstall(self):
+        """ Test mozinstall's uninstall capabilites """
+        # Uninstall after installing
+
+        if mozinfo.isLinux:
+            installdir = mozinstall.install(self.bz2, self.tempdir)
+            mozinstall.uninstall(installdir)
+            self.assertFalse(os.path.exists(installdir))
+
+        elif mozinfo.isWin:
+            # Exe installer for Windows
+            installdir_exe = mozinstall.install(self.exe,
+                                                os.path.join(self.tempdir, 'exe'))
+            mozinstall.uninstall(installdir_exe)
+            self.assertFalse(os.path.exists(installdir_exe))
+
+            # Zip installer for Windows
+            installdir_zip = mozinstall.install(self.zipfile,
+                                                os.path.join(self.tempdir, 'zip'))
+            mozinstall.uninstall(installdir_zip)
+            self.assertFalse(os.path.exists(installdir_zip))
+
+        elif mozinfo.isMac:
+            installdir = mozinstall.install(self.dmg, self.tempdir)
+            mozinstall.uninstall(installdir)
+            self.assertFalse(os.path.exists(installdir))
+
+if __name__ == '__main__':
+    unittest.main()