def create_document(self): # Create the document my_document = Document(self) my_document.check_for_quotas() # Create the measures table my_document.get_duties("preferences") tariff_data = my_document.print_tariffs() # Create the quotas table my_document.get_duties("quotas") my_document.get_quota_order_numbers() my_document.get_quota_balances() my_document.get_quota_measures() my_document.get_quota_definitions() quota_data = my_document.print_quotas() context_data = { 'AGREEMENT_NAME': self.agreement.agreement_name, 'VERSION': self.agreement.version, 'AGREEMENT_DATE': self.agreement.agreement_date_long, 'AGREEMENT_DATE_SHORT': self.agreement.agreement_date_short, 'COUNTRY_NAME': self.agreement.country_name, **tariff_data, **quota_data, } # Personalise and write the document my_document.create_document(context_data) update_document_status(self.agreement, DocumentStatus.AVAILABLE)
def test_get_duties(instrument_type, measure_type_id, expected_duty_list, expected_measure_list, expected_commodity_list, expected_quota_order_number_list): measure = get_mfn_siv_product(1, geographical_area_id='1011', measure_type_id=measure_type_id) current_measure = CurrentMeasureFactory( measure_sid=measure.measure_sid, geographical_area_id=measure.geographical_area_id, measure_type_id=measure.measure_type_id, validity_start_date=measure.validity_start_date, validity_end_date=measure.validity_end_date, ordernumber=measure.quota_order_number_id, goods_nomenclature_item_id=measure.goods_nomenclature_item_id, reduction_indicator=measure.reduction_indicator, ) GoodsNomenclatureFactory( goods_nomenclature_item_id=current_measure.goods_nomenclature_item_id) AgreementFactory(country_name='Espana', slug='spain', country_codes=['1011']) application = Application(country_profile='spain') document = Document(application) document.get_duties(instrument_type) assert len(document.duty_list) == len(expected_duty_list) if document.duty_list: assert_object(document.duty_list[0], expected_duty_list[0]) assert len(document.measure_list) == len(expected_measure_list) if document.measure_list: assert_object(document.measure_list[0], expected_measure_list[0]) assert len(document.commodity_list) == len(expected_commodity_list) if document.commodity_list: assert_object(document.commodity_list[0], expected_commodity_list[0]) assert len(document.quota_order_number_list) == len( expected_quota_order_number_list) if document.quota_order_number_list: assert_object(document.quota_order_number_list[0], expected_quota_order_number_list[0])