Report buildbucket ID for all builds

infra_libs/event_mon/test/monitoring_test.py

 # -*- encoding:utf-8 -*-
 # Copyright 2015 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 unittest
 
 from infra_libs import event_mon
 from infra_libs.event_mon import config, router
 from infra_libs.event_mon import monitoring
@@ skipping 295 matching lines @@
     patch_url = 'http://foo.bar/123#456'
     log_event = monitoring.get_build_event(
         'BUILD', 'bot.host.name', 'build_name', patch_url=patch_url).log_event()
     self.assertIsInstance(log_event, LogRequestLite.LogEventLite)
 
     # Check that source_extension deserializes to the right thing.
     event = ChromeInfraEvent.FromString(log_event.source_extension)
     self.assertTrue(event.HasField('build_event'))
     self.assertEquals(event.build_event.patch_url, patch_url)
 
+  def test_get_build_event_with_bbucket_id(self):
+    log_event = monitoring.get_build_event(
+        'BUILD', 'bot.host.name', 'build_name', bbucket_id=123).log_event()
+    self.assertIsInstance(log_event, LogRequestLite.LogEventLite)
+
+    # Check that source_extension deserializes to the right thing.
+    event = ChromeInfraEvent.FromString(log_event.source_extension)
+    self.assertTrue(event.HasField('build_event'))
+    self.assertEquals(event.build_event.bbucket_id, 123)
+
+    # Try invalid value. This should not throw any exceptions.
+    log_event = monitoring.get_build_event(
+        'BUILD', 'bot.host.name', 'build_name', bbucket_id='foo').log_event()
+    self.assertIsInstance(log_event, LogRequestLite.LogEventLite)
+
   def test_get_build_event_invalid_type(self):
     # An invalid type is a critical error.
     log_event = monitoring.get_build_event('INVALID_TYPE',
                                            'bot.host.name',
                                            'build_name').log_event()
     self.assertIsNone(log_event)
 
   def test_get_build_event_invalid_build_name(self):
     # an invalid builder name is not a critical error.
     hostname = 'bot.host.name'
@@ skipping 888 matching lines @@
         'BUILD',
         'bot2.host.name',
         'build_name2',
         build_number=1,
         build_scheduling_time=123456789,
         result='FAILURE',
         timestamp_kind='POINT',
         event_timestamp=None),
     ]
     self.assertTrue(monitoring.send_events(events))