Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(563)

Side by Side Diff: infra_libs/logs/test/logs_test.py

Issue 2213143002: Add infra_libs as a bootstrap dependency. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Removed the ugly import hack Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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())
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698