from atst.eda_client import MockEDAClient


client = MockEDAClient()


def test_list_contracts():
    results = client.list_contracts()
    assert len(results) == 3


def test_get_contract():
    result = client.get_contract("DCA10096D0052", "y")
    assert result["contract_no"] == "DCA10096D0052"
    assert result["amount"] == 2_000_000


def test_contract_not_found():
    result = client.get_contract("abc", "y")
    assert result is None