[IMP] l10n_it_fatturapa_in_purchase: test with intermediary wizard check

Author: Chionne27

l10n_it_fatturapa_in_purchase/tests/test_l10n_it_fatturapa_in_purchase.py

@@ -39,15 +39,46 @@ def setUp(self):
         )
 
     def test_fatturapa_in_purchase(self):
-        """Check that `name_get` works properly. verify to_invoice field"""
+        """
+        Check that `name_get` works properly and verify the `to_invoice` field.
+        Handle the intermediary VAT check wizard if triggered during import.
+        """
         res = self.run_wizard("test_fatturapa_in_purchase", "IT05979361218_008.xml")
-        invoice_id = res.get("domain")[0][2][0]
-        invoice = self.invoice_model.browse(invoice_id)
 
+        # Handle the intermediary VAT check wizard if it appears
+        if res.get("res_model") == "wizard.check.intermediary":
+            context = res.get("context", {})
+            intermediary_id = context.get("intermediary_id")
+            invoice_id = context.get("invoice_id")
+
+            self.assertTrue(intermediary_id, "intermediary_id is missing in context")
+            self.assertTrue(invoice_id, "invoice_id is missing in context")
+
+            wizard = (
+                self.env["wizard.check.intermediary"]
+                .with_context(context)
+                .create(
+                    {
+                        "message": context.get("default_message"),
+                        "intermediary_id": intermediary_id,
+                        "invoice_id": invoice_id,
+                    }
+                )
+            )
+
+            wizard.action_confirm()
+            invoice = self.env["account.move"].browse(invoice_id)
+
+        else:
+            invoice_id = res.get("domain")[0][2][0]
+            invoice = self.invoice_model.browse(invoice_id)
+
+        # Link the invoice line to the purchase order line
         inv_line = first(invoice.invoice_line_ids)
         order_line = first(self.order.order_line)
         inv_line.purchase_line_id = order_line
 
+        # Check the display name and `to_invoice` field logic
         self.assertEqual(order_line.display_name, "test_po: Large Desk")
         self.assertFalse(order_line.to_invoice)
         order_line.product_id.write({"purchase_method": "purchase"})