2ecb282f13
This commit includes the exact Exception class that the 'except' statement catches. According to the new flake8 3.5.0 version it should be explicit rather than implicit.
65 lines
1.5 KiB
Python
65 lines
1.5 KiB
Python
from functools import wraps
|
|
|
|
import kiwi.logger
|
|
import sys
|
|
import logging
|
|
from io import BytesIO
|
|
from mock import MagicMock, patch
|
|
|
|
# default log level, overwrite when needed
|
|
kiwi.logger.log.setLevel(logging.WARN)
|
|
|
|
# default commandline used for any test, overwrite when needed
|
|
sys.argv = [
|
|
sys.argv[0], 'system', 'prepare',
|
|
'--description', 'description', '--root', 'directory'
|
|
]
|
|
argv_kiwi_tests = sys.argv
|
|
|
|
# mock open calls
|
|
patch_open = patch("{0}.open".format(
|
|
sys.version_info.major < 3 and "__builtin__" or "builtins")
|
|
)
|
|
|
|
|
|
class raises(object):
|
|
"""
|
|
exception decorator as used in nose, tools/nontrivial.py
|
|
"""
|
|
def __init__(self, *exceptions):
|
|
self.exceptions = exceptions
|
|
self.valid = ' or '.join([e.__name__ for e in exceptions])
|
|
|
|
def __call__(self, func):
|
|
name = func.__name__
|
|
|
|
def newfunc(*args, **kw):
|
|
try:
|
|
func(*args, **kw)
|
|
except self.exceptions:
|
|
pass
|
|
except Exception:
|
|
raise
|
|
else:
|
|
message = "%s() did not raise %s" % (name, self.valid)
|
|
raise AssertionError(message)
|
|
newfunc = wraps(func)(newfunc)
|
|
return newfunc
|
|
|
|
|
|
def mock_open(data=None):
|
|
'''
|
|
Mock "open" function.
|
|
|
|
:param data:
|
|
:return:
|
|
'''
|
|
data = BytesIO(data)
|
|
mock = MagicMock()
|
|
handle = MagicMock()
|
|
handle.write.return_value = None
|
|
handle.__enter__.return_value = data or handle
|
|
mock.return_value = handle
|
|
|
|
return mock
|