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