telemetry/examples/browser_tests/process_tests.py

# Copyright 2017 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from __future__ import absolute_import
import sys

from telemetry.testing import serially_executed_browser_test_case


class FailIfSetUpProcessCalledTwice(
    serially_executed_browser_test_case.SeriallyExecutedBrowserTestCase):
  count = 0

  @classmethod
  def SetUpProcess(cls):
    cls.count += 1
    if cls.count >= 2:
      assert False, 'This should not be called more than once'

  @classmethod
  def GenerateTestCases_DummyTest(cls, options):
    del options  # Unused.
    for i in range(0, 3):
      yield 'Dummy_%i' % i, ()

  def DummyTest(self):
    pass


class FailIfTearDownProcessCalledTwice(
    serially_executed_browser_test_case.SeriallyExecutedBrowserTestCase):
  count = 0

  @classmethod
  def TearDownProcess(cls):
    cls.count += 1
    if cls.count >= 2:
      assert False, 'This should not be called more than once'

  @classmethod
  def GenerateTestCases_DummyTest(cls, options):
    del options  # Unused.
    for i in range(0, 3):
      yield 'Dummy_%i' % i, ()

  def DummyTest(self):
    pass


def load_tests(loader, tests, pattern): # pylint: disable=invalid-name
  del loader, tests, pattern  # Unused.
  return serially_executed_browser_test_case.LoadAllTestsInModule(
      sys.modules[__name__])