server-tools/attachment_synchronize/tests/common.py

# Copyright 2020 Akretion (http://www.akretion.com).
# @author Sébastien BEAU <sebastien.beau@akretion.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

import mock
import os
from odoo.addons.storage_backend.tests.common import Common


class SyncCommon(Common):
    def _clean_testing_directory(self):
        for test_dir in [
            self.directory_input,
            self.directory_output,
            self.directory_archived,
        ]:
            for filename in self.backend._list(test_dir):
                self.backend._delete(os.path.join(test_dir, filename))

    def _create_test_file(self):
        self.backend._add_b64_data(
            os.path.join(self.directory_input, "bar.txt"),
            self.filedata,
            mimetype=u"text/plain",
        )

    def setUp(self):
        super().setUp()
        self.env.cr.commit = mock.Mock()
        self.registry.enter_test_mode(self.env.cr)
        self.directory_input = "test_import"
        self.directory_output = "test_export"
        self.directory_archived = "test_archived"
        self._clean_testing_directory()
        self._create_test_file()
        self.task = self.env.ref(
            "attachment_synchronize.import_from_filestore"
        )

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