1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
# -*- 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):
def test_01_get_current_website_id(self):
"""Make sure `_get_current_website_id works`."""
Website = self.env['website']
# clean initial state
website1 = self.env.ref('website.default_website')
website1.domain = ''
website1.country_group_ids = False
website2 = Website.create({
'name': 'My Website 2',
'domain': '',
'country_group_ids': False,
})
country1 = self.env['res.country'].create({'name': "My Country 1"})
country2 = self.env['res.country'].create({'name': "My Country 2"})
country3 = self.env['res.country'].create({'name': "My Country 3"})
country4 = self.env['res.country'].create({'name': "My Country 4"})
country5 = self.env['res.country'].create({'name': "My Country 5"})
country_group_1_2 = self.env['res.country.group'].create({
'name': "My Country Group 1-2",
'country_ids': [(6, 0, (country1 + country2 + country5).ids)],
})
country_group_3 = self.env['res.country.group'].create({
'name': "My Country Group 3",
'country_ids': [(6, 0, (country3 + country5).ids)],
})
# CASE: no domain, no country: get first
self.assertEqual(Website._get_current_website_id('', False), website1.id)
self.assertEqual(Website._get_current_website_id('', country3.id), website1.id)
# CASE: no domain, given country: get by country
website1.country_group_ids = country_group_1_2
website2.country_group_ids = country_group_3
self.assertEqual(Website._get_current_website_id('', country1.id), website1.id)
self.assertEqual(Website._get_current_website_id('', country2.id), website1.id)
self.assertEqual(Website._get_current_website_id('', country3.id), website2.id)
# CASE: no domain, wrong country: get first
self.assertEqual(Website._get_current_website_id('', country4.id), Website.search([]).sorted('country_group_ids')[0].id)
# CASE: no domain, multiple country: get first
self.assertEqual(Website._get_current_website_id('', country5.id), website1.id)
# setup domain
website1.domain = 'my-site-1.fr'
website2.domain = 'https://my2ndsite.com:80'
website1.country_group_ids = False
website2.country_group_ids = False
# CASE: domain set: get matching domain
self.assertEqual(Website._get_current_website_id('my-site-1.fr', False), website1.id)
# CASE: domain set: get matching domain (scheme and port supported)
self.assertEqual(Website._get_current_website_id('my-site-1.fr:8069', False), website1.id)
self.assertEqual(Website._get_current_website_id('my2ndsite.com:80', False), website2.id)
self.assertEqual(Website._get_current_website_id('my2ndsite.com:8069', False), website2.id)
self.assertEqual(Website._get_current_website_id('my2ndsite.com', False), website2.id)
# CASE: domain set, wrong domain: get first
self.assertEqual(Website._get_current_website_id('test.com', False), website1.id)
# CASE: subdomain: not supported
self.assertEqual(Website._get_current_website_id('www.my2ndsite.com', False), website1.id)
# CASE: domain set, given country: get by domain in priority
website1.country_group_ids = country_group_1_2
website2.country_group_ids = country_group_3
self.assertEqual(Website._get_current_website_id('my2ndsite.com', country1.id), website2.id)
self.assertEqual(Website._get_current_website_id('my2ndsite.com', country2.id), website2.id)
self.assertEqual(Website._get_current_website_id('my-site-1.fr', country3.id), website1.id)
# CASE: domain set, wrong country: get first for domain
self.assertEqual(Website._get_current_website_id('my2ndsite.com', country4.id), website2.id)
# CASE: domain set, multiple country: get first for domain
website1.domain = website2.domain
self.assertEqual(Website._get_current_website_id('my2ndsite.com', country5.id), 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', False), website1.id)
self.assertEqual(Website._get_current_website_id('even-better-site-1.com', False), 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', False), website1.id)
self.assertEqual(Website._get_current_website_id('even-beTTer-site-1.com', False), 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', False), website1.id)
self.assertEqual(Website._get_current_website_id('site-1.com:81', False), website2.id)
self.assertEqual(Website._get_current_website_id('site-1.com:82', False), website1.id)
self.assertEqual(Website._get_current_website_id('site-1.com', False), website1.id)
def test_02_signup_user_website_id(self):
website = self.env.ref('website.default_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)
|
