Index: telemetry/telemetry/testing/serially_executed_browser_test_case.py |
diff --git a/telemetry/telemetry/testing/serially_executed_browser_test_case.py b/telemetry/telemetry/testing/serially_executed_browser_test_case.py |
index 1bb1f6e72584626c1c300160c226ec0af1869945..3a469cb3cfdd9ee49c86b43efab8d7df81c9f906 100644 |
--- a/telemetry/telemetry/testing/serially_executed_browser_test_case.py |
+++ b/telemetry/telemetry/testing/serially_executed_browser_test_case.py |
@@ -8,7 +8,7 @@ |
from py_utils import cloud_storage |
from telemetry.internal.browser import browser_finder |
-from telemetry.testing import browser_test_context |
+from telemetry.testing import options_for_unittests |
from telemetry.util import wpr_modes |
@@ -30,12 +30,8 @@ |
pass |
@classmethod |
- def SetUpProcess(cls): |
- """ Set up testing logic before running the test case. |
- This is guaranteed to be called only once for all the tests before the test |
- suite runs. |
- """ |
- cls._finder_options = browser_test_context.GetCopy().finder_options |
+ def setUpClass(cls): |
+ cls._finder_options = options_for_unittests.GetCopy() |
cls.platform = None |
cls.browser = None |
cls._browser_to_create = None |
@@ -96,12 +92,7 @@ |
cls.browser = None |
@classmethod |
- def TearDownProcess(cls): |
- """ Tear down the testing logic after running the test cases. |
- This is guaranteed to be called only once for all the tests after the test |
- suite finishes running. |
- """ |
- |
+ def tearDownClass(cls): |
if cls.platform: |
cls.platform.StopAllLocalServers() |
cls.platform.network_controller.Close() |
@@ -151,24 +142,15 @@ |
test cases to be run. |
""" |
suite = unittest.TestSuite() |
- test_context = browser_test_context.GetCopy() |
- if not test_context: |
+ finder_options = options_for_unittests.GetCopy() |
+ if not hasattr(finder_options, 'browser_test_runner_running'): |
return suite |
for _, obj in inspect.getmembers(module): |
if (inspect.isclass(obj) and |
issubclass(obj, SeriallyExecutedBrowserTestCase)): |
- # We bail out early if this class doesn't match the targeted |
- # test_class in test_context to avoid calling GenerateTestCases |
- # for tests that we don't intend to run. This is to avoid possible errors |
- # in GenerateTestCases as the test class may define custom options in |
- # the finder_options object, and hence would raise error if they can't |
- # find their custom options in finder_options object. |
- if test_context.test_class != obj: |
- continue |
for test in GenerateTestCases( |
- test_class=obj, finder_options=test_context.finder_options): |
- if test.id() in test_context.test_case_ids_to_run: |
- suite.addTest(test) |
+ test_class=obj, finder_options=finder_options): |
+ suite.addTest(test) |
return suite |