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 |