diff options
author | Victor Stinner <victor.stinner@gmail.com> | 2014-07-14 22:26:57 +0200 |
---|---|---|
committer | Victor Stinner <victor.stinner@gmail.com> | 2014-07-14 22:26:57 +0200 |
commit | bbbf45b08d49bd74341cccc24d8fe345cf1eea85 (patch) | |
tree | e8f95984dd9ab27fcdf176e45bd4d88651d0dc15 /Lib/asyncio/test_utils.py | |
parent | ed589f5354d2f757c43505b9e1b1f3dabeb15428 (diff) | |
parent | 1cae9ec96667e80e85fcdb034206dfd614708cd1 (diff) | |
download | cpython-git-bbbf45b08d49bd74341cccc24d8fe345cf1eea85.tar.gz |
Merge with Python 3.4
Diffstat (limited to 'Lib/asyncio/test_utils.py')
-rw-r--r-- | Lib/asyncio/test_utils.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/Lib/asyncio/test_utils.py b/Lib/asyncio/test_utils.py index 6abcaf1d37..840bbf94a3 100644 --- a/Lib/asyncio/test_utils.py +++ b/Lib/asyncio/test_utils.py @@ -3,6 +3,7 @@ import collections import contextlib import io +import logging import os import re import socket @@ -28,6 +29,7 @@ from . import futures from . import selectors from . import tasks from .coroutines import coroutine +from .log import logger if sys.platform == 'win32': # pragma: no cover @@ -401,3 +403,17 @@ class TestCase(unittest.TestCase): def tearDown(self): events.set_event_loop(None) + + +@contextlib.contextmanager +def disable_logger(): + """Context manager to disable asyncio logger. + + For example, it can be used to ignore warnings in debug mode. + """ + old_level = logger.level + try: + logger.setLevel(logging.CRITICAL+1) + yield + finally: + logger.setLevel(old_level) |