from unittest.mock import patch

from werkzeug.exceptions import Forbidden

from odoo.exceptions import ValidationError
from odoo.tests import tagged
from odoo.tools import mute_logger

from odoo.addons.payment import utils as payment_utils
from odoo.addons.payment.tests.http_common import PaymentHttpCommon
from odoo.addons.payment_toss_payments import const
from odoo.addons.payment_toss_payments.controllers.main import TossPaymentsController
from odoo.addons.payment_toss_payments.tests.common import TossPaymentsCommon


@tagged('post_install', '-at_install')
class TestProcessingFlows(TossPaymentsCommon, PaymentHttpCommon):
    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_returning_from_successful_payment_initiation_triggers_processing(self):
        """Test that successfully initiating a payment triggers the processing of the payment
        data."""
        tx = self._create_transaction('direct')
        redirect_success_params = {'orderId': tx.reference, 'paymentKey': 'test-pk', 'amount': 750}
        url = self._build_url(const.PAYMENT_SUCCESS_RETURN_ROUTE)
        with (
            patch(
                'odoo.addons.payment.models.payment_transaction.PaymentTransaction'
                '._send_api_request'
            ),
            patch(
                'odoo.addons.payment.models.payment_transaction.PaymentTransaction._process'
            ) as process_mock,
        ):
            self._make_http_get_request(url, params=redirect_success_params)
        self.assertEqual(process_mock.call_count, 1)

    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_failing_to_confirm_payment_sets_the_transaction_in_error(self):
        """Test that the transaction is set in error if the payment confirmation request fails."""
        tx = self._create_transaction('direct')
        redirect_success_params = {'orderId': tx.reference, 'paymentKey': "test-pk", 'amount': 750}
        url = self._build_url(const.PAYMENT_SUCCESS_RETURN_ROUTE)
        with patch(
            'odoo.addons.payment.models.payment_transaction.PaymentTransaction._send_api_request',
            side_effect=ValidationError('dummy response'),
        ):
            self._make_http_get_request(url, params=redirect_success_params)
        self.assertEqual(tx.state, 'error')

    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_returning_from_failing_payment_initiation_sets_transaction_in_error(self):
        """Test that failing to initiate a payment set the transaction in error."""
        tx = self._create_transaction('direct')
        url = self._build_url(const.PAYMENT_FAILURE_RETURN_ROUTE)
        # Generate DB secret to verify the request is coming from Toss Payments since we
        # don't have paymentKey in the failure return URL to verify the request via API call.
        access_token = payment_utils.generate_access_token(tx.reference, env=self.env)
        error_data = {
            'code': 'ERR',
            'message': 'Payment refused',
            'orderId': tx.reference,
            'access_token': access_token,
        }
        self._make_http_get_request(url, params=error_data)
        self.assertEqual(tx.state, 'error')

    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_webhook_notification_triggers_processing(self):
        """Test that receiving a valid webhook notification triggers the processing of the payment
        data."""
        self._create_transaction('direct')
        url = self._build_url(const.WEBHOOK_ROUTE)
        with (
            patch(
                'odoo.addons.payment_toss_payments.controllers.main.TossPaymentsController'
                '._verify_signature'
            ),
            patch(
                'odoo.addons.payment.models.payment_transaction.PaymentTransaction._process'
            ) as process_mock,
        ):
            self._make_json_request(url, data=self.webhook_data)
        self.assertEqual(process_mock.call_count, 1)

    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_webhook_notification_triggers_signature_check(self):
        """Test that receiving a webhook notification triggers a signature check."""
        self._create_transaction('direct')
        url = self._build_url(const.WEBHOOK_ROUTE)
        with (
            patch(
                'odoo.addons.payment_toss_payments.controllers.main.TossPaymentsController'
                '._verify_signature'
            ) as signature_check_mock,
            patch('odoo.addons.payment.models.payment_transaction.PaymentTransaction._process'),
        ):
            self._make_json_request(url, data=self.webhook_data)
        self.assertEqual(signature_check_mock.call_count, 1)

    def test_accept_payment_data_with_valid_signature(self):
        """Test the verification of payment data with a valid signature."""
        tx = self._create_transaction('direct')
        tx.write({'toss_payments_payment_secret': 'test-secret'})
        self._assert_does_not_raise(
            Forbidden, TossPaymentsController._verify_signature, self.webhook_data['data'], tx
        )

    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_reject_payment_data_with_missing_signature(self):
        """Test the verification of payment data with a missing signature."""
        tx = self._create_transaction('direct')
        tx.write({'toss_payments_payment_secret': 'test-secret'})
        webhook_data = dict(self.webhook_data['data'], status='DONE', secret=None)
        self.assertRaises(Forbidden, TossPaymentsController._verify_signature, webhook_data, tx)

    @mute_logger('odoo.addons.payment_toss_payments.controllers.main')
    def test_reject_payment_data_with_invalid_signature(self):
        """Test the verification of payment data with an invalid signature."""
        tx = self._create_transaction('direct')
        tx.write({'toss_payments_payment_secret': 'test-secret'})
        webhook_data = dict(self.webhook_data['data'], status='DONE', secret='dummy')
        self.assertRaises(Forbidden, TossPaymentsController._verify_signature, webhook_data, tx)