# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. from odoo.tests import tagged from odoo.tests.common import TransactionCase @tagged('post_install', '-at_install') class TestGetCurrentWebsite(TransactionCase): @classmethod def setUpClass(cls): super().setUpClass() cls.website = cls.env.ref('website.default_website') def test_01_get_current_website_id(self): """Make sure `_get_current_website_id works`.""" Website = self.env['website'] # clean initial state website1 = self.website website1.domain = False website2 = Website.create({'name': 'My Website 2'}) # CASE: no domain: get first self.assertEqual(Website._get_current_website_id(''), website1.id) # setup domain website1.domain = 'my-site-1.fr' website2.domain = 'https://my2ndsite.com:80' # CASE: domain set: get matching domain self.assertEqual(Website._get_current_website_id('my-site-1.fr'), website1.id) # CASE: domain set: get matching domain (scheme and port supported) self.assertEqual(Website._get_current_website_id('my-site-1.fr:8069'), website1.id) self.assertEqual(Website._get_current_website_id('my2ndsite.com:80'), website2.id) self.assertEqual(Website._get_current_website_id('my2ndsite.com:8069'), website2.id) self.assertEqual(Website._get_current_website_id('my2ndsite.com'), website2.id) # CASE: domain set, wrong domain: get first self.assertEqual(Website._get_current_website_id('test.com'), website1.id) # CASE: subdomain: not supported self.assertEqual(Website._get_current_website_id('www.my2ndsite.com'), website1.id) # CASE: domain set: get by domain in priority self.assertEqual(Website._get_current_website_id('my2ndsite.com'), website2.id) self.assertEqual(Website._get_current_website_id('my-site-1.fr'), website1.id) # CASE: overlapping domain: get exact match website1.domain = 'site-1.com' website2.domain = 'even-better-site-1.com' self.assertEqual(Website._get_current_website_id('site-1.com'), website1.id) self.assertEqual(Website._get_current_website_id('even-better-site-1.com'), website2.id) # CASE: case insensitive website1.domain = 'Site-1.com' website2.domain = 'Even-Better-site-1.com' self.assertEqual(Website._get_current_website_id('sitE-1.com'), website1.id) self.assertEqual(Website._get_current_website_id('even-beTTer-site-1.com'), website2.id) # CASE: same domain, different port website1.domain = 'site-1.com:80' website2.domain = 'site-1.com:81' self.assertEqual(Website._get_current_website_id('site-1.com:80'), website1.id) self.assertEqual(Website._get_current_website_id('site-1.com:81'), website2.id) self.assertEqual(Website._get_current_website_id('site-1.com:82'), website1.id) self.assertEqual(Website._get_current_website_id('site-1.com'), website1.id) def test_02_signup_user_website_id(self): website = self.website website.specific_user_account = True user = self.env['res.users'].create({'website_id': website.id, 'login': 'sad@mail.com', 'name': 'Hope Fully'}) self.assertTrue(user.website_id == user.partner_id.website_id == website)