mirror of
https://github.com/OCA/knowledge.git
synced 2025-07-24 17:38:42 -06:00
58 lines
1.7 KiB
Python
58 lines
1.7 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',
|
|
'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
|
|
)
|
|
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)
|
|
|
|
modules = self.env['ir.module.module'].search([])
|
|
module = modules.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)
|
|
|
|
att1 = self.env['ir.attachment'].create({
|
|
'name': 'Test Attachment Preview'
|
|
})
|
|
res4 = self.env['ir.attachment'].get_binary_extension(
|
|
'ir.attachment',
|
|
att1.id,
|
|
'datas'
|
|
)
|
|
self.assertFalse(res4)
|