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

72 lines
3.3 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from unittest.mock import patch
from werkzeug.exceptions import Forbidden
from odoo.tests import tagged
from odoo.tools import mute_logger
from odoo.addons.payment.tests.http_common import PaymentHttpCommon
from odoo.addons.payment_asiapay.controllers.main import AsiaPayController
from odoo.addons.payment_asiapay.tests.common import AsiaPayCommon
@tagged('post_install', '-at_install')
class TestProcessingFlows(AsiaPayCommon, PaymentHttpCommon):
@mute_logger('odoo.addons.payment_asiapay.controllers.main')
def test_webhook_notification_triggers_processing(self):
""" Test that receiving a valid webhook notification triggers the processing of the
notification data. """
self._create_transaction('redirect')
url = self._build_url(AsiaPayController._webhook_url)
with patch(
'odoo.addons.payment_asiapay.controllers.main.AsiaPayController.'
'_verify_notification_signature'
), patch(
'odoo.addons.payment.models.payment_transaction.PaymentTransaction'
'._handle_notification_data'
) as handle_notification_data_mock:
self._make_http_post_request(url, data=self.webhook_notification_data)
self.assertEqual(handle_notification_data_mock.call_count, 1)
@mute_logger('odoo.addons.payment_asiapay.controllers.main')
def test_webhook_notification_triggers_signature_check(self):
""" Test that receiving a webhook notification triggers a signature check. """
self._create_transaction('redirect')
url = self._build_url(AsiaPayController._webhook_url)
with patch(
'odoo.addons.payment_asiapay.controllers.main.AsiaPayController'
'._verify_notification_signature'
) as signature_check_mock, patch(
'odoo.addons.payment.models.payment_transaction.PaymentTransaction'
'._handle_notification_data'
):
self._make_http_post_request(url, data=self.webhook_notification_data)
self.assertEqual(signature_check_mock.call_count, 1)
def test_accept_webhook_notification_with_valid_signature(self):
""" Test the verification of a webhook notification with a valid signature. """
tx = self._create_transaction('redirect')
self._assert_does_not_raise(
Forbidden,
AsiaPayController._verify_notification_signature,
self.webhook_notification_data,
tx,
)
@mute_logger('odoo.addons.payment_asiapay.controllers.main')
def test_reject_notification_with_missing_signature(self):
""" Test the verification of a notification with a missing signature. """
tx = self._create_transaction('redirect')
payload = dict(self.webhook_notification_data, secureHash='dummy')
self.assertRaises(Forbidden, AsiaPayController._verify_notification_signature, payload, tx)
@mute_logger('odoo.addons.payment_asiapay.controllers.main')
def test_reject_notification_with_invalid_signature(self):
""" Test the verification of a notification with an invalid signature. """
tx = self._create_transaction('redirect')
payload = dict(self.webhook_notification_data, secureHash='dummy')
self.assertRaises(Forbidden, AsiaPayController._verify_notification_signature, payload, tx)