From 05ae347bdd5875aeb35d2f10f5c7fb1a1f75a460 Mon Sep 17 00:00:00 2001 From: Yannick Tivisse Date: Mon, 4 Nov 2019 16:50:30 +0100 Subject: [IMP] website: Adapt tests to work with/without demo data --- addons/website/tests/test_get_current_website.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'addons/website/tests/test_get_current_website.py') diff --git a/addons/website/tests/test_get_current_website.py b/addons/website/tests/test_get_current_website.py index ee02a4d5ff8..da36762acee 100644 --- a/addons/website/tests/test_get_current_website.py +++ b/addons/website/tests/test_get_current_website.py @@ -14,13 +14,15 @@ class TestGetCurrentWebsite(TransactionCase): Website = self.env['website'] # clean initial state - website1 = Website.browse(1) + website1 = self.env.ref('website.default_website') website1.domain = '' website1.country_group_ids = False - website2 = Website.browse(2) - website2.domain = '' - website2.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"}) @@ -114,7 +116,7 @@ class TestGetCurrentWebsite(TransactionCase): self.assertEqual(Website._get_current_website_id('site-1.com', False), website1.id) def test_02_signup_user_website_id(self): - website = self.env['website'].browse(1) + 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'}) -- cgit v1.2.3