| OLD | NEW |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 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 import datetime | 5 import datetime |
| 6 | 6 |
| 7 from testing_utils import testing | 7 from testing_utils import testing |
| 8 | 8 |
| 9 from model.wf_build import WfBuild | 9 from model.wf_build import WfBuild |
| 10 from waterfall import build_util | 10 from waterfall import build_util |
| (...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 159 | 159 |
| 160 self.UpdateUnitTestConfigSettings( | 160 self.UpdateUnitTestConfigSettings( |
| 161 'download_build_data_settings', {'use_chrome_build_extract': False}) | 161 'download_build_data_settings', {'use_chrome_build_extract': False}) |
| 162 self._MockUrlfetchWithBuildData(master_name, builder_name, build_number, | 162 self._MockUrlfetchWithBuildData(master_name, builder_name, build_number, |
| 163 build_data='Test get build data') | 163 build_data='Test get build data') |
| 164 | 164 |
| 165 build_util.DownloadBuildData(master_name, builder_name, build_number) | 165 build_util.DownloadBuildData(master_name, builder_name, build_number) |
| 166 | 166 |
| 167 self.assertEqual(build.data_source, build_util.BUILDBOT_MASTER) | 167 self.assertEqual(build.data_source, build_util.BUILDBOT_MASTER) |
| 168 self.assertEqual(build.data, 'Test get build data from build master') | 168 self.assertEqual(build.data, 'Test get build data from build master') |
| 169 |
| 170 def testGetBuildEndTime(self): |
| 171 cases = { |
| 172 'null': None, |
| 173 '1467740016': datetime.datetime(2016, 7, 5, 17, 33, 36), |
| 174 } |
| 175 for end_time, expected_time in cases.iteritems(): |
| 176 master_name = 'm' |
| 177 builder_name = 'b' |
| 178 build_number = 123 |
| 179 build = WfBuild.Create(master_name, builder_name, build_number) |
| 180 build.data = '{"times": [1467738821, %s]}' % end_time |
| 181 build.completed = True |
| 182 build.last_crawled_time = self._TimeBeforeNowBySeconds(10) |
| 183 build.put() |
| 184 |
| 185 self.assertEqual(expected_time, |
| 186 build_util.GetBuildEndTime( |
| 187 master_name, builder_name, build_number)) |
| OLD | NEW |