mirror of
https://github.com/OCA/knowledge.git
synced 2025-12-22 21:32:18 -06:00
[IMP] attachment_preview: black, isort, prettier
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user