| OLD | NEW |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import argparse | 5 import argparse |
| 6 import unittest | 6 import unittest |
| 7 | 7 |
| 8 from infra.libs import event_mon | 8 from infra.libs import event_mon |
| 9 from infra.libs.event_mon import config, router | 9 from infra.libs.event_mon import config, router |
| 10 | 10 |
| 11 | 11 |
| 12 class MonitoringTest(unittest.TestCase): | 12 class MonitoringTest(unittest.TestCase): |
| 13 | 13 |
| 14 def _set_up_args(self, args=None): | 14 def _set_up_args(self, args=None): |
| 15 parser = argparse.ArgumentParser() | 15 parser = argparse.ArgumentParser() |
| 16 event_mon.add_argparse_options(parser) | 16 event_mon.add_argparse_options(parser) |
| 17 args = parser.parse_args((args or []) + ['--event-mon-dry-run', 'True']) | 17 args = parser.parse_args((args or [])) |
| 18 self.assertEquals(args.event_mon_dry_run, True) | 18 self.assertEquals(args.event_mon_run_type, 'dry') |
| 19 event_mon.process_argparse_options(args) | 19 event_mon.process_argparse_options(args) |
| 20 self.assertIsInstance(config._router, router._Router) | 20 self.assertIsInstance(config._router, router._Router) |
| 21 | 21 |
| 22 def tearDown(self): | 22 def tearDown(self): |
| 23 event_mon.close() | 23 event_mon.close() |
| 24 | 24 |
| 25 def test_send_service_smoke(self): | 25 def test_send_service_smoke(self): |
| 26 self._set_up_args() | 26 self._set_up_args() |
| 27 self.assertTrue(event_mon.send_service_event('START')) | 27 self.assertTrue(event_mon.send_service_event('START')) |
| 28 self.assertTrue(event_mon.send_service_event('START', | 28 self.assertTrue(event_mon.send_service_event('START', |
| 29 timestamp_kind=None)) | 29 timestamp_kind=None)) |
| 30 self.assertTrue(event_mon.send_service_event('START', | 30 self.assertTrue(event_mon.send_service_event('START', |
| 31 timestamp_kind='BEGIN')) | 31 timestamp_kind='BEGIN')) |
| 32 self.assertTrue(event_mon.send_service_event('STOP', | 32 self.assertTrue(event_mon.send_service_event('STOP', |
| 33 timestamp_kind='END', | 33 timestamp_kind='END', |
| 34 event_timestamp=1234)) | 34 event_timestamp=1234)) |
| 35 | 35 |
| 36 def test_send_service_errors(self): | 36 def test_send_service_errors(self): |
| 37 self._set_up_args() | 37 self._set_up_args() |
| 38 self.assertFalse(event_mon.send_service_event('invalid')) | 38 self.assertFalse(event_mon.send_service_event('invalid')) |
| 39 self.assertFalse(event_mon.send_service_event('START', | 39 self.assertFalse(event_mon.send_service_event('START', |
| 40 timestamp_kind='invalid')) | 40 timestamp_kind='invalid')) |
| 41 self.assertFalse(event_mon.send_service_event( | 41 self.assertFalse(event_mon.send_service_event( |
| 42 'START', event_timestamp='2015-01-25')) | 42 'START', event_timestamp='2015-01-25')) |
| OLD | NEW |