src/tools/telemetry/telemetry/page/actions/all_page_actions.py

   1 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4 import os
   5
   6 from telemetry.core import discover
   7 from telemetry.core import util
   8 from telemetry.page.actions import page_action
   9
  10 _page_action_classes = discover.DiscoverClasses(
  11     os.path.dirname(__file__), util.GetTelemetryDir(), page_action.PageAction)
  12
  13 def GetAllClasses():
  14   return list(_page_action_classes.values())
  15
  16 def FindClassWithName(name):
  17   return _page_action_classes.get(name)
  18
  19 def RegisterClassForTest(name, clazz):
  20   _page_action_classes[name] = clazz