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])
