Add support for build category in BuildEvent

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 310 matching lines @@
     # 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_with_category(self):
+    log_event = monitoring.get_build_event(
+        'BUILD', 'bot.host.name', 'build_name',
+        category='git_cl_try').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.category, BuildEvent.CATEGORY_GIT_CL_TRY)
+
+    # Try unknown value. Should produce CATEGORY_UNKNOWN.
+    log_event = monitoring.get_build_event(
+        'BUILD', 'bot.host.name', 'build_name', category='foobar').log_event()
+    self.assertIsInstance(log_event, LogRequestLite.LogEventLite)
+    event = ChromeInfraEvent.FromString(log_event.source_extension)
+    self.assertTrue(event.HasField('build_event'))
+    self.assertEquals(
+        event.build_event.category, BuildEvent.CATEGORY_UNKNOWN)
+
+    # Try empty value. Should not set category.
+    log_event = monitoring.get_build_event(
+        'BUILD', 'bot.host.name', 'build_name', category='').log_event()
+    self.assertIsInstance(log_event, LogRequestLite.LogEventLite)
+    event = ChromeInfraEvent.FromString(log_event.source_extension)
+    self.assertTrue(event.HasField('build_event'))
+    self.assertFalse(event.build_event.HasField('category'))
+
   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))