"""Unit Test for legacy_contract handlers."""

from unittest.mock import patch

from owsresponse import response


@patch('abacus_contract.blueprints.legacy_contract.legacy_contract')
def test_create_legacy_contract(mock_logic, fixture_client):
    """Test to create a legact_contract."""
    contract_id = 123
    mock_post_data = {'oa_contract_id': 1}
    mock_logic.create_legacy_contract.return_value = response.Response(
        message='ok', status=201
    )

    res = fixture_client.post(
        f'/contract/{contract_id}/legacy-contract/', json=mock_post_data
    )

    assert res.status_code == 201
    mock_logic.create_legacy_contract.assert_called_once_with(
        contract_id=contract_id, **mock_post_data
    )


@patch('abacus_contract.blueprints.legacy_contract.legacy_contract')
def test_getlegacy_contract(mock_logic, fixture_client):
    """Test to get a legact_contract."""
    contract_id = 123
    mock_logic.get_legacy_contract_by_contract_id.return_value = response.Response(
        message='ok', status=200
    )

    res = fixture_client.get(f'/contract/{contract_id}/legacy-contract/')

    assert res.status_code == 200
    mock_logic.get_legacy_contract_by_contract_id.assert_called_once_with(contract_id)