Report buildbucket ID for all builds

infra_libs/event_mon/test/monitoring_test.py

Index: infra_libs/event_mon/test/monitoring_test.py
diff --git a/infra_libs/event_mon/test/monitoring_test.py b/infra_libs/event_mon/test/monitoring_test.py
index 2fb67437c05445ff7460970cf6e74306800b373f..055122ba5c2314b093dc27c360fc5118630b9257 100644
--- a/infra_libs/event_mon/test/monitoring_test.py
+++ b/infra_libs/event_mon/test/monitoring_test.py
@@ -313,6 +313,21 @@ class GetBuildEventTest(unittest.TestCase):
     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',