fix unit tests - keep product_origin_country_id on top of new product…

…_origin_country_code to avoid breaking the localisation modules
OCA · Apr 10, 2022 · 8487f1c · 8487f1c
1 parent e49bac5
commit 8487f1c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/intrastat_product/models/intrastat_product_declaration.py b/intrastat_product/models/intrastat_product_declaration.py
@@ -721,6 +721,9 @@ def _gather_invoices(self, notedict):
                 total_inv_product_cc += amount_company_currency
 
                 if inv_intrastat_line:
+                    product_origin_country = (
+                        inv_intrastat_line.product_origin_country_id
+                    )
                     product_origin_country_code = (
                         inv_intrastat_line.product_origin_country_code
                     )
@@ -745,6 +748,7 @@ def _gather_invoices(self, notedict):
                     "amount_company_currency": amount_company_currency,
                     "amount_accessory_cost_company_currency": 0.0,
                     "transaction_id": intrastat_transaction.id,
+                    "product_origin_country_id": product_origin_country.id or False,
                     "product_origin_country_code": product_origin_country_code,
                     "region_id": region and region.id or False,
                     "vat": vat,
@@ -868,6 +872,7 @@ def _prepare_grouped_fields(self, computation_line, fields_to_sum):
             "transport_id": computation_line.transport_id.id,
             "region_id": computation_line.region_id.id,
             "parent_id": computation_line.parent_id.id,
+            "product_origin_country_id": computation_line.product_origin_country_id.id,
             "product_origin_country_code": computation_line.product_origin_country_code,
             "amount_company_currency": 0.0,
             "vat": computation_line.vat,