server-tools/sql_export_mail/tests/test_sql_query_mail.py

# Copyright (C) 2019 Akretion (<http://www.akretion.com>)
# @author: Florian da Costa
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from odoo.tests.common import TransactionCase
from odoo import SUPERUSER_ID


class TestExportSqlQueryMail(TransactionCase):

    def setUp(self):
        super(TestExportSqlQueryMail, self).setUp()
        self.sql_report_demo = self.env.ref('sql_export.sql_export_partner')
        self.sql_report_demo.write({'mail_user_ids': [(4, SUPERUSER_ID)]})

    def test_sql_query_mail(self):
        mail_obj = self.env['mail.mail']
        mails = mail_obj.search(
            [('model', '=', 'sql.export'),
             ('res_id', '=', self.sql_report_demo.id)])
        self.assertFalse(mails)
        self.sql_report_demo.create_cron()
        self.assertTrue(self.sql_report_demo.cron_ids)
        self.sql_report_demo.cron_ids.method_direct_trigger()
        mails = mail_obj.search(
            [('model', '=', 'sql.export'),
             ('res_id', '=', self.sql_report_demo.id)])
        self.assertTrue(mails)
        self.assertTrue(mails.attachment_ids)