"""Buffers utils test module."""

from unittest.mock import MagicMock
from unittest.mock import patch

from src import buffers_utils


@patch('io.BytesIO')
def test_allocate_buffers(bytesio_mock):
    """Test allocate_buffers function."""
    # mocking
    output_buffers_count = 1
    expected_mock = MagicMock()
    bytesio_mock.return_value = expected_mock
    # test function call
    file_buffers = buffers_utils.allocate_buffers(output_buffers_count)
    # checking
    assert expected_mock == file_buffers[0]


def test_release_allocated_buffers():
    """Test release_allocated_buffers function."""
    # mocking
    buffers = [MagicMock() for i in range(3)]
    # test function call
    buffers_utils.release_allocated_buffers(buffers)
    # checking
    for buffer in buffers:
        assert buffer.close.called