summaryrefslogtreecommitdiffstats
path: root/testing/marionette/harness/marionette_harness/tests/unit/test_geckoinstance.py
blob: 3d35217bc49bb634d6eebbc53dd4d8b9dc703045 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# 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/.

from marionette_driver.geckoinstance import apps, GeckoInstance

from marionette_harness import MarionetteTestCase


class TestGeckoInstance(MarionetteTestCase):
    def test_create(self):
        """Test that the correct gecko instance is determined."""
        for app in apps:
            # If app has been specified we directly return the appropriate instance class
            self.assertEqual(type(GeckoInstance.create(app=app, bin="n/a")), apps[app])

        # Unknown applications and binaries should fail
        self.assertRaises(
            NotImplementedError,
            GeckoInstance.create,
            app="n/a",
            bin=self.marionette.bin,
        )
        self.assertRaises(NotImplementedError, GeckoInstance.create, bin="n/a")
        self.assertRaises(NotImplementedError, GeckoInstance.create, bin=None)