"""Tests for contract_terms_delete."""
from unittest.mock import call
from unittest.mock import patch

from json_contract_file_import import contract_terms_delete


@patch('json_contract_file_import.contract_terms_delete'
       '.ows_royalties.get_contract_terms')
@patch('json_contract_file_import.contract_terms_delete'
       '.ows_royalties.delete_contract_term')
def test_delete_contract_terms(
        mock_delete_contract_term,
        mock_get_contract_terms,
):
    """Test delete contract terms."""
    mock_get_contract_terms.return_value.json.return_value = [
        {'contract_term_id': 1},
        {'contract_term_id': 2},
        {'contract_term_id': 3},
    ]

    contract_terms_delete.delete_contract_terms(100)

    mock_delete_contract_term.assert_has_calls([
        call(1),
        call(2),
        call(3)
    ])