oca-ocb-sale/odoo-bringout-oca-ocb-website_sale/website_sale/tests/test_main_controller.py
Ernad Husremovic 73afc09215 19.0 vanilla
2026-03-09 09:32:12 +01:00

54 lines
2.3 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from unittest.mock import patch
from odoo.tests import tagged
from odoo.addons.payment.models.payment_provider import PaymentProvider
from odoo.addons.payment.tests.http_common import PaymentHttpCommon
from odoo.addons.sale.tests.common import SaleCommon
@tagged('post_install', '-at_install')
class TestPaymentProviderVisibility(PaymentHttpCommon, SaleCommon):
def test_payment_provider_visibility_with_portal(self):
"""Check providers availability on the sales portal.
The current website must be considered to filter the providers.
"""
website_portal = self.env['website'].get_current_website()
website_shop = self.env['website'].create({'name': "Shop Website"})
base_url = self.env['ir.config_parameter'].sudo().get_base_url()
website_portal.write({'domain': base_url})
self.provider.write({'website_id': website_portal.id})
restricted_provider = self.env['payment.provider'].sudo().search([('name', '=', 'Demo'), ('company_id', '=', website_shop.company_id.id)])
restricted_provider.write({'state': 'test', 'website_id': website_shop.id})
url_so = self.sale_order.get_portal_url()
self.sale_order.require_payment = True
portal_url = f"{website_portal.domain}{url_so}"
with patch(
'odoo.addons.website_payment.models.payment_provider.PaymentProvider._get_compatible_providers',
side_effect=lambda *args, **kwargs: PaymentProvider._get_compatible_providers(
self.env['payment.provider'], *args, **kwargs
),
) as mock_method:
self.url_open(portal_url, allow_redirects=True)
mock_method.assert_called_once()
self.assertEqual(mock_method.call_args.kwargs['website_id'], website_portal.id)
mock_method.call_args.kwargs.pop('show_non_tokenize_provider', None)
providers = self.env['payment.provider']._get_compatible_providers(
*mock_method.call_args.args, **mock_method.call_args.kwargs
)
self.assertIn(self.provider.id, providers.ids, "The visible provider should be visible.")
self.assertNotIn(
restricted_provider.id, providers.ids, "The restricted provider shouldn't be visible."
)