Allow easily silencing the server log during tests

This commit is contained in:
Kovid Goyal 2015-10-25 20:43:43 +05:30
parent 7bc5e491b3
commit d2535aa9cf
2 changed files with 8 additions and 0 deletions

View File

@ -86,6 +86,7 @@ class TestServer(Thread):
log=ServerLog(level=ServerLog.WARN), log=ServerLog(level=ServerLog.WARN),
) )
self.log = self.loop.log self.log = self.loop.log
self.silence_log = self.log
specialize(self) specialize(self)
def setup_defaults(self, kwargs): def setup_defaults(self, kwargs):

View File

@ -159,6 +159,13 @@ class Log(object):
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
self.prints(INFO, *args, **kwargs) self.prints(INFO, *args, **kwargs)
def __enter__(self):
self.orig_filter_level = self.filter_level
self.filter_level = self.ERROR + 100
def __exit__(self, *args):
self.filter_level = self.orig_filter_level
class DevNull(Log): class DevNull(Log):
def __init__(self): def __init__(self):