forked from constverum/ProxyBroker
-
-
Notifications
You must be signed in to change notification settings - Fork 117
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
2a09c0f
commit d560297
Showing
11 changed files
with
591 additions
and
673 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,5 +1,6 @@ | ||
__dev__ | ||
__pycache__ | ||
*.eggs | ||
*.egg-info | ||
.idea | ||
.DS_Store | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -3,6 +3,7 @@ python: | |
- 3.5 | ||
- 3.6 | ||
os: | ||
- osx | ||
- linux | ||
install: | ||
- pip install -r requirements.txt | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,80 @@ | ||
""" | ||
The module was copied from | ||
https://github.com/aio-libs/aiohttp/blob/master/tests/conftest.py | ||
""" | ||
|
||
import collections | ||
import logging | ||
|
||
import pytest | ||
|
||
|
||
_LoggingWatcher = collections.namedtuple("_LoggingWatcher", | ||
["records", "output"]) | ||
|
||
|
||
class _CapturingHandler(logging.Handler): | ||
""" | ||
A logging handler capturing all (raw and formatted) logging output. | ||
""" | ||
|
||
def __init__(self): | ||
logging.Handler.__init__(self) | ||
self.watcher = _LoggingWatcher([], []) | ||
|
||
def flush(self): | ||
pass | ||
|
||
def emit(self, record): | ||
self.watcher.records.append(record) | ||
msg = self.format(record) | ||
self.watcher.output.append(msg) | ||
|
||
|
||
class _AssertLogsContext: | ||
"""A context manager used to implement TestCase.assertLogs().""" | ||
|
||
LOGGING_FORMAT = "%(levelname)s:%(name)s:%(message)s" | ||
|
||
def __init__(self, logger_name=None, level=None): | ||
self.logger_name = logger_name | ||
if level: | ||
self.level = logging._nameToLevel.get(level, level) | ||
else: | ||
self.level = logging.INFO | ||
self.msg = None | ||
|
||
def __enter__(self): | ||
if isinstance(self.logger_name, logging.Logger): | ||
logger = self.logger = self.logger_name | ||
else: | ||
logger = self.logger = logging.getLogger(self.logger_name) | ||
formatter = logging.Formatter(self.LOGGING_FORMAT) | ||
handler = _CapturingHandler() | ||
handler.setFormatter(formatter) | ||
self.watcher = handler.watcher | ||
self.old_handlers = logger.handlers[:] | ||
self.old_level = logger.level | ||
self.old_propagate = logger.propagate | ||
logger.handlers = [handler] | ||
logger.setLevel(self.level) | ||
logger.propagate = False | ||
return handler.watcher | ||
|
||
def __exit__(self, exc_type, exc_value, tb): | ||
self.logger.handlers = self.old_handlers | ||
self.logger.propagate = self.old_propagate | ||
self.logger.setLevel(self.old_level) | ||
if exc_type is not None: | ||
# let unexpected exceptions pass through | ||
return False | ||
if len(self.watcher.records) == 0: | ||
__tracebackhide__ = True | ||
assert 0, ("no logs of level {} or higher triggered on {}" | ||
.format(logging.getLevelName(self.level), | ||
self.logger.name)) | ||
|
||
|
||
@pytest.fixture | ||
def log(): | ||
return _AssertLogsContext |
Oops, something went wrong.