32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from odoo import Command
|
|
from odoo.addons.documents.tests.test_documents_common import TEXT
|
|
from odoo.addons.documents.tools import attachment_read
|
|
from odoo.tests.common import TransactionCase
|
|
|
|
|
|
class TestTools(TransactionCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
cls.document_manager = cls.env['res.users'].create([
|
|
{
|
|
'email': "dtdm@yourcompany.com",
|
|
'group_ids': [Command.link(cls.env.ref('documents.group_documents_manager').id)],
|
|
'login': "dtdm",
|
|
'name': "Documents Manager",
|
|
}
|
|
])
|
|
cls.document_txt = cls.env['documents.document'].create({
|
|
'type': 'binary',
|
|
'datas': TEXT,
|
|
'name': 'file.txt',
|
|
'mimetype': 'text/plain',
|
|
'owner_id': cls.document_manager.id,
|
|
})
|
|
|
|
def test_attachment_read(self):
|
|
attachment_sudo = self.document_txt.attachment_id.sudo()
|
|
self.assertEqual(attachment_sudo.raw, b'TEST')
|
|
self.assertEqual(attachment_read(attachment_sudo, 2), b'TE')
|