Анонс pytest-stub 0.1.0 для подмены пакетов, модулей, атрибутов

На днях, для проверки работы поддержки Django в новой версии uwsgiconf (подробнее об этом в одной из следующих статей) я написал некоторое количество тестов, которые в ходе исполнения задействовали сущности из самого Django. Понятное дело, без наличия этого каркаса тесты будут падать, а каждый раз для их прогона тащить Django на какой-нибудь Travis — только время тратить, поэтому нужно было имитировать наличие пакета, благо, что в данном случае имитация была целесообразна.

В комплекте pytest идёт фикстура monkeypatch, однако, к сожалению, она не решает задачу имитации наличия пакетов, а лишь позволяет подменить атрибуты уже имеющихся. Беглый поиск по существующим плагинам результатов отчего-то не дал, поэтому я создал фикстуру stub, позволяющую имитировать наличие пакета/модуля путём временной манипуляции sys.modules.

Решение сработало, я решил им поделиться — так появился плагин pytest-stub.
Ниже вы можете видеть пример использования фикстуры stub.

    
    def test_django_related(stub):
    
        stub.apply({
            # Заменим 'call_command' функцией, сгенерированной налету.
            'django.core.management.call_command': '[func]',
    
            # А здесь заменим 'BaseCommand' классом, сгенерированным налету.
            'django.core.management.base.BaseCommand': '[cls]',
    
            # Подменим сразу несколько атрибутов модуля 'conf' своими объектами.
            'django.conf': {
                'settings': object(),
                'some': True,
            },
        })
        
        # Теперь следующие импорты сработают, даже если Django недоступен.
        from django.core.management import call_command
        from django.core.management.base import BaseCommand
        from django.conf import settings, some

Из кода видно, что имеется возможность не только прописывать свои значения атрибутам модуля (можно заменить их, к примеру, на MagicMock ), но есть и средства для быстрого генерирования функций и классов. Сгенерированные функции принимают любые аргументы и возвращают None, а сгенерированные классы просто наследуются от object.

Ввиду того, что до окончания поддержки Python 2.7 остаётся лишь год с небольшим, а также того, что для работы со старой версией потребовалось бы совершить много дополнительных телодвижений, pytest-stub работает только с Python 3.

pytest-stub разрабатывается открыто. Желающие могут принять участие.


Удачной подмены.