[IMP] attachment_preview: black, isort, prettier

This commit is contained in:
vancouver29
2022-08-16 17:54:32 +02:00
parent aa48e38d3f
commit c6e0419f98
10 changed files with 324 additions and 243 deletions

View File

@@ -2,45 +2,44 @@
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import base64
from odoo.tests.common import TransactionCase
class TestAttachmentPreview(TransactionCase):
def test_get_extension(self):
attachment = self.env['ir.attachment'].create({
'datas': base64.b64encode(b'from this, to that.'),
'name': 'doc.txt',
'datas_fname': 'doc.txt'
})
attachment2 = self.env['ir.attachment'].create({
'datas': base64.b64encode(b'Png'),
'name': 'image.png',
'datas_fname': 'image.png'
})
res = self.env['ir.attachment'].get_attachment_extension(
attachment.id
attachment = self.env["ir.attachment"].create(
{
"datas": base64.b64encode(b"from this, to that."),
"name": "doc.txt",
"datas_fname": "doc.txt",
}
)
self.assertEqual(res, 'txt')
attachment2 = self.env["ir.attachment"].create(
{
"datas": base64.b64encode(b"Png"),
"name": "image.png",
"datas_fname": "image.png",
}
)
res = self.env["ir.attachment"].get_attachment_extension(attachment.id)
self.assertEqual(res, "txt")
res = self.env['ir.attachment'].get_attachment_extension(
res = self.env["ir.attachment"].get_attachment_extension(
[attachment.id, attachment2.id]
)
self.assertEqual(res[attachment.id], 'txt')
self.assertEqual(res[attachment2.id], 'png')
self.assertEqual(res[attachment.id], "txt")
self.assertEqual(res[attachment2.id], "png")
res2 = self.env['ir.attachment'].get_binary_extension(
'ir.attachment',
attachment.id,
'datas'
res2 = self.env["ir.attachment"].get_binary_extension(
"ir.attachment", attachment.id, "datas"
)
self.assertTrue(res2)
module = self.env['ir.module.module'].search([]).filtered(
lambda m: m.icon_image
)[0]
res3 = self.env['ir.attachment'].get_binary_extension(
'ir.module.module',
module.id,
'icon_image'
module = (
self.env["ir.module.module"].search([]).filtered(lambda m: m.icon_image)[0]
)
res3 = self.env["ir.attachment"].get_binary_extension(
"ir.module.module", module.id, "icon_image"
)
self.assertTrue(res3)