initial commit 2

1 files changed, 16 insertions, 0 deletions

diff --git a/addons/test_website/tests/test_multi_company.py b/addons/test_website/tests/test_multi_company.py
new file mode 100644
index 00000000..1395466a
--- /dev/null
+++ b/addons/test_website/tests/test_multi_company.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo.tests.common import HttpCase, tagged
+
+
+@tagged('post_install', '-at_install')
+class TestMultiCompany(HttpCase):
+
+    def test_company_in_context(self):
+        """ Test website company is set in context """
+        website = self.env.ref('website.default_website')
+        company = self.env['res.company'].create({'name': "Adaa"})
+        website.company_id = company
+        response = self.url_open('/multi_company_website')
+        self.assertEqual(response.json()[0], company.id)