Почему заменённая mock функция не отображается в mock_calls?

Question

[alekssamos]

Заменяю mock функцию и она после замены не отображается в mock_calls, как исправить?

from unittest.mock import MagicMock, mock_open, patch
mo = mock_open()
def mywrite(b):
    assert type(b) == bytearray or type(b) == bytes
    assert b == b'data'
    return len(b)
mo.return_value.write = mywrite
with patch("builtins.open", mo) as mock_file:
    with open("a.mp3", "wb") as f:
        total_size = f.write(b"data")

    print("MP3 file size: ", total_size)

    print(mock_file.mock_calls) # !!!***

вызова write нет.

Comments

[Andy_U]

Как то странно Вы unittest используете...

[alekssamos]

Andy_U, А как правильно? Я только учусь.
Не просто замокать, но и проверить.
То есть допустим случайно удалим флаг бинарного режима из кода, тест пройдёт, а при настоящем запуске какая-нибудь UnicodeDecodeError.

А зачем тестировать write?
Например, я хочу проверить, действительно ли данные приходят.

total_size: int = 0
        r = self.session.get(url, stream=True)
        r.raise_for_status()
        with open(filename, "wb") as f:
            for chunk in r.iter_content(chunk_size=1024):
                if chunk:
                    total_size += f.write(chunk)
                    f.flush()

Разве я не имею право протестировать корректность этого кода в различных симуляционных условиях?
Ну вот. Я и написал такой код, что в вопросе.

А как правильно надо?

[Andy_U]

alekssamos, Начните, что-ли отсюда. Или просто в google поищите по словам "python unittest tutorial".

А то сейчас вы подменяете своим тот код, который хотите протестировать на словах. А нужно только (при необходимости), вызываемые из вашего кода другие функции.