forked from Techsystech/web
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
# Copyright 2024 OERP Canada <https://www.oerp.ca>
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)
|
|
|
|
from PIL import Image
|
|
|
|
from odoo.tests import tagged
|
|
from odoo.tests.common import TransactionCase
|
|
from odoo.tools import base64_to_image, image_to_base64
|
|
|
|
from odoo.addons.website.tools import MockRequest
|
|
|
|
|
|
@tagged("post_install", "-at_install")
|
|
class TestWebFavicon(TransactionCase):
|
|
def test_01_web_favicon(self):
|
|
"""The goal of this test is to make sure the favicon is correctly
|
|
handled on the backend."""
|
|
|
|
# Test setting an Ico file directly, done through create
|
|
Company = self.env["res.company"]
|
|
|
|
company = Company.create(
|
|
{
|
|
"name": "Test Company",
|
|
"favicon": Company._get_default_favicon(),
|
|
}
|
|
)
|
|
|
|
image = base64_to_image(company.favicon)
|
|
self.assertEqual(image.format, "ICO")
|
|
|
|
# Test setting a JPEG file that is too big, done through write
|
|
bg_color = (135, 90, 123)
|
|
image = Image.new("RGB", (1920, 1080), color=bg_color)
|
|
company.favicon = image_to_base64(image, "JPEG")
|
|
image = base64_to_image(company.favicon)
|
|
self.assertEqual(image.format, "JPEG")
|
|
self.assertEqual(image.size, (1920, 1080))
|
|
self.assertEqual(image.getpixel((0, 0)), bg_color)
|
|
with MockRequest(self.env) as mock_request:
|
|
mock_request.httprequest.cookies = {"cids": str(company.id)}
|
|
self.assertTrue(Company._get_favicon())
|
|
|
|
def test_02_default_favicon_creation(self):
|
|
"""Test if default favicon is set when creating a company without favicon."""
|
|
Company = self.env["res.company"]
|
|
company = Company.create({"name": "Test Company"})
|
|
self.assertTrue(company.favicon, "Default favicon not set on company creation.")
|
|
|
|
def test_03_website_favicon(self):
|
|
"""Test if favicon URL is correctly returned when website_id is in context."""
|
|
if self.env["ir.module.module"].search(
|
|
[("name", "=", "website"), ("state", "=", "installed")]
|
|
):
|
|
company = self.env["res.company"].create(
|
|
{
|
|
"name": "Test Company with Website",
|
|
}
|
|
)
|
|
website = self.env["website"].create(
|
|
{
|
|
"name": "Test Website",
|
|
"domain": "www.test.com",
|
|
"company_id": company.id,
|
|
}
|
|
)
|
|
website.favicon = website._default_favicon()
|
|
favicon_url = company.with_context(website_id=website.id)._get_favicon()
|
|
expected_favicon_url = website.image_url(website, "favicon")
|
|
self.assertEqual(
|
|
favicon_url,
|
|
expected_favicon_url,
|
|
"The favicon URL should match the expected value.",
|
|
)
|