mirror of
https://github.com/OCA/knowledge.git
synced 2025-07-20 10:16:45 -06:00
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
# Copyright 2018 Onestein
|
|
# 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"}
|
|
)
|
|
attachment2 = self.env["ir.attachment"].create(
|
|
{"datas": base64.b64encode(b"Png"), "name": "image.png"}
|
|
)
|
|
res = self.env["ir.attachment"].get_attachment_extension(attachment.id)
|
|
self.assertEqual(res, "txt")
|
|
|
|
res = self.env["ir.attachment"].get_attachment_extension(
|
|
[attachment.id, attachment2.id]
|
|
)
|
|
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"
|
|
)
|
|
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"
|
|
)
|
|
self.assertTrue(res3)
|