mirror of
https://github.com/OCA/knowledge.git
synced 2025-07-21 03:27:25 -06:00
3d42b35eac
[FIX] Lint [FIX] lint and flake [ADD] tests [ADD] tests [ADD] tests [ADD] Tests [ADD] Package python-magic [ADD] Tests [FIX] Lint
47 lines
1.4 KiB
Python
47 lines
1.4 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)
|
|
|
|
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)
|