summaryrefslogtreecommitdiffstats
path: root/python/mozperftest/mozperftest/test/__init__.py
blob: c7d7d6e04950882791cb5a3798aa4661cb3223fb (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 mozperftest.layers import Layers
from mozperftest.test.androidlog import AndroidLog
from mozperftest.test.browsertime import BrowsertimeRunner
from mozperftest.test.webpagetest import WebPageTest
from mozperftest.test.xpcshell import XPCShell


def get_layers():
    return BrowsertimeRunner, AndroidLog, XPCShell, WebPageTest


def pick_test(env, flavor, mach_cmd):
    if flavor == "xpcshell":
        return Layers(env, mach_cmd, (XPCShell,))
    if flavor == "desktop-browser":
        return Layers(env, mach_cmd, (BrowsertimeRunner,))
    if flavor == "mobile-browser":
        return Layers(env, mach_cmd, (BrowsertimeRunner, AndroidLog))
    if flavor == "webpagetest":
        return Layers(env, mach_cmd, (WebPageTest,))

    raise NotImplementedError(flavor)