| Index: tools/telemetry/telemetry/page/page_test_runner.py
|
| diff --git a/tools/telemetry/telemetry/page/page_test_runner.py b/tools/telemetry/telemetry/page/page_test_runner.py
|
| index 22a05176d7e1c51644d32f3c0601e0b13d362d7c..2801312866b92125408215cdb90de141ddd42ec7 100644
|
| --- a/tools/telemetry/telemetry/page/page_test_runner.py
|
| +++ b/tools/telemetry/telemetry/page/page_test_runner.py
|
| @@ -8,7 +8,7 @@ from telemetry import test as test_module
|
| from telemetry.core import browser_options
|
| from telemetry.core import discover
|
| from telemetry.core import profile_types
|
| -from telemetry.page import page_test
|
| +from telemetry.page import page_test as page_test_module
|
| from telemetry.page import page_runner
|
| from telemetry.page import page_set
|
|
|
| @@ -29,7 +29,7 @@ class PageTestRunner(object):
|
|
|
| @property
|
| def test_class(self):
|
| - return page_test.PageTest
|
| + return page_test_module.PageTest
|
|
|
| @property
|
| def test_class_name(self):
|
| @@ -138,9 +138,12 @@ class PageTestRunner(object):
|
| test = None
|
| if test_name:
|
| test = test_constructors[test_name]()
|
| - if not isinstance(test, test_module.Test):
|
| - test.AddOutputOptions(self._parser)
|
| - test.AddCommandLineOptions(self._parser)
|
| + if isinstance(test, test_module.Test):
|
| + page_test = test.test()
|
| + else:
|
| + page_test = test
|
| + page_test.AddOutputOptions(self._parser)
|
| + page_test.AddCommandLineOptions(self._parser)
|
| page_runner.AddCommandLineOptions(self._parser)
|
|
|
| _, self._args = self._parser.parse_args()
|
| @@ -165,10 +168,7 @@ class PageTestRunner(object):
|
| if len(self._args) > 2:
|
| self.PrintParseError('Too many arguments.')
|
|
|
| - if isinstance(test, test_module.Test):
|
| - return test.test(), ps
|
| - else:
|
| - return test, ps
|
| + return page_test, ps
|
|
|
| def PrintParseError(self, message):
|
| self._parser.error(message)
|
|
|