Odoo18-Base/addons/l10n_sa_edi/tests/test_edi_zatca.py
2025-01-06 10:57:38 +07:00

144 lines
8.0 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from datetime import datetime
from freezegun import freeze_time
import logging
from pytz import timezone
from odoo.tests import tagged
from odoo.tools import misc
from .common import TestSaEdiCommon
_logger = logging.getLogger(__name__)
@tagged('post_install_l10n', '-at_install', 'post_install')
class TestEdiZatca(TestSaEdiCommon):
def testInvoiceStandard(self):
with freeze_time(datetime(year=2022, month=9, day=5, hour=8, minute=20, second=2, tzinfo=timezone('Etc/GMT-3'))):
standard_invoice = misc.file_open('l10n_sa_edi/tests/compliance/standard/invoice.xml', 'rb').read()
expected_tree = self.get_xml_tree_from_string(standard_invoice)
expected_tree = self.with_applied_xpath(expected_tree, self.invoice_applied_xpath)
self.partner_us.vat = 'US12345677'
move = self._create_invoice(name='INV/2022/00014', date='2022-09-05', date_due='2022-09-22', partner_id=self.partner_us,
product_id=self.product_a, price=320.0)
move._l10n_sa_generate_unsigned_data()
generated_file = self.env['account.edi.format']._l10n_sa_generate_zatca_template(move)
current_tree = self.get_xml_tree_from_string(generated_file)
current_tree = self.with_applied_xpath(current_tree, self.remove_ubl_extensions_xpath)
self.assertXmlTreeEqual(current_tree, expected_tree)
def testCreditNoteStandard(self):
with freeze_time(datetime(year=2022, month=9, day=5, hour=9, minute=39, second=15, tzinfo=timezone('Etc/GMT-3'))):
applied_xpath = self.credit_note_applied_xpath + \
'''
<xpath expr="(//*[local-name()='AdditionalDocumentReference']/*[local-name()='UUID'])[1]" position="replace">
<UUID>___ignore___</UUID>
</xpath>
'''
standard_credit_note = misc.file_open('l10n_sa_edi/tests/compliance/standard/credit.xml', 'rb').read()
expected_tree = self.get_xml_tree_from_string(standard_credit_note)
expected_tree = self.with_applied_xpath(expected_tree, applied_xpath)
credit_note = self._create_credit_note(name='INV/2022/00014', date='2022-09-05', date_due='2022-09-22',
partner_id=self.partner_us, product_id=self.product_a, price=320.0)
credit_note._l10n_sa_generate_unsigned_data()
generated_file = self.env['account.edi.format']._l10n_sa_generate_zatca_template(credit_note)
current_tree = self.get_xml_tree_from_string(generated_file)
current_tree = self.with_applied_xpath(current_tree, self.remove_ubl_extensions_xpath)
self.assertXmlTreeEqual(current_tree, expected_tree)
def testDebitNoteStandard(self):
with freeze_time(datetime(year=2022, month=9, day=5, hour=9, minute=45, second=27, tzinfo=timezone('Etc/GMT-3'))):
applied_xpath = self.debit_note_applied_xpath + \
'''
<xpath expr="(//*[local-name()='AdditionalDocumentReference']/*[local-name()='UUID'])[1]" position="replace">
<UUID>___ignore___</UUID>
</xpath>
'''
standard_debit_note = misc.file_open('l10n_sa_edi/tests/compliance/standard/debit.xml', 'rb').read()
expected_tree = self.get_xml_tree_from_string(standard_debit_note)
expected_tree = self.with_applied_xpath(expected_tree, applied_xpath)
debit_note = self._create_debit_note(name='INV/2022/00001', date='2022-09-05', date_due='2022-09-22',
partner_id=self.partner_us, product_id=self.product_b, price=15.80)
debit_note._l10n_sa_generate_unsigned_data()
generated_file = self.env['account.edi.format']._l10n_sa_generate_zatca_template(debit_note)
current_tree = self.get_xml_tree_from_string(generated_file)
current_tree = self.with_applied_xpath(current_tree, self.remove_ubl_extensions_xpath)
self.assertXmlTreeEqual(current_tree, expected_tree)
def testInvoiceSimplified(self):
with freeze_time(datetime(year=2023, month=3, day=10, hour=14, minute=56, second=55, tzinfo=timezone('Etc/GMT-3'))):
simplified_invoice = misc.file_open('l10n_sa_edi/tests/compliance/simplified/invoice.xml', 'rb').read()
expected_tree = self.get_xml_tree_from_string(simplified_invoice)
expected_tree = self.with_applied_xpath(expected_tree, self.invoice_applied_xpath)
move = self._create_invoice(name='INV/2023/00034', date='2023-03-10', date_due='2023-03-10', partner_id=self.partner_sa_simplified,
product_id=self.product_burger, price=265.00, quantity=3.0)
move._l10n_sa_generate_unsigned_data()
generated_file = self.env['account.edi.format']._l10n_sa_generate_zatca_template(move)
current_tree = self.get_xml_tree_from_string(generated_file)
current_tree = self.with_applied_xpath(current_tree, self.remove_ubl_extensions_xpath)
self.assertXmlTreeEqual(current_tree, expected_tree)
def testCreditNoteSimplified(self):
with freeze_time(datetime(year=2023, month=3, day=10, hour=14, minute=59, second=38, tzinfo=timezone('Etc/GMT-3'))):
simplified_credit_note = misc.file_open('l10n_sa_edi/tests/compliance/simplified/credit.xml', 'rb').read()
expected_tree = self.get_xml_tree_from_string(simplified_credit_note)
expected_tree = self.with_applied_xpath(expected_tree, self.credit_note_applied_xpath)
move = self._create_credit_note(name='INV/2023/00034', date='2023-03-10', date_due='2023-03-10',
partner_id=self.partner_sa_simplified, product_id=self.product_burger,
price=265.00, quantity=3.0)
move._l10n_sa_generate_unsigned_data()
generated_file = self.env['account.edi.format']._l10n_sa_generate_zatca_template(move)
current_tree = self.get_xml_tree_from_string(generated_file)
current_tree = self.with_applied_xpath(current_tree, self.remove_ubl_extensions_xpath)
self.assertXmlTreeEqual(current_tree, expected_tree)
def testDebitNoteSimplified(self):
with freeze_time(datetime(year=2023, month=3, day=10, hour=15, minute=1, second=46, tzinfo=timezone('Etc/GMT-3'))):
simplified_credit_note = misc.file_open('l10n_sa_edi/tests/compliance/simplified/debit.xml', 'rb').read()
expected_tree = self.get_xml_tree_from_string(simplified_credit_note)
expected_tree = self.with_applied_xpath(expected_tree, self.debit_note_applied_xpath)
move = self._create_debit_note(name='INV/2023/00034', date='2023-03-10', date_due='2023-03-10',
partner_id=self.partner_sa_simplified, product_id=self.product_burger,
price=265.00, quantity=2.0)
move._l10n_sa_generate_unsigned_data()
generated_file = self.env['account.edi.format']._l10n_sa_generate_zatca_template(move)
current_tree = self.get_xml_tree_from_string(generated_file)
current_tree = self.with_applied_xpath(current_tree, self.remove_ubl_extensions_xpath)
self.assertXmlTreeEqual(current_tree, expected_tree)
def test_invoice_tax_amount_calculation(self):
taxes = self.env['account.tax'].create([
{
'l10n_sa_is_retention': True,
'name': "Retention 10%",
'amount': -10
},
{
'name': "15% sales",
'amount': 15
},
])
invoice = self.init_invoice('out_invoice', amounts=[1000], taxes=taxes)
self.assertRecordValues(invoice.line_ids.filtered(lambda l: l.display_type == 'product'), [{
'l10n_gcc_invoice_tax_amount': 150
}])