tstcls.py

import inspect

from pytest import fixture

__all__ = ["TestClassBase"]


class TestClassBase(object):
    @classmethod
    def find_fixtures(cls, func, request):
        fixtures = {}
        arg_spec = inspect.getfullargspec(func)
        for index, arg in enumerate(arg_spec.args):
            if index == 0 and arg in ["self", "cls"]:
                continue
            fixtures[arg] = request.getfixturevalue(arg)

        return fixtures

    @fixture(autouse=True, scope="class")
    def init_class(self, request):
        setup_fixtures = self.find_fixtures(self.setup_test_class, request)
        teardown_fixtures = self.find_fixtures(self.teardown_test_class, request)

        self.__class__.setup_test_class(**setup_fixtures)
        yield
        self.__class__.teardown_test_class(**teardown_fixtures)

    @fixture(autouse=True)
    def init(self, request):
        setup_fixtures = self.find_fixtures(self.setup_test, request)
        teardown_fixtures = self.find_fixtures(self.teardown_test, request)

        self.setup_test(**setup_fixtures)
        yield
        self.teardown_test(**teardown_fixtures)

    def setup_test(self, **fixtures):
        pass

    def teardown_test(self, **fixtures):
        pass

    @classmethod
    def setup_test_class(cls, **fixtures):
        pass

    @classmethod
    def teardown_test_class(cls, **fixtures):
        pass