"""Test accounting_run_calculate tasks' helpers."""

from unittest.mock import patch

from tasks.accounting_run_calculate import helpers


@patch('tasks.accounting_run_calculate.helpers.ows')
def test_get_sales_file_ids(mock_ows):
    """Test method to get sales files in parent accounting_period."""
    accounting_period_id = 1
    mock_sales_files = [
        {
            'sales_file_id': 1,
            'sales_file_name': 'Sales 1'
        },
        {
            'sales_file_id': 2,
            'sales_file_name': 'Sales 2'
        },
        {
            'sales_file_id': 3,
            'sales_file_name': 'Sales 3'
        }
    ]
    mock_ows.get_accounting_period_sales_files.return_value = mock_sales_files

    res = helpers.get_sales_file_ids(accounting_period_id)

    assert res == [1, 2, 3]
    mock_ows.get_accounting_period_sales_files.assert_called_once_with(
        accounting_period_id
    )