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

from unittest.mock import patch

from src.connectors.ows_royalties import OwsRoyalties
from tests.conftest import CONTRACT_ID


@patch('src.connectors.ows_royalties.OwsService.get')
def test_get_statement_period(mock_owsservice):
    """Test getting statement period."""
    statement_period_id = 123
    path = f'/statement-period/{statement_period_id}'
    mock_owsservice.return_value = {}

    OwsRoyalties.get_statement_period(statement_period_id)

    mock_owsservice.assert_called_with(path)


@patch('src.connectors.ows_royalties.OwsService.get')
def test_get_contract(mock_owsservice):
    """Test getting an account."""
    path = f'/contract/{CONTRACT_ID}'
    mock_owsservice.return_value = {}

    OwsRoyalties.get_contract(CONTRACT_ID)

    mock_owsservice.assert_called_with(path)


@patch('src.connectors.ows_royalties.OwsService.get')
def test_get_reference_payment_entity(mock_owsservice):
    """Test getting a reference payment entity."""
    reference_payment_entity_id = 1
    path = f'/reference-payment-entity/{reference_payment_entity_id}/'
    mock_owsservice.return_value = {}

    OwsRoyalties.get_reference_payment_entity(reference_payment_entity_id)

    mock_owsservice.assert_called_with(path)


@patch('src.connectors.ows_royalties.OwsService.get')
def test_get_reference_psigningt_entity(mock_owsservice):
    """Test getting a reference signing entity."""
    reference_signing_entity_id = 1
    path = f'/reference-signing-entity/{reference_signing_entity_id}/'
    mock_owsservice.return_value = {}

    OwsRoyalties.get_reference_signing_entity(reference_signing_entity_id)

    mock_owsservice.assert_called_with(path)