from markupsafe import Markup from odoo.fields import Command from odoo.addons.account.tests.common import AccountTestInvoicingCommon from odoo.tests import tagged from unittest.mock import patch from freezegun import freeze_time @tagged('post_install_l10n', 'post_install', '-at_install') class TestQris(AccountTestInvoicingCommon): """ Test QRIS QR generation on invoices and auto-payment registration""" @classmethod @AccountTestInvoicingCommon.setup_chart_template('id') def setUpClass(cls): super().setUpClass() cls.company_data['company'].qr_code = True cls.company_data['company'].partner_id.update({ 'country_id': cls.env.ref('base.id').id, 'city': 'Jakarta', }) cls.acc_qris_id = cls.env['res.partner.bank'].create({ 'acc_number': '123456789012345678', 'partner_id': cls.company_data['company'].partner_id.id, 'l10n_id_qris_api_key': 'apikey', 'l10n_id_qris_mid': 'mid', }) cls.qris_qr_invoice = cls.env['account.move'].create({ 'move_type': 'out_invoice', 'partner_id': cls.partner_a.id, 'currency_id': cls.env.ref('base.IDR').id, 'partner_bank_id': cls.acc_qris_id.id, 'company_id': cls.company_data['company'].id, 'invoice_line_ids': [Command.create({'quantity': 1, 'price_unit': 100})], 'qr_code_method': 'id_qr', })._post() cls.success_qris_get = { "status": "success", "data": { "qris_content": "Test Content", "qris_request_date": "2024-02-27 11:13:42", "qris_invoiceid": "413255111", "qris_nmid": "ID1020021181745" } } cls.success_qris_get_second = { "status": "success", "data": { "qris_content": "Test Content Second", "qris_request_date": "2024-02-27 11:13:42", "qris_invoiceid": "413255111", "qris_nmid": "ID1020021181745" } } cls.qris_status_fail = { "status": "failed", "data": { "qris_status": "unpaid" } } cls.qris_status_success = { "status": "success", "data": { "qris_status": "paid", "qris_payment_customername": "Zainal Arief", "qris_payment_methodby": "Sakuku" }, "qris_api_version_code": "2206091709" } @freeze_time("2024-02-27 04:15:00") def test_qris_qr_code_generation(self): """ QR-Code generation conditions: - should only come from portal side - only call the API when the QRIS transaction not found or lasted for > 30 minutes - if transaction found and < 30 minutes, no API call needed """ with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get ) as patched: # QR code shouldn't be generated without the context. result = self.qris_qr_invoice._generate_qr_code() self.assertIsNone(result) # But of course, should be with it. result = self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() self.assertIsNotNone(result) # Confirm that a transaction should be registered on the invoice qr_details = self.qris_qr_invoice.l10n_id_qris_transaction_ids self.assertEqual(len(qr_details), 1) qris_data = self.success_qris_get['data'] self.assertEqual(qr_details.qris_invoice_id, qris_data['qris_invoiceid']) self.assertEqual(qr_details.qris_content, qris_data['qris_content']) patched.assert_called_once() with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get ) as patched: # Check the API is not called again, as it should reuse the existing QR until it is expired self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() patched.assert_not_called() # Check that if the qr code has expired, we correctly generate a new one. qr_details.qris_creation_datetime = '2024-02-27 03:00:00' self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() patched.assert_called_once() self.assertEqual(len(self.qris_qr_invoice.l10n_id_qris_transaction_ids), 2) with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get_second ): self.qris_qr_invoice.l10n_id_qris_transaction_ids[-1]['qris_creation_datetime'] = '2024-02-27 03:00:00' self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() # Ensure that the l10n_id_latest_qris_content is the new one, while now there are 3 qris transactions linked to the inovice qr_details = self.qris_qr_invoice.l10n_id_qris_transaction_ids qris_data = self.success_qris_get_second['data'] self.assertEqual(qr_details[-1].qris_invoice_id, qris_data['qris_invoiceid']) self.assertEqual(qr_details[-1].qris_content, qris_data['qris_content']) self.assertEqual(len(self.qris_qr_invoice.l10n_id_qris_transaction_ids), 3) @freeze_time("2024-02-27 04:15:00") def test_fetch_payment_status_fail(self): """ If API return unpaid status """ with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get ): self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.qris_status_fail ): self.qris_qr_invoice.action_l10n_id_update_payment_status() self.assertEqual(self.qris_qr_invoice.payment_state, 'not_paid') @freeze_time("2024-02-27 04:15:00") def test_fetch_payment_status_success(self): """ If API returns 'paid' status """ with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get ): self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.qris_status_success ): self.qris_qr_invoice.action_l10n_id_update_payment_status() self.assertEqual(self.qris_qr_invoice.payment_state, self.env['account.move']._get_invoice_in_payment_state()) # Ensure that the message is logged as expected. self.assertEqual( self.qris_qr_invoice.message_ids[0].body, Markup('

This invoice was paid by Zainal Arief using QRIS with the payment method Sakuku.

') ) # One of the QRIS transactions linked to the invoice should be paid already self.assertTrue(any(self.qris_qr_invoice.l10n_id_qris_transaction_ids.mapped('paid'))) @freeze_time("2024-02-27 04:15:00") def test_gc_no_remove_transactions_unpaid_within_30(self): # Case 2: Unsuccessful, latest transaction is < 30 minutes, should avoid garbage-collection with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get ): self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.qris_status_fail ): self.qris_qr_invoice.action_l10n_id_update_payment_status() self.assertEqual(self.env['l10n_id.qris.transaction'].search_count([]), 1) self.env['l10n_id.qris.transaction']._gc_remove_pointless_qris_transactions() self.assertEqual(self.env['l10n_id.qris.transaction'].search_count([]), 1) @freeze_time("2024-02-27 05:15:00") def test_gc_remove_transactions_unpaid_after_30(self): # Case 3: Unsuccessful, latest transaction is > 30 minutes, should be garbage-collected with patch( 'odoo.addons.l10n_id.models.res_bank._l10n_id_make_qris_request', return_value=self.success_qris_get ): self.qris_qr_invoice.with_context({'is_online_qr': True})._generate_qr_code() self.assertEqual(self.env['l10n_id.qris.transaction'].search_count([]), 1) self.env['l10n_id.qris.transaction']._gc_remove_pointless_qris_transactions() self.assertEqual(self.env['l10n_id.qris.transaction'].search_count([]), 0)