setup_command.py

#!/usr/bin/env python
"""
This module describes class implementing `python setup.py test`.
"""

import unittest

import setuptools

try:
    from setuptools.errors import BaseError
except (ModuleNotFoundError, ImportError):
    # pylint: disable=deprecated-module
    from distutils.errors import DistutilsError as BaseError


class Test(setuptools.Command):
    """
    Class implementing `python setup.py test`.
    """
    # pylint: disable=bad-option-value,no-self-use

    user_options = []
    description = 'Run tests'

    def initialize_options(self):
        """
        Do nothing. setuptools requires to override this abstract
        method.
        """

    def finalize_options(self):
        """
        Do nothing. setuptools requires to override this abstract
        method.
        """

    def run(self):
        """
        Find all tests in test/tarantool/ and run them
        """

        tests = unittest.defaultTestLoader.discover('test', pattern='suites')
        test_runner = unittest.TextTestRunner(verbosity=2)
        result = test_runner.run(tests)
        if not result.wasSuccessful():
            raise BaseError('There are failed tests')