OLD | NEW |
| (Empty) |
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 | |
3 # found in the LICENSE file. | |
4 | |
5 import copy | |
6 import logging | |
7 import sys | |
8 import unittest | |
9 | |
10 import mock | |
11 | |
12 from infra_libs.logs import logs | |
13 | |
14 class InfraFilterTest(unittest.TestCase): | |
15 | |
16 def test_infrafilter_adds_correct_fields(self): | |
17 record = logging.makeLogRecord({}) | |
18 infrafilter = logs.InfraFilter('US/Pacific') | |
19 infrafilter.filter(record) | |
20 self.assertTrue(hasattr(record, "severity")) | |
21 self.assertTrue(hasattr(record, "iso8601")) | |
22 | |
23 def test_infraformatter_adds_full_module_name(self): | |
24 record = logging.makeLogRecord({}) | |
25 infrafilter = logs.InfraFilter('US/Pacific') | |
26 infrafilter.filter(record) | |
27 self.assertEqual('infra_libs.logs.test.logs_test', record.fullModuleName) | |
28 | |
29 def test_filters_by_module_name(self): | |
30 record = logging.makeLogRecord({}) | |
31 infrafilter = logs.InfraFilter( | |
32 'US/Pacific', module_name_blacklist=r'^other\.module') | |
33 self.assertTrue(infrafilter.filter(record)) | |
34 | |
35 infrafilter = logs.InfraFilter( | |
36 'US/Pacific', | |
37 module_name_blacklist=r'^infra_libs\.logs\.test\.logs_test$') | |
38 self.assertFalse(infrafilter.filter(record)) | |
39 | |
40 | |
41 class DefaultProgramNameTest(unittest.TestCase): | |
42 | |
43 def setUp(self): | |
44 self.old_argv = copy.copy(sys.argv) | |
45 self.old_package = sys.modules['__main__'].__package__ | |
46 | |
47 def tearDown(self): | |
48 sys.argv = self.old_argv | |
49 sys.modules['__main__'].__package__ = self.old_package | |
50 | |
51 def test_from_argv(self): | |
52 sys.argv = ['/foo/bar/program', '--args'] | |
53 self.assertEquals('program', logs.default_program_name()) | |
54 | |
55 def test_from_main_package(self): | |
56 sys.argv = ['/foo/bar/__main__.py', '--args'] | |
57 sys.modules['__main__'].__package__ = 'foo.bar.program' | |
58 self.assertEquals('program', logs.default_program_name()) | |
59 | |
60 def test_from_argv_when_main_package_is_none(self): | |
61 sys.argv = ['/foo/bar/__main__.py', '--args'] | |
62 sys.modules['__main__'].__package__ = None | |
63 self.assertEquals('__main__.py', logs.default_program_name()) | |
OLD | NEW |