triage_lib, sync_stages: allow commit message to specify pre-cq-configs

BUG=chromium:464503
TEST=unit tests

Change-Id: I8d431aa147e2ff3800510da7b0123588d978e3d7
Reviewed-on: https://chromium-review.googlesource.com/256593
Tested-by: Aviv Keshet <akeshet@chromium.org>
Reviewed-by: Prathmesh Prabhu <pprabhu@chromium.org>
Commit-Queue: Aviv Keshet <akeshet@chromium.org>

1 files changed, 34 insertions, 0 deletions

diff --git a/cbuildbot/stages/sync_stages_unittest.py b/cbuildbot/stages/sync_stages_unittest.py
index 1c85fdc9a..0f57620cd 100644
--- a/cbuildbot/stages/sync_stages_unittest.py
+++ b/cbuildbot/stages/sync_stages_unittest.py
@@ -499,6 +499,40 @@ class PreCQLauncherStageTest(MasterCQSyncTestCase):
     self.assertEqual(self.sync_stage.VerificationsForChange(change),
                      constants.PRE_CQ_DEFAULT_CONFIGS)
 
+  def testVerificationsForChangeFromInvalidCommitMessage(self):
+    change = MockPatch(commit_message="""First line.
+
+Third line.
+pre-cq-configs: insect-pre-cq
+""")
+    self.PatchObject(triage_lib, 'GetOptionForChange',
+                     return_value='lumpy-pre-cq')
+    self.assertEqual(self.sync_stage.VerificationsForChange(change),
+                     ['lumpy-pre-cq'])
+
+  def testVerificationsForChangeFromCommitMessage(self):
+    change = MockPatch(commit_message="""First line.
+
+Third line.
+pre-cq-configs: stumpy-pre-cq
+""")
+    self.PatchObject(triage_lib, 'GetOptionForChange',
+                     return_value='lumpy-pre-cq')
+    self.assertEqual(self.sync_stage.VerificationsForChange(change),
+                     ['stumpy-pre-cq'])
+
+  def testMultiVerificationsForChangeFromCommitMessage(self):
+    change = MockPatch(commit_message="""First line.
+
+Third line.
+pre-cq-configs: stumpy-pre-cq
+pre-cq-configs: link-pre-cq
+""")
+    self.PatchObject(triage_lib, 'GetOptionForChange',
+                     return_value='lumpy-pre-cq')
+    self.assertEqual(self.sync_stage.VerificationsForChange(change),
+                     ['stumpy-pre-cq', 'link-pre-cq'])
+
   def _PrepareChangesWithPendingVerifications(self, verifications=None):
     """Prepare changes and pending verifications for them.