OLD | NEW |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 """Unit tests for the fetch_builds module.""" | 5 """Unit tests for the fetch_builds module.""" |
6 | 6 |
7 import errno | 7 import errno |
8 import unittest | 8 import unittest |
9 | 9 |
10 # The third-party mock module is expected to be available in PYTHONPATH. | 10 # The third-party mock module is expected to be available in PYTHONPATH. |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
64 archive._platform = 'linux' | 64 archive._platform = 'linux' |
65 self.assertEqual('chromium-linux-archive', archive.BucketName()) | 65 self.assertEqual('chromium-linux-archive', archive.BucketName()) |
66 self.assertEqual( | 66 self.assertEqual( |
67 'chromium.linux/Linux Builder/full-build-linux_1234567890abcdef.zip', | 67 'chromium.linux/Linux Builder/full-build-linux_1234567890abcdef.zip', |
68 archive.FilePath('1234567890abcdef')) | 68 archive.FilePath('1234567890abcdef')) |
69 | 69 |
70 def test_FullBuildArchive_Android(self): | 70 def test_FullBuildArchive_Android(self): |
71 archive = fetch_build.FullBuildArchive() | 71 archive = fetch_build.FullBuildArchive() |
72 archive._platform = 'android' | 72 archive._platform = 'android' |
73 self.assertEqual('chromium-android', archive.BucketName()) | 73 self.assertEqual('chromium-android', archive.BucketName()) |
74 self.assertEqual('android_main_rel/full-build-linux_1234567890abcdef.zip', | 74 self.assertEqual( |
| 75 'android_main_rel/full-build-linux_1234567890abcdef.zip', |
75 archive.FilePath('1234567890abcdef')) | 76 archive.FilePath('1234567890abcdef')) |
76 | 77 |
77 def test_FullBuildArchive_Linux_BuilderName(self): | 78 def test_FullBuildArchive_Linux_BuilderName(self): |
78 archive = fetch_build.FullBuildArchive() | 79 archive = fetch_build.FullBuildArchive() |
79 archive._platform = 'linux' | 80 archive._platform = 'linux' |
80 self.assertEqual('linux_full_bisect_builder', archive.GetBuilderName()) | 81 self.assertEqual('linux_full_bisect_builder', archive.GetBuilderName()) |
81 | 82 |
82 def test_FullBuildArchive_Windows_BuildTime(self): | 83 def test_FullBuildArchive_Windows_BuildTime(self): |
83 archive = fetch_build.FullBuildArchive() | 84 archive = fetch_build.FullBuildArchive() |
84 archive._platform = 'win' | 85 archive._platform = 'win' |
(...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
160 archive = fetch_build.PerfBuildArchive() | 161 archive = fetch_build.PerfBuildArchive() |
161 archive._platform = 'mac' | 162 archive._platform = 'mac' |
162 self.assertEqual('mac_perf_bisect_builder', archive.GetBuilderName()) | 163 self.assertEqual('mac_perf_bisect_builder', archive.GetBuilderName()) |
163 | 164 |
164 def test_PerfBuildArchive_mac_BuildTime(self): | 165 def test_PerfBuildArchive_mac_BuildTime(self): |
165 archive = fetch_build.PerfBuildArchive() | 166 archive = fetch_build.PerfBuildArchive() |
166 archive._platform = 'mac' | 167 archive._platform = 'mac' |
167 self.assertEqual(14400, archive.GetBuilderBuildTime()) | 168 self.assertEqual(14400, archive.GetBuilderBuildTime()) |
168 | 169 |
169 def test_GetBuildBotUrl_Perf(self): | 170 def test_GetBuildBotUrl_Perf(self): |
170 self.assertEqual(fetch_build.PERF_TRY_SERVER_URL, | 171 self.assertEqual( |
| 172 fetch_build.PERF_TRY_SERVER_URL, |
171 fetch_build.GetBuildBotUrl(fetch_build.PERF_BUILDER)) | 173 fetch_build.GetBuildBotUrl(fetch_build.PERF_BUILDER)) |
172 | 174 |
173 def test_GetBuildBotUrl_full(self): | 175 def test_GetBuildBotUrl_full(self): |
174 self.assertEqual(fetch_build.LINUX_TRY_SERVER_URL, | 176 self.assertEqual( |
| 177 fetch_build.LINUX_TRY_SERVER_URL, |
175 fetch_build.GetBuildBotUrl(fetch_build.FULL_BUILDER)) | 178 fetch_build.GetBuildBotUrl(fetch_build.FULL_BUILDER)) |
176 | 179 |
| 180 |
177 class UnzipTest(unittest.TestCase): | 181 class UnzipTest(unittest.TestCase): |
178 | 182 |
179 def setUp(self): | 183 def setUp(self): |
180 # Mocks of the os and bisect_utils modules are used in the methods below. | 184 # Mocks of the os and bisect_utils modules are used in the methods below. |
181 os_patcher = mock.patch('fetch_build.os') | 185 os_patcher = mock.patch('fetch_build.os') |
182 self.mock_os = os_patcher.start() | 186 self.mock_os = os_patcher.start() |
183 self.addCleanup(os_patcher.stop) | 187 self.addCleanup(os_patcher.stop) |
184 | 188 |
185 bisect_utils_patcher = mock.patch('fetch_build.bisect_utils') | 189 bisect_utils_patcher = mock.patch('fetch_build.bisect_utils') |
186 self.mock_bisect_utils = bisect_utils_patcher.start() | 190 self.mock_bisect_utils = bisect_utils_patcher.start() |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
240 @mock.patch('fetch_build.shutil') | 244 @mock.patch('fetch_build.shutil') |
241 def test_RemoveDirectoryTree(self, mock_shutil): | 245 def test_RemoveDirectoryTree(self, mock_shutil): |
242 # _RemoveDirectoryTree uses shutil.rmtree. | 246 # _RemoveDirectoryTree uses shutil.rmtree. |
243 fetch_build._RemoveDirectoryTree('some/path') | 247 fetch_build._RemoveDirectoryTree('some/path') |
244 mock_shutil.rmtree.assert_called_with('some/path') | 248 mock_shutil.rmtree.assert_called_with('some/path') |
245 | 249 |
246 | 250 |
247 if __name__ == '__main__': | 251 if __name__ == '__main__': |
248 unittest.main() | 252 unittest.main() |
249 | 253 |
OLD | NEW |