python/mozperftest/mozperftest/system/__init__.py

   1 # This Source Code Form is subject to the terms of the Mozilla Public
   2 # License, v. 2.0. If a copy of the MPL was not distributed with this
   3 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   4 from mozperftest.layers import Layers
   5 from mozperftest.system.android import AndroidDevice
   6 from mozperftest.system.android_startup import AndroidStartUp
   7 from mozperftest.system.macos import MacosDevice
   8 from mozperftest.system.pingserver import PingServer
   9 from mozperftest.system.profile import Profile
  10 from mozperftest.system.proxy import ProxyRunner
  11
  12
  13 def get_layers():
  14     return PingServer, Profile, ProxyRunner, AndroidDevice, MacosDevice, AndroidStartUp
  15
  16
  17 def pick_system(env, flavor, mach_cmd):
  18     if flavor in ("desktop-browser", "xpcshell", "mochitest"):
  19         return Layers(
  20             env,
  21             mach_cmd,
  22             (
  23                 PingServer,  # needs to come before Profile
  24                 MacosDevice,
  25                 Profile,
  26                 ProxyRunner,
  27             ),
  28         )
  29     if flavor == "mobile-browser":
  30         return Layers(
  31             env, mach_cmd, (Profile, ProxyRunner, AndroidDevice, AndroidStartUp)
  32         )
  33     if flavor == "webpagetest":
  34         return Layers(env, mach_cmd, (Profile,))
  35     raise NotImplementedError(flavor)