changes

fusion_accounting/fusion_accounting_ocr/tests/test_field_parser.py

@@ -0,0 +1,74 @@
+from unittest.mock import MagicMock
+
+from odoo.tests import tagged
+from odoo.tests.common import TransactionCase
+
+from odoo.addons.fusion_accounting_ocr.services.invoice_field_parser import (
+    parse_invoice_fields,
+)
+
+
+@tagged('post_install', '-at_install')
+class TestFieldParser(TransactionCase):
+
+    def test_parser_handles_empty_text(self):
+        result = parse_invoice_fields(self.env, '')
+        self.assertIsNone(result['total'])
+        self.assertEqual(result['line_items'], [])
+
+    def test_parser_handles_no_provider_gracefully(self):
+        # Without an LLM provider configured, parse should return an empty
+        # result dict rather than crashing.
+        result = parse_invoice_fields(self.env, 'INVOICE 12345 Total $100')
+        self.assertIn('total', result)
+        self.assertIn('line_items', result)
+        self.assertIsInstance(result['line_items'], list)
+
+    def test_parser_consumes_clean_json(self):
+        provider = MagicMock()
+        provider.complete.return_value = {
+            'content': (
+                '{"vendor_name": "Acme Co", "invoice_number": "INV-1",'
+                ' "invoice_date": "2026-04-20", "due_date": null,'
+                ' "currency": "CAD", "subtotal": 90.0, "tax_total": 10.0,'
+                ' "total": 100.0, "line_items": ['
+                '{"description": "Widget", "quantity": 1, "unit_price": 90.0,'
+                ' "amount": 90.0}]}'
+            ),
+        }
+        result = parse_invoice_fields(self.env, 'raw text', provider=provider)
+        self.assertEqual(result['vendor_name'], 'Acme Co')
+        self.assertEqual(result['invoice_number'], 'INV-1')
+        self.assertEqual(result['total'], 100.0)
+        self.assertEqual(len(result['line_items']), 1)
+        self.assertEqual(result['line_items'][0]['description'], 'Widget')
+
+    def test_parser_strips_markdown_fences(self):
+        provider = MagicMock()
+        provider.complete.return_value = {
+            'content': (
+                '```json\n'
+                '{"vendor_name": "Beta Ltd", "invoice_number": "B-2",'
+                ' "invoice_date": null, "due_date": null, "currency": null,'
+                ' "subtotal": null, "tax_total": null, "total": 5.5,'
+                ' "line_items": []}\n'
+                '```'
+            ),
+        }
+        result = parse_invoice_fields(self.env, 'raw text', provider=provider)
+        self.assertEqual(result['vendor_name'], 'Beta Ltd')
+        self.assertEqual(result['total'], 5.5)
+
+    def test_parser_returns_empty_on_invalid_json(self):
+        provider = MagicMock()
+        provider.complete.return_value = {'content': 'not json at all'}
+        result = parse_invoice_fields(self.env, 'raw text', provider=provider)
+        self.assertIsNone(result['total'])
+        self.assertEqual(result['line_items'], [])
+
+    def test_parser_returns_empty_on_provider_exception(self):
+        provider = MagicMock()
+        provider.complete.side_effect = RuntimeError('boom')
+        result = parse_invoice_fields(self.env, 'raw text', provider=provider)
+        self.assertIsNone(result['total'])
+        self.assertEqual(result['line_items'], [])