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_order_numbers(quota_order_numbers, has_quotas, expected_q):
AgreementFactory(country_name='Espana',
slug='spain',
country_codes=['1011'])
application = Application(country_profile='spain')
ignored_measure = get_mfn_siv_product(1,
geographical_area_id='1011',
measure_type_id='142',
measure_quota_number='99999')
CurrentMeasureFactory(
measure_sid=ignored_measure.measure_sid,
geographical_area_id=ignored_measure.geographical_area_id,
measure_type_id=ignored_measure.measure_type_id,
validity_start_date=ignored_measure.validity_start_date,
validity_end_date=ignored_measure.validity_end_date,
ordernumber=ignored_measure.quota_order_number_id,
goods_nomenclature_item_id=ignored_measure.goods_nomenclature_item_id,
reduction_indicator=ignored_measure.reduction_indicator,
)
for quota_order_number in quota_order_numbers:
measure = get_mfn_siv_product(1,
geographical_area_id='1011',
measure_type_id='143',
measure_quota_number=quota_order_number)
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,
)
document = Document(application)
document.get_quota_order_numbers()
assert document.has_quotas is has_quotas
assert document.q == expected_q
assert len(document.quota_order_number_list) == len(expected_q)
for qon in document.quota_order_number_list:
assert qon.quota_order_number_id in expected_q
