commit 4e29b51b97c50672d878c9bb5eb124bab7309d24
author Garry Wang <xianuowang@chromium.org> Thu Aug 26 19:32:59 2021 -0700
committer Commit Bot <commit-bot@chromium.org> Sat Aug 28 00:15:18 2021 +0000
tree d1f0c51f528dc6531f8e02c9ef69163294cddbaa
parent a0a80fa77e715d378e9a2c35dac7d2dd1fc94331

suite-scheduler: Get build from wider time range for multi-DUTs

For non tot branch there could be only 1 build per day, so one
failed build can cause us failed to locate most recent build
for a secondary board, and result in skip build for primary
board.

BUG=None
TEST=bin/run_tests

Change-Id: I9aad2c13e78e998c13a1cf73fb60cbe482b9e14c
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/infra/suite_scheduler/+/3123969
Commit-Queue: Xianuo Wang <xianuowang@chromium.org>
Tested-by: Xianuo Wang <xianuowang@chromium.org>
Reviewed-by: Sean McAllister <smcallis@google.com>

task_unittest.py

diff --git a/task_unittest.py b/task_unittest.py
index adc1b76..f702573 100644
--- a/task_unittest.py
+++ b/task_unittest.py
@@ -659,7 +659,7 @@
     ('hatch', 'kled', 'release', '55'): '0000.00.00',
     ('hatch', 'kled', 'release', '56'): '0000.00.00',
   }
-  DAILY_BUILD_DICT = {
+  RECENT_BUILD_DICT = {
     ('coral', None, 'release', '55'): '0000.00.00',
     ('coral', None, 'release', '56'): '0000.00.00',
     ('hatch', None, 'release', '55'): '0000.00.00',
@@ -777,19 +777,19 @@
     """Test schedule multi DUTs cros build with any_model set to True."""
     self.task.any_model = True
     self.task._schedule_multi_duts_cros_builds(
-        self.BUILD_DICT, self.DAILY_BUILD_DICT, self._fake_configs)
+        self.BUILD_DICT, self.RECENT_BUILD_DICT, self._fake_configs)
     self.assertEqual(self._mock_push.call_count, 4)
 
   def testScheduleMultiDUTsCrosWithoutAnyModel(self):
     """Test schedule multi DUTs cros build with any_model set to False."""
     self.task.any_model = False
     self.task._schedule_multi_duts_cros_builds(
-        self.BUILD_DICT, self.DAILY_BUILD_DICT, self._fake_configs)
+        self.BUILD_DICT, self.RECENT_BUILD_DICT, self._fake_configs)
     self.assertEqual(self._mock_push.call_count, 10)
 
   def testScheduleMultiDUTsWithMissingSecondaryBuild(self):
     """Test schedule multi DUTs cros build when missing secondary build
-    info in daily dict, the expectation is no suite will be pushed.
+    info in recent dict, the expectation is no suite will be pushed.
     """
     self.task._schedule_multi_duts_cros_builds(
         self.BUILD_DICT, {}, self._fake_configs)
@@ -836,5 +836,5 @@
   def testScheduleMultiDUTsCrosWithoutAnyModel(self):
     """Test schedule multi DUTs cros build."""
     self.task._schedule_multi_duts_cros_builds(
-        self.BUILD_DICT, self.DAILY_BUILD_DICT, self._fake_configs)
+        self.BUILD_DICT, self.RECENT_BUILD_DICT, self._fake_configs)
     self.assertEqual(self._mock_push.call_count, 4)