1 files changed, 25 insertions, 0 deletions
diff --git a/python/mozperftest/mozperftest/test/__init__.py b/python/mozperftest/mozperftest/test/__init__.py
new file mode 100644
index 0000000000..c7d7d6e049
--- /dev/null
+++ b/python/mozperftest/mozperftest/test/__init__.py
@@ -0,0 +1,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)