"""Test ows-royalties requests."""

from unittest.mock import patch

import httpx
from owsclient.test import OwsClientMock

from payoneer_payments_payout.ows_royalties import get_payment_entity


@patch('payoneer_payments_payout.ows_royalties.raise_service_error')
def test_get_payment_entity(
    mock_raise_service_error,
    mock_payment_entity,
    ows_client_mock: OwsClientMock,
):
    """Test get payment entity success."""
    mock_payment_entity_id = mock_payment_entity['reference_payment_entity_id']
    request_url = f'/reference-payment-entity/{mock_payment_entity_id}'
    ows_client_mock.get('ows-royalties', request_url).mock(
        return_value=httpx.Response(
            200,
            json=mock_payment_entity
        )
    )

    res = get_payment_entity(mock_payment_entity_id)
    assert res == mock_payment_entity

    mock_raise_service_error.assert_not_called()


@patch('payoneer_payments_payout.ows_royalties.raise_service_error')
def test_get_payment_entity_failure(
    mock_raise_service_error,
    mock_payment_entity,
    ows_client_mock: OwsClientMock
):
    """Test get payment entity failure."""
    mock_payment_entity_id = mock_payment_entity['reference_payment_entity_id']
    request_url = f'/reference-payment-entity/{mock_payment_entity_id}'
    ows_client_mock.get('ows-royalties', request_url).mock(
        return_value=httpx.Response(
            400,
            json='error'
        )
    )

    get_payment_entity(mock_payment_entity_id)

    mock_raise_service_error.assert_called_once_with(
        'ERROR in GET /reference-payment-entity/4', 'ows-royalties'
    )