"""Tests for ContractHistory model."""

from contracts.constants import contracts_constants
from contracts.models import contract_history


def test_set_track_change_history():
    """Test set track change history."""
    vendor_id = 1
    isrc = 'QM4TW1664434'
    contract_id = 6523
    oa_user = 123
    delete_message = contracts_constants.DELETE_TRACK_SPLIT_MESSAGE

    change_history_text = delete_message.format(isrc=isrc, vendor_id=vendor_id)
    response = contract_history.set_track_change_history(
        delete_message, vendor_id, isrc, contract_id, oa_user
    )

    assert response.contract_id == contract_id
    assert response.change_detail == change_history_text


def test_set_product_territory_change_history():
    """Test set track change history."""
    upc = 191018329785
    country_id = 1
    contract_id = 6523
    oa_user = 123
    country_name = 'India'
    delete_message = contracts_constants.DELETE_PRODUCT_TERRITORY_SPLIT_MESSAGE

    change_history_text = delete_message.format(upc=upc, country_name=country_name)
    response = contract_history.set_product_territory_change_history(
        delete_message, upc, country_id, contract_id, oa_user, country_name
    )

    assert response.contract_id == contract_id
    assert response.change_detail == change_history_text