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 |