| 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 |