"""Tests for ows-royalties connector."""

import httpx
import pytest

from moneyhub.connectors import ows_royalties


@pytest.mark.parametrize('account_id', [
    1234,
    4321
])
def test_get_contracts_for_account(account_id, ows_client_mock):
    """Test get_contracts_for_account method."""
    contracts = [
        {
            'account_id': account_id,
            'contract_id': 111
        },
        {
            'account_id': account_id,
            'contract_id': 222
        }
    ]
    ows_client_mock.get(
        'ows-royalties',
        f'/contracts/account/{account_id}',
    ).mock(
        return_value=httpx.Response(
            200,
            json={'items': contracts, 'total_count': 2},
        )
    )
    result = ows_royalties.get_contracts_for_account(account_id)

    assert result == contracts