53 lines
2.0 KiB
Python
53 lines
2.0 KiB
Python
|
# Copyright 2017 Akretion (http://www.akretion.com).
|
||
|
|
# @author Sébastien BEAU <sebastien.beau@akretion.com>
|
||
|
|
# Copyright 2021 Camptocamp (http://www.camptocamp.com).
|
||
|
|
# @author Iván Todorovich <ivan.todorovich@gmail.com>
|
||
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
||
|
|
|
||
|
|
import base64
|
||
|
|
import os
|
||
|
|
|
||
|
|
from odoo.addons.component.tests.common import SavepointComponentCase
|
||
|
|
|
||
|
|
|
||
|
|
class StorageImageCommonCase(SavepointComponentCase):
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
super().setUpClass()
|
||
|
|
cls.base_path = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
cls.env = cls.env(context=dict(cls.env.context, tracking_disable=True))
|
||
|
|
# Run tests with demo user
|
||
|
|
cls.user = cls.env.ref("base.user_demo")
|
||
|
|
cls.user.write(
|
||
|
|
{"groups_id": [(4, cls.env.ref("storage_image.group_image_manager").id)]}
|
||
|
|
)
|
||
|
|
cls.env = cls.env(user=cls.user)
|
||
|
|
# Storage backend
|
||
|
|
cls.backend = cls.env.ref("storage_backend.default_storage_backend")
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _get_file_content(cls, name, base_path=None):
|
||
|
|
path = base_path or cls.base_path
|
||
|
|
with open(os.path.join(path, name), "rb") as f:
|
||
|
|
data = f.read()
|
||
|
|
return data
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _create_storage_image(cls, filename, data):
|
||
|
|
return cls.env["storage.image"].create({"name": filename, "data": data})
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _create_storage_image_from_file(cls, filename, base_path=None):
|
||
|
|
data = cls._get_file_content(filename, base_path=base_path)
|
||
|
|
data = base64.b64encode(data)
|
||
|
|
return cls._create_storage_image(os.path.basename(filename), data)
|
||
|
|
|
||
|
|
def _check_thumbnail(self, image):
|
||
|
|
self.assertEqual(len(image.thumbnail_ids), 2)
|
||
|
|
medium = image._get_thumb("medium")
|
||
|
|
small = image._get_thumb("small")
|
||
|
|
self.assertEqual(medium.size_x, 128)
|
||
|
|
self.assertEqual(medium.size_y, 128)
|
||
|
|
self.assertEqual(small.size_x, 64)
|
||
|
|
self.assertEqual(small.size_y, 64)
|