| # Copyright 2016 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import optparse |
| |
| from webkitpy.tool.commands.optimize_baselines import OptimizeBaselines |
| from webkitpy.tool.commands.rebaseline_unittest import BaseTestCase |
| |
| |
| class TestOptimizeBaselines(BaseTestCase): |
| command_constructor = OptimizeBaselines |
| |
| def _write_test_file(self, port, path, contents): |
| abs_path = self.tool.filesystem.join(port.layout_tests_dir(), path) |
| self.tool.filesystem.write_text_file(abs_path, contents) |
| |
| def setUp(self): |
| super(TestOptimizeBaselines, self).setUp() |
| |
| def test_optimize_all_suffixes_by_default(self): |
| test_port = self.tool.port_factory.get('test') |
| self._write_test_file(test_port, 'another/test.html', 'Dummy test contents') |
| self._write_test_file(test_port, 'platform/test-mac-mac10.10/another/test-expected.txt', 'result A') |
| self._write_test_file(test_port, 'platform/test-mac-mac10.10/another/test-expected.png', 'result A png') |
| self._write_test_file(test_port, 'another/test-expected.txt', 'result A') |
| self._write_test_file(test_port, 'another/test-expected.png', 'result A png') |
| |
| self.command.execute( |
| optparse.Values({'suffixes': 'txt,wav,png', 'no_modify_git': True, 'platform': 'test-mac-mac10.10'}), |
| ['another/test.html'], |
| self.tool) |
| |
| self.assertFalse( |
| self.tool.filesystem.exists(self.tool.filesystem.join( |
| test_port.layout_tests_dir(), 'platform/mac/another/test-expected.txt'))) |
| self.assertFalse( |
| self.tool.filesystem.exists(self.tool.filesystem.join( |
| test_port.layout_tests_dir(), 'platform/mac/another/test-expected.png'))) |
| self.assertTrue( |
| self.tool.filesystem.exists(self.tool.filesystem.join( |
| test_port.layout_tests_dir(), 'another/test-expected.txt'))) |
| self.assertTrue( |
| self.tool.filesystem.exists(self.tool.filesystem.join( |
| test_port.layout_tests_dir(), 'another/test-expected.png'))) |