| Index: tools/telemetry/telemetry/unittest/run_tests_unittest.py
|
| diff --git a/tools/telemetry/telemetry/unittest/run_tests_unittest.py b/tools/telemetry/telemetry/unittest/run_tests_unittest.py
|
| index 20bc88ddb4a55ff95c795f525c6952b75c88dd58..ee02d93d25b305db2ff6b06d0214330e0fc9b544 100644
|
| --- a/tools/telemetry/telemetry/unittest/run_tests_unittest.py
|
| +++ b/tools/telemetry/telemetry/unittest/run_tests_unittest.py
|
| @@ -32,8 +32,9 @@ class MockPlatform(object):
|
| class RunTestsUnitTest(unittest.TestCase):
|
|
|
| def setUp(self):
|
| - self.suite = run_tests.Discover(
|
| - util.GetTelemetryDir(), util.GetTelemetryDir(), 'disabled_cases.py')
|
| + self.suite = unittest.TestSuite()
|
| + self.suite.addTests(run_tests.Discover(
|
| + util.GetTelemetryDir(), util.GetTelemetryDir(), 'disabled_cases.py'))
|
|
|
| def _GetEnabledTests(self, browser_type, os_name, os_version_name,
|
| supports_tab_control):
|
| @@ -45,8 +46,8 @@ class RunTestsUnitTest(unittest.TestCase):
|
|
|
| enabled_tests = set()
|
| for i in run_tests.FilterSuite(self.suite, MockPredicate)._tests:
|
| - for j in i._tests:
|
| - for k in j._tests:
|
| + for j in i:
|
| + for k in j:
|
| enabled_tests.add(k._testMethodName)
|
| return enabled_tests
|
|
|
|
|