atst/tests/test_eda_client.py
2018-10-17 16:12:19 -04:00

27 lines
665 B
Python

from atst.eda_client import MockEDAClient, parse_eda_xml
mock_client = MockEDAClient()
def test_list_contracts():
results = mock_client.list_contracts()
assert len(results) == 3
def test_get_contract():
result = mock_client.get_contract("DCA10096D0052", "y")
assert result["number"] == "DCA10096D0052"
def test_contract_not_found():
result = mock_client.get_contract("abc", "y")
assert result is None
def test_eda_xml_parser():
with open("tests/fixtures/eda_contract.xml") as contract:
eda_data = parse_eda_xml(contract.read())
assert eda_data["clin_0001"] == 200000.00
assert not eda_data["clin_0003"]