"""Tests for account logic."""

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

from abacus_account.logic.account_dataload import dataload_accounts_by_ids
from abacus_account.schemas import AccountDetailSchema
from tests.utils.factories import AccountFactory


@patch('abacus_account.models.account.Account.get_filtered_query')
def test_dataload_accounts_by_ids(mock_query):
    """Test dataload_accounts_by_ids function."""
    account_ids = [1, 2, 3]

    accounts = [AccountFactory.create(
        account_id=account_id,
        account_name=f'Account {account_id}',
    ) for account_id in account_ids]

    mock_query.return_value = MagicMock(all=MagicMock(return_value=accounts))

    res = dataload_accounts_by_ids(account_ids)

    assert res.message == {'items': [
        {
            'data': AccountDetailSchema().dump(accounts[0])
        },
        {
            'data': AccountDetailSchema().dump(accounts[1])
        },
        {
            'data': AccountDetailSchema().dump(accounts[2])
        },
    ]}
    mock_query.assert_called_once_with(account_ids=account_ids)