def test_get_duty_suffix(measurement_unit_code,
measurement_unit_qualifier_code, expected_result):
duty = Duty(
measurement_unit_code=measurement_unit_code,
measurement_unit_qualifier_code=measurement_unit_qualifier_code,
)
assert duty.get_duty_string_suffix() == expected_result
def test_initialise():
duty = Duty(
commodity_code=1,
additional_code_type_id=2,
additional_code_id=3,
measure_type_id=4,
duty_expression_id=5,
duty_amount=6,
monetary_unit_code=7,
measurement_unit_code=8,
measurement_unit_qualifier_code=9,
measure_sid=10,
)
assert duty.commodity_code == "1"
assert duty.additional_code_type_id == "2"
assert duty.additional_code_id == "3"
assert duty.measure_type_id == "4"
assert duty.duty_expression_id == "5"
assert duty.duty_amount == 6
assert duty.monetary_unit_code == "7"
assert duty.measurement_unit_code == "8"
assert duty.measurement_unit_qualifier_code == "9"
assert duty.measure_sid == 10
assert duty.duty_string == ""
assert duty.measure_type_description == ""
def test_combine_duties(
duty_list,
expected_combined_duty,
):
application = mock.MagicMock()
commodity = Commodity(application)
duty_list = [Duty(**duty_dict) for duty_dict in duty_list]
commodity.duty_list = duty_list
commodity.combine_duties()
assert commodity.combined_duty == expected_combined_duty
def test_get_duty_string(duty_expression_id, duty_amount, monetary_unit_code,
measurement_unit_code,
measurement_unit_qualifier_code,
expected_duty_string):
duty = Duty(
duty_expression_id=duty_expression_id,
duty_amount=duty_amount,
monetary_unit_code=monetary_unit_code,
measurement_unit_code=measurement_unit_code,
measurement_unit_qualifier_code=measurement_unit_qualifier_code,
)
assert duty.duty_string == expected_duty_string
def test_empty_initialise():
duty = Duty()
assert duty.commodity_code == ""
assert duty.additional_code_type_id == ""
assert duty.additional_code_id == ""
assert duty.measure_type_id == ""
assert duty.duty_expression_id == ""
assert duty.duty_amount == 0
assert duty.monetary_unit_code == ""
assert duty.measurement_unit_code == ""
assert duty.measurement_unit_qualifier_code == ""
assert duty.measure_sid == 0
assert duty.duty_string == ""
assert duty.measure_type_description == ""
def test_assign_duties_to_commodities(self):
application = mock.MagicMock(
document_type=SCHEDULE,
OUTPUT_DIR='/tmp',
)
commodity_1 = Commodity(application,
product_line_suffix='80',
commodity_code='1234567891')
commodity_2 = Commodity(application,
product_line_suffix='82',
commodity_code='1234567892')
Commodity(application,
product_line_suffix='80',
commodity_code='1234567893')
duty_1 = Duty(commodity_code=commodity_1.commodity_code)
Duty(commodity_code=commodity_2.commodity_code)
db_chapter = ChapterFactory(id=1, description='Test')
chapter = ScheduleChapter(application, db_chapter)
chapter.duty_list = [duty_1]
chapter.assign_duties_to_commodities([commodity_1])
assert commodity_1.assigned is True
assert commodity_1.duty_list == [duty_1]
assert commodity_1.commodity_code_formatted == '1234 56 78 91'
def test_get_duty_prefix(duty_expression_id, expected_result):
duty = Duty(duty_expression_id=duty_expression_id, )
assert duty.get_duty_string_prefix() == expected_result
def test_get_duty_string_additional_abbreviation(duty_expression_id,
expected_abbreviation):
duty = Duty(duty_expression_id=duty_expression_id, )
assert duty.get_duty_string_additional_abbreviation(
) == expected_abbreviation