# 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)