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_quota_measures(): measure = get_mfn_siv_product(1, geographical_area_id='1011', measure_type_id='143') 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, ) AgreementFactory(country_name='Espana', slug='spain', country_codes=['1011']) application = Application(country_profile='spain') document = Document(application) document.get_quota_measures() actual_measure_list = document.measure_list assert len(actual_measure_list) == 1, 'Not the correct things to assert' actual_measure = actual_measure_list[0] assert actual_measure.commodity_code == str( current_measure.goods_nomenclature_item_id) assert actual_measure.measure_sid == current_measure.measure_sid assert actual_measure.quota_order_number_id == str( current_measure.ordernumber) assert actual_measure.validity_start_date == current_measure.validity_start_date assert actual_measure.validity_end_date == current_measure.validity_end_date assert actual_measure.geographical_area_id == current_measure.geographical_area_id assert actual_measure.reduction_indicator == current_measure.reduction_indicator