"""Functional tests for getting a TransferWise batch."""

from oto import status

from tests.testutils import db, mock_auth


@db.test_schema_default_seed
def test_get_batches(auth_client, mocker, mock_account):
    """Test getting TransferWise batches for account."""
    mock_auth(mocker, mock_account.id)

    result = auth_client.get(
        "/transferwise/batches", query_string={"vendor_id": mock_account.id}
    )

    assert result.status_code == status.OK
    batch_ids = [batch["id"] for batch in result.json]
    assert batch_ids == [1, 2, 3]