def test_assign_measures_to_commodities(mock_combine_duties, commodity_list,
measure_list,
expected_assigned_measure):
mock_combine_duties.return_value = None
application = mock.MagicMock(country_name='spain')
document = Document(application)
commodity_list = [
Commodity(commodity_code) for commodity_code in commodity_list
]
measure_list = [
get_measure(**measure_properties)
for measure_properties in measure_list
]
document.commodity_list = commodity_list
document.measure_list = measure_list
for commodity in commodity_list:
assert commodity.measure_list == []
document.assign_measures_to_commodities()
for commodity in document.commodity_list:
if expected_assigned_measure:
assert len(commodity.measure_list) == 1
# TODO: MPP not a very dynamic test as it makes assumptions. Consider a better way to assert
assert commodity.measure_list[0].measure_sid == measure_list[
0].measure_sid
else:
assert commodity.measure_list == []
def test_assign_duties_to_measures(mock_combine_duties, duty_list,
measure_list, expected_assigned_duty):
mock_combine_duties.return_value = None
application = mock.MagicMock(country_name='spain')
document = Document(application)
duty_list = [
get_duty_object(**duty_properties) for duty_properties in duty_list
]
measure_list = [
get_measure(**measure_properties)
for measure_properties in measure_list
]
document.duty_list = duty_list
document.measure_list = measure_list
for measure in measure_list:
assert measure.duty_list == []
document.assign_duties_to_measures()
for measure in document.measure_list:
if expected_assigned_duty:
assert len(measure.duty_list) == 1
# TODO: MPP not a very dynamic test as it makes assumptions. Consider a better way to assert
assert measure.duty_list[0].measure_sid == measure_list[
0].measure_sid
else:
assert measure.duty_list == []
def test_combine_duties(mock_combine_duties, measure_list,
expected_combine_duties_call_count):
mock_combine_duties.return_value = None
application = mock.MagicMock(country_name='spain')
document = Document(application)
document.measure_list = [
get_measure(measure_sid=measure_sid) for measure_sid in measure_list
]
document.combine_duties()
assert mock_combine_duties.call_count == expected_combine_duties_call_count
