server-tools/base_attachment_queue/tests/test_attachment_base_queue.py

# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo.tests.common import TransactionCase
import odoo
from odoo import api


class TestAttachmentBaseQueue(TransactionCase):

    def setUp(self):
        super().setUp()
        self.registry.enter_test_mode(self.env.cr)
        self.env = api.Environment(self.registry.test_cr, self.env.uid,
                                   self.env.context)
        self.attachment = self.env.ref(
            'base_attachment_queue.attachment_metadata')
        self.ir_attachment_metadata = self.env['ir.attachment.metadata']

    def tearDown(self):
        self.registry.leave_test_mode()
        super().tearDown()

    def test_attachment_metadata(self):
        """Test run_attachment_metadata_scheduler to ensure set state to done
        """
        self.assertEqual(
            self.attachment.state,
            'pending'
        )
        self.ir_attachment_metadata.run_attachment_metadata_scheduler()
        self.env.cache.invalidate()
        with odoo.registry(self.env.cr.dbname).cursor() as new_cr:
            new_env = api.Environment(
                new_cr, self.env.uid, self.env.context)
            attach = self.attachment.with_env(new_env)
            self.assertEqual(
                attach.state,
                'done'
            )

    def test_set_done(self):
        """Test set_done manually
        """
        self.assertEqual(
            self.attachment.state,
            'pending'
        )
        self.attachment.set_done()
        self.assertEqual(
            self.attachment.state,
            'done'
        )