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
