1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
|
import sys
import os.path
import logging
import gc
if sys.version_info < (3,3):
raise SystemExit('Python version is %d.%d.%d, but Dugong requires 3.3 or newer'
% sys.version_info[:3])
# Enable output checks
pytest_plugins = ('pytest_checklogs')
def pytest_addoption(parser):
group = parser.getgroup("general")
group._addoption("--installed", action="store_true", default=False,
help="Test the installed package.")
group = parser.getgroup("terminal reporting")
group._addoption("--logdebug", action="append", metavar='<module>',
help="Activate debugging output from <module> for tests. Use `all` "
"to get debug messages from all modules. This option can be "
"specified multiple times.")
def pytest_configure(config):
# If we are running from the source directory, make sure that we load
# modules from here
basedir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
if not config.getoption('installed'):
if (os.path.exists(os.path.join(basedir, 'setup.py')) and
os.path.exists(os.path.join(basedir, 'dugong', '__init__.py'))):
sys.path.insert(0, basedir)
# Make sure that called processes use the same path
pp = os.environ.get('PYTHONPATH', None)
if pp:
pp = '%s:%s' % (basedir, pp)
else:
pp = basedir
os.environ['PYTHONPATH'] = pp
# When running from VCS repo, enable all warnings
if os.path.exists(os.path.join(basedir, 'MANIFEST.in')):
import warnings
warnings.resetwarnings()
warnings.simplefilter('default')
# Configure logging. We don't set a default handler but rely on
# the catchlog pytest plugin.
logdebug = config.getoption('logdebug')
root_logger = logging.getLogger()
if logdebug is not None:
logging.disable(logging.NOTSET)
if 'all' in logdebug:
root_logger.setLevel(logging.DEBUG)
else:
for module in logdebug:
logging.getLogger(module).setLevel(logging.DEBUG)
else:
root_logger.setLevel(logging.INFO)
logging.disable(logging.DEBUG)
logging.captureWarnings(capture=True)
# Run gc.collect() at the end of every test, so that we get ResourceWarnings
# as early as possible.
def pytest_runtest_teardown(item, nextitem):
gc.collect()
|