| 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 ConfigTest(unittest.TestCase): | 12 class ConfigTest(unittest.TestCase): |
| 13 def _set_up_args(self, args=None): | 13 def _set_up_args(self, args=None): |
| 14 parser = argparse.ArgumentParser() | 14 parser = argparse.ArgumentParser() |
| 15 event_mon.add_argparse_options(parser) | 15 event_mon.add_argparse_options(parser) |
| 16 args = parser.parse_args((args or []) + ['--event-mon-dry-run', 'True']) | 16 args = parser.parse_args((args or [])) |
| 17 self.assertEquals(args.event_mon_dry_run, True) | 17 self.assertEquals(args.event_mon_run_type, 'dry') |
| 18 event_mon.process_argparse_options(args) | 18 event_mon.process_argparse_options(args) |
| 19 r = config._router | 19 r = config._router |
| 20 self.assertIsInstance(r, router._Router) | 20 self.assertIsInstance(r, router._Router) |
| 21 # Check that process_argparse_options is idempotent | 21 # Check that process_argparse_options is idempotent |
| 22 event_mon.process_argparse_options(args) | 22 event_mon.process_argparse_options(args) |
| 23 self.assertIs(config._router, r) | 23 self.assertIs(config._router, r) |
| 24 | 24 |
| 25 def _close(self): | 25 def _close(self): |
| 26 event_mon.close() | 26 event_mon.close() |
| 27 # Test that calling it twice does not raise an exception. | 27 # Test that calling it twice does not raise an exception. |
| (...skipping 12 matching lines...) Expand all Loading... |
| 40 | 40 |
| 41 args = ['--event-mon-hostname', hostname, | 41 args = ['--event-mon-hostname', hostname, |
| 42 '--event-mon-service-name', service_name, | 42 '--event-mon-service-name', service_name, |
| 43 '--event-mon-appengine-name', appengine_name] | 43 '--event-mon-appengine-name', appengine_name] |
| 44 self._set_up_args(args=args) | 44 self._set_up_args(args=args) |
| 45 event = config.cache['default_event'] | 45 event = config.cache['default_event'] |
| 46 self.assertEquals(event.event_source.host_name, hostname) | 46 self.assertEquals(event.event_source.host_name, hostname) |
| 47 self.assertEquals(event.event_source.service_name, service_name) | 47 self.assertEquals(event.event_source.service_name, service_name) |
| 48 self.assertEquals(event.event_source.appengine_name, appengine_name) | 48 self.assertEquals(event.event_source.appengine_name, appengine_name) |
| 49 self._close() | 49 self._close() |
| OLD | NEW |