Pytest: как замокать переменную метода?

Question

[Jekson]

Есть джанговская manager command в которой обабатывается csv файл.
app/my_app/my_command.py

class Command(BaseCommand):
    def handle(self,  *args, **options):
        path = (os.path.join(os.path.abspath(os.path.dirname(__name__)), 'data.csv'))
        # остальная логика

Пишу под нее тест на pytest, проблема в том что не могу понять как замокать переменную path чтобы тест обращался не к реальному data.csv а временному тестовому файлу

@pytest.fixture
def create_test_csv_file(tmpdir_factory):
    # логка по созданию и заполнению тестового файла
    # фикстура возвращает путь к тестовому файлу
    return str(path)

@pytest.mark.django_db
def test_function(mocker, create_test_csv_file):
   # здесь как то надо замокать path 
   что то типа mock_path =  create_test_csv_file
    call_command('my_command')

Comments

[Михаил]

При запуске передать в контекст вызова пробовали?

[Jekson]

Михаил, не хотелось бы менять сигнатуру def handle и дополнять там логику ради теста, поэтому и хочу замокать

[Михаил]

Jekson, так а в чём собственно проблема переопределить переменную внутри теста?

[Jekson]

Михаил, видимо я не совсем понимаю о чем речь . Каким образом??

[Михаил]

Jekson, ну у вас тест обращается к переменной в внешнем скоупе, верно? Вы же явно не передаёте путь туда, следовательно он и подтягивает тот что видит. Почему бы просто не добавить:

@pytest.fixture
def create_test_csv_file(tmpdir_factory):
    # логка по созданию и заполнению тестового файла
    # фикстура возвращает путь к тестовому файлу
    path = (os.path.join(os.path.abspath(os.path.dirname(__name__)), 'test.csv'))
    return str(path)

[Jekson]

Михаил, так сейчас и есть так. фикстура возвращает путь к файлу test.csv но внутри def handle path ссылается на реальный файл. Надо переопределить этот момент что бы path стал равен create_test_csv_file

[Михаил]

Jekson, тогда тут только наследование поможет.

class TestCommand(Command):
    def handle(self, *args, **options):
        path = (os.path.join(os.path.abspath(os.path.dirname(__name__)), 'test.csv'))

Answer 1

[Jekson]

Михаил спасибо за помощь. Похоже всетаки придется менять manager command. Сейчас остановился на таком решении через дополнительную переменную с дефолтным значением, возможно кому-то пригодится

class Command(BaseCommand):
    def add_arguments(self, parser):
        parser.add_argument(
            "--path",
            dest="path",
            default=os.path.join(os.path.abspath(os.path.dirname(__name__)), 'data.csv')
        )

    def handle(self, *args, **options):
        path = options.get("path")
        ...

Вызов в тесте

call_command('my_command', path=create_test_csv_file)