| # Copyright (C) 2012 Google Inc. All rights reserved. |
| # |
| # Redistribution and use in source and binary forms, with or without |
| # modification, are permitted provided that the following conditions are |
| # met: |
| # |
| # * Redistributions of source code must retain the above copyright |
| # notice, this list of conditions and the following disclaimer. |
| # * Redistributions in binary form must reproduce the above |
| # copyright notice, this list of conditions and the following disclaimer |
| # in the documentation and/or other materials provided with the |
| # distribution. |
| # * Neither the name of Google Inc. nor the names of its |
| # contributors may be used to endorse or promote products derived from |
| # this software without specific prior written permission. |
| # |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| import StringIO |
| import json |
| import math |
| import unittest2 as unittest |
| |
| from webkitpy.common.host_mock import MockHost |
| from webkitpy.common.system.outputcapture import OutputCapture |
| from webkitpy.layout_tests.port.driver import DriverOutput |
| from webkitpy.layout_tests.port.test import TestDriver |
| from webkitpy.layout_tests.port.test import TestPort |
| from webkitpy.performance_tests.perftest import ChromiumStylePerfTest |
| from webkitpy.performance_tests.perftest import PerfTest |
| from webkitpy.performance_tests.perftest import PerfTestMetric |
| from webkitpy.performance_tests.perftest import PerfTestFactory |
| from webkitpy.performance_tests.perftest import ReplayPerfTest |
| from webkitpy.performance_tests.perftest import SingleProcessPerfTest |
| |
| |
| class MockPort(TestPort): |
| def __init__(self, custom_run_test=None): |
| super(MockPort, self).__init__(host=MockHost(), custom_run_test=custom_run_test) |
| |
| |
| class TestPerfTestMetric(unittest.TestCase): |
| def test_init_set_missing_unit(self): |
| self.assertEqual(PerfTestMetric('Time', iterations=[1, 2, 3, 4, 5]).unit(), 'ms') |
| self.assertEqual(PerfTestMetric('Malloc', iterations=[1, 2, 3, 4, 5]).unit(), 'bytes') |
| self.assertEqual(PerfTestMetric('JSHeap', iterations=[1, 2, 3, 4, 5]).unit(), 'bytes') |
| |
| def test_init_set_time_metric(self): |
| self.assertEqual(PerfTestMetric('Time', 'ms').name(), 'Time') |
| self.assertEqual(PerfTestMetric('Time', 'fps').name(), 'FrameRate') |
| self.assertEqual(PerfTestMetric('Time', 'runs/s').name(), 'Runs') |
| |
| def test_has_values(self): |
| self.assertFalse(PerfTestMetric('Time').has_values()) |
| self.assertTrue(PerfTestMetric('Time', iterations=[1]).has_values()) |
| |
| def test_append(self): |
| metric = PerfTestMetric('Time') |
| metric2 = PerfTestMetric('Time') |
| self.assertFalse(metric.has_values()) |
| self.assertFalse(metric2.has_values()) |
| |
| metric.append_group([1]) |
| self.assertTrue(metric.has_values()) |
| self.assertFalse(metric2.has_values()) |
| self.assertEqual(metric.grouped_iteration_values(), [[1]]) |
| self.assertEqual(metric.flattened_iteration_values(), [1]) |
| |
| metric.append_group([2]) |
| self.assertEqual(metric.grouped_iteration_values(), [[1], [2]]) |
| self.assertEqual(metric.flattened_iteration_values(), [1, 2]) |
| |
| metric2.append_group([3]) |
| self.assertTrue(metric2.has_values()) |
| self.assertEqual(metric.flattened_iteration_values(), [1, 2]) |
| self.assertEqual(metric2.flattened_iteration_values(), [3]) |
| |
| metric.append_group([4, 5]) |
| self.assertEqual(metric.grouped_iteration_values(), [[1], [2], [4, 5]]) |
| self.assertEqual(metric.flattened_iteration_values(), [1, 2, 4, 5]) |
| |
| |
| class TestPerfTest(unittest.TestCase): |
| def _assert_results_are_correct(self, test, output): |
| test.run_single = lambda driver, path, time_out_ms: output |
| self.assertTrue(test._run_with_driver(None, None)) |
| self.assertEqual(test._metrics.keys(), ['Time']) |
| self.assertEqual(test._metrics['Time'].flattened_iteration_values(), [1080, 1120, 1095, 1101, 1104]) |
| |
| def test_parse_output(self): |
| output = DriverOutput(""" |
| Running 20 times |
| Ignoring warm-up run (1115) |
| |
| Time: |
| values 1080, 1120, 1095, 1101, 1104 ms |
| avg 1100 ms |
| median 1101 ms |
| stdev 14.50862 ms |
| min 1080 ms |
| max 1120 ms |
| """, image=None, image_hash=None, audio=None) |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| try: |
| test = PerfTest(MockPort(), 'some-test', '/path/some-dir/some-test') |
| self._assert_results_are_correct(test, output) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, '') |
| |
| def test_parse_output_with_failing_line(self): |
| output = DriverOutput(""" |
| Running 20 times |
| Ignoring warm-up run (1115) |
| |
| some-unrecognizable-line |
| |
| Time: |
| values 1080, 1120, 1095, 1101, 1104 ms |
| avg 1100 ms |
| median 1101 ms |
| stdev 14.50862 ms |
| min 1080 ms |
| max 1120 ms |
| """, image=None, image_hash=None, audio=None) |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| try: |
| test = PerfTest(MockPort(), 'some-test', '/path/some-dir/some-test') |
| test.run_single = lambda driver, path, time_out_ms: output |
| self.assertFalse(test._run_with_driver(None, None)) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, 'ERROR: some-unrecognizable-line\n') |
| |
| def test_parse_output_with_description(self): |
| output = DriverOutput(""" |
| Description: this is a test description. |
| |
| Running 20 times |
| Ignoring warm-up run (1115) |
| |
| Time: |
| values 1080, 1120, 1095, 1101, 1104 ms |
| avg 1100 ms |
| median 1101 ms |
| stdev 14.50862 ms |
| min 1080 ms |
| max 1120 ms""", image=None, image_hash=None, audio=None) |
| test = PerfTest(MockPort(), 'some-test', '/path/some-dir/some-test') |
| self._assert_results_are_correct(test, output) |
| self.assertEqual(test.description(), 'this is a test description.') |
| |
| def test_ignored_stderr_lines(self): |
| test = PerfTest(MockPort(), 'some-test', '/path/some-dir/some-test') |
| output_with_lines_to_ignore = DriverOutput('', image=None, image_hash=None, audio=None, error=""" |
| Unknown option: --foo-bar |
| Should not be ignored |
| [WARNING:proxy_service.cc] bad moon a-rising |
| [WARNING:chrome.cc] Something went wrong |
| [INFO:SkFontHost_android.cpp(1158)] Use Test Config File Main /data/local/tmp/drt/android_main_fonts.xml, Fallback /data/local/tmp/drt/android_fallback_fonts.xml, Font Dir /data/local/tmp/drt/fonts/ |
| [ERROR:main.cc] The sky has fallen""") |
| test._filter_output(output_with_lines_to_ignore) |
| self.assertEqual(output_with_lines_to_ignore.error, |
| "Should not be ignored\n" |
| "[WARNING:chrome.cc] Something went wrong\n" |
| "[ERROR:main.cc] The sky has fallen") |
| |
| def test_parse_output_with_subtests(self): |
| output = DriverOutput(""" |
| Running 20 times |
| some test: [1, 2, 3, 4, 5] |
| other test = else: [6, 7, 8, 9, 10] |
| Ignoring warm-up run (1115) |
| |
| Time: |
| values 1080, 1120, 1095, 1101, 1104 ms |
| avg 1100 ms |
| median 1101 ms |
| stdev 14.50862 ms |
| min 1080 ms |
| max 1120 ms |
| """, image=None, image_hash=None, audio=None) |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| try: |
| test = PerfTest(MockPort(), 'some-test', '/path/some-dir/some-test') |
| self._assert_results_are_correct(test, output) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, '') |
| |
| |
| class TestSingleProcessPerfTest(unittest.TestCase): |
| def test_use_only_one_process(self): |
| called = [0] |
| |
| def run_single(driver, path, time_out_ms): |
| called[0] += 1 |
| return DriverOutput(""" |
| Running 20 times |
| Ignoring warm-up run (1115) |
| |
| Time: |
| values 1080, 1120, 1095, 1101, 1104 ms |
| avg 1100 ms |
| median 1101 ms |
| stdev 14.50862 ms |
| min 1080 ms |
| max 1120 ms""", image=None, image_hash=None, audio=None) |
| |
| test = SingleProcessPerfTest(MockPort(), 'some-test', '/path/some-dir/some-test') |
| test.run_single = run_single |
| self.assertTrue(test.run(0)) |
| self.assertEqual(called[0], 1) |
| |
| |
| class TestReplayPerfTest(unittest.TestCase): |
| class ReplayTestPort(MockPort): |
| def __init__(self, custom_run_test=None): |
| |
| class ReplayTestDriver(TestDriver): |
| def run_test(self, text_input, stop_when_done): |
| return custom_run_test(text_input, stop_when_done) if custom_run_test else None |
| |
| self._custom_driver_class = ReplayTestDriver |
| super(self.__class__, self).__init__() |
| |
| def _driver_class(self): |
| return self._custom_driver_class |
| |
| class MockReplayServer(object): |
| def __init__(self, wait_until_ready=True): |
| self.wait_until_ready = lambda: wait_until_ready |
| |
| def stop(self): |
| pass |
| |
| def _add_file(self, port, dirname, filename, content=True): |
| port.host.filesystem.maybe_make_directory(dirname) |
| port.host.filesystem.write_binary_file(port.host.filesystem.join(dirname, filename), content) |
| |
| def _setup_test(self, run_test=None): |
| test_port = self.ReplayTestPort(run_test) |
| self._add_file(test_port, '/path/some-dir', 'some-test.replay', 'http://some-test/') |
| test = ReplayPerfTest(test_port, 'some-test.replay', '/path/some-dir/some-test.replay') |
| test._start_replay_server = lambda archive, record: self.__class__.MockReplayServer() |
| return test, test_port |
| |
| def test_run_single(self): |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| |
| loaded_pages = [] |
| |
| def run_test(test_input, stop_when_done): |
| if test_input.test_name == test.force_gc_test: |
| loaded_pages.append(test_input) |
| return |
| if test_input.test_name != "about:blank": |
| self.assertEqual(test_input.test_name, 'http://some-test/') |
| loaded_pages.append(test_input) |
| self._add_file(port, '/path/some-dir', 'some-test.wpr', 'wpr content') |
| return DriverOutput('actual text', 'actual image', 'actual checksum', |
| audio=None, crash=False, timeout=False, error=False, test_time=12345) |
| |
| test, port = self._setup_test(run_test) |
| test._archive_path = '/path/some-dir/some-test.wpr' |
| test._url = 'http://some-test/' |
| |
| try: |
| driver = port.create_driver(worker_number=1, no_timeout=True) |
| output = test.run_single(driver, '/path/some-dir/some-test.replay', time_out_ms=100) |
| self.assertTrue(output) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| |
| self.assertEqual(len(loaded_pages), 2) |
| self.assertEqual(loaded_pages[0].test_name, test.force_gc_test) |
| self.assertEqual(loaded_pages[1].test_name, 'http://some-test/') |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, '') |
| self.assertEqual(port.host.filesystem.read_binary_file('/path/some-dir/some-test-actual.png'), 'actual image') |
| self.assertEqual(output.test_time, 12345) |
| |
| def test_run_single_fails_without_webpagereplay(self): |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| |
| test, port = self._setup_test() |
| test._start_replay_server = lambda archive, record: None |
| test._archive_path = '/path/some-dir.wpr' |
| test._url = 'http://some-test/' |
| |
| try: |
| driver = port.create_driver(worker_number=1, no_timeout=True) |
| self.assertEqual(test.run_single(driver, '/path/some-dir/some-test.replay', time_out_ms=100), None) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, "Web page replay didn't start.\n") |
| |
| def test_run_with_driver_accumulates_results(self): |
| port = MockPort() |
| test, port = self._setup_test() |
| counter = [0] |
| |
| def mock_run_signle(drive, path, timeout): |
| counter[0] += 1 |
| return DriverOutput('some output', image=None, image_hash=None, audio=None, test_time=counter[0], measurements={}) |
| |
| test.run_single = mock_run_signle |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| try: |
| driver = port.create_driver(worker_number=1, no_timeout=True) |
| self.assertTrue(test._run_with_driver(driver, None)) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, '') |
| |
| self.assertEqual(test._metrics.keys(), ['Time']) |
| self.assertEqual(test._metrics['Time'].flattened_iteration_values(), [float(i * 1000) for i in range(2, 7)]) |
| |
| def test_run_with_driver_accumulates_memory_results(self): |
| port = MockPort() |
| test, port = self._setup_test() |
| counter = [0] |
| |
| def mock_run_signle(drive, path, timeout): |
| counter[0] += 1 |
| return DriverOutput('some output', image=None, image_hash=None, audio=None, test_time=counter[0], measurements={'Malloc': 10, 'JSHeap': 5}) |
| |
| test.run_single = mock_run_signle |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| try: |
| driver = port.create_driver(worker_number=1, no_timeout=True) |
| self.assertTrue(test._run_with_driver(driver, None)) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, '') |
| |
| metrics = test._metrics |
| self.assertEqual(sorted(metrics.keys()), ['JSHeap', 'Malloc', 'Time']) |
| self.assertEqual(metrics['Time'].flattened_iteration_values(), [float(i * 1000) for i in range(2, 7)]) |
| self.assertEqual(metrics['Malloc'].flattened_iteration_values(), [float(10)] * 5) |
| self.assertEqual(metrics['JSHeap'].flattened_iteration_values(), [float(5)] * 5) |
| |
| def test_prepare_fails_when_wait_until_ready_fails(self): |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| |
| test, port = self._setup_test() |
| test._start_replay_server = lambda archive, record: self.__class__.MockReplayServer(wait_until_ready=False) |
| test._archive_path = '/path/some-dir.wpr' |
| test._url = 'http://some-test/' |
| |
| try: |
| driver = port.create_driver(worker_number=1, no_timeout=True) |
| self.assertEqual(test.run_single(driver, '/path/some-dir/some-test.replay', time_out_ms=100), None) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, "Web page replay didn't start.\n") |
| |
| def test_prepare(self): |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| |
| def run_test(test_input, stop_when_done): |
| self._add_file(port, '/path/some-dir', 'some-test.wpr', 'wpr content') |
| return DriverOutput('actual text', 'actual image', 'actual checksum', |
| audio=None, crash=False, timeout=False, error=False) |
| |
| test, port = self._setup_test(run_test) |
| |
| try: |
| self.assertTrue(test.prepare(time_out_ms=100)) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, 'Preparing replay for some-test.replay\nPrepared replay for some-test.replay\n') |
| self.assertEqual(port.host.filesystem.read_binary_file('/path/some-dir/some-test-expected.png'), 'actual image') |
| |
| def test_prepare_calls_run_single(self): |
| output_capture = OutputCapture() |
| output_capture.capture_output() |
| called = [False] |
| |
| def run_single(driver, url, time_out_ms, record): |
| self.assertTrue(record) |
| self.assertEqual(url, '/path/some-dir/some-test.wpr') |
| called[0] = True |
| return False |
| |
| test, port = self._setup_test() |
| test.run_single = run_single |
| |
| try: |
| self.assertFalse(test.prepare(time_out_ms=100)) |
| finally: |
| actual_stdout, actual_stderr, actual_logs = output_capture.restore_output() |
| self.assertTrue(called[0]) |
| self.assertEqual(test._archive_path, '/path/some-dir/some-test.wpr') |
| self.assertEqual(test._url, 'http://some-test/') |
| self.assertEqual(actual_stdout, '') |
| self.assertEqual(actual_stderr, '') |
| self.assertEqual(actual_logs, "Preparing replay for some-test.replay\nFailed to prepare a replay for some-test.replay\n") |
| |
| |
| class TestPerfTestFactory(unittest.TestCase): |
| def test_regular_test(self): |
| test = PerfTestFactory.create_perf_test(MockPort(), 'some-dir/some-test', '/path/some-dir/some-test') |
| self.assertEqual(test.__class__, PerfTest) |
| |
| def test_inspector_test(self): |
| test = PerfTestFactory.create_perf_test(MockPort(), 'inspector/some-test', '/path/inspector/some-test') |
| self.assertEqual(test.__class__, ChromiumStylePerfTest) |