# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. import json from datetime import date from unittest.mock import patch from odoo import Command from odoo.addons.account.tests.common import AccountTestInvoicingCommon from odoo.addons.mail.tests.common import MailCommon from odoo.exceptions import UserError from odoo.tests.common import users, warmup from odoo.tests import tagged from odoo.tools import formataddr, mute_logger @tagged('post_install_l10n', 'post_install', '-at_install') class TestAccountComposerPerformance(AccountTestInvoicingCommon, MailCommon): @classmethod def setUpClass(cls): super().setUpClass() # ensure print params cls.user_accountman = cls.env.user # main account setup shadows users, better save it cls.company_main = cls.company_data['company'] cls.company_main.invoice_is_email = True cls.company_main.invoice_is_download = False cls.move_template = cls.env['mail.template'].create({ 'auto_delete': True, 'body_html': '

TemplateBody for

', 'description': 'Sent to customers with their invoices in attachment', 'email_from': "{{ (object.invoice_user_id.email_formatted or user.email_formatted) }}", 'mail_server_id': cls.mail_server_default.id, 'model_id': cls.env['ir.model']._get_id('account.move'), 'name': "Invoice: Test Sending", 'partner_to': "{{ object.partner_id.id }}", 'subject': "{{ object.company_id.name }} Invoice (Ref {{ object.name or 'n/a' }})", 'report_template_ids': [(4, cls.env.ref('account.account_invoices').id)], 'lang': "{{ object.partner_id.lang }}", }) cls.attachments = cls.env['ir.attachment'].create( cls._generate_attachments_data( 2, cls.move_template._name, cls.move_template.id ) ) cls.move_template.write({ 'attachment_ids': [(6, 0, cls.attachments.ids)] }) # test users + fetch admin user for testing (recipient, ...) cls.user_account = cls.env['res.users'].with_context(cls._test_context).create({ 'company_id': cls.company_main.id, 'company_ids': [ (6, 0, (cls.company_data['company'] + cls.company_data_2['company']).ids) ], 'country_id': cls.env.ref('base.be').id, 'email': 'e.e@example.com', 'groups_id': [ (6, 0, [cls.env.ref('base.group_user').id, cls.env.ref('account.group_account_invoice').id, cls.env.ref('base.group_partner_manager').id ]) ], 'login': 'user_account', 'name': 'Ernest Employee Account', 'notification_type': 'inbox', 'signature': '--\nErnest', }) cls.user_account_other = cls.env['res.users'].with_context(cls._test_context).create({ 'company_id': cls.company_admin.id, 'company_ids': [(4, cls.company_admin.id)], 'country_id': cls.env.ref('base.be').id, 'email': 'e.e.other@example.com', 'groups_id': [ (6, 0, [cls.env.ref('base.group_user').id, cls.env.ref('account.group_account_invoice').id, cls.env.ref('base.group_partner_manager').id ]) ], 'login': 'user_account_other', 'name': 'Eglantine Employee AccountOther', 'notification_type': 'inbox', 'signature': '--\nEglantine', }) # mass mode: 10 invoices with their customer country_id = cls.env.ref('base.be').id langs = ['en_US', 'es_ES'] cls.env['res.lang']._activate_lang('es_ES') cls.test_customers = cls.env['res.partner'].create([ {'country_id': country_id, 'email': f'test_partner_{idx}@test.example.com', 'mobile': f'047500{idx:2d}{idx:2d}', 'lang': langs[idx % len(langs)], 'name': f'Partner_{idx}', } for idx in range(0, 10) ]) cls.test_account_moves = cls.env['account.move'].create([{ 'invoice_date': date(2022, 3, 2), 'invoice_date_due': date(2022, 3, 10), 'invoice_line_ids': [ (0, 0, {'name': 'Line1', 'price_unit': 100.0 } ), (0, 0, {'name': 'Line2', 'price_unit': 200.0 } ), ], 'invoice_user_id': cls.user_account_other.id, 'move_type': 'out_invoice', 'name': f'INVOICE_{idx:02d}', 'partner_id': cls.test_customers[idx].id, } for idx in range(0, 10)]) cls.test_account_moves.action_post() # test impact of multi language support cls._activate_multi_lang( test_record=cls.test_account_moves, test_template=cls.move_template, ) def setUp(self): super().setUp() # patch registry to simulate a ready environment self.patch(self.env.registry, 'ready', True) self.flush_tracking() def test_assert_initial_values(self): """ Test initial values to ease understanding of results and notifications """ for move in self.test_account_moves: with self.subTest(move=move): self.assertEqual( move.invoice_user_id, self.user_account_other, ) self.assertEqual( move.message_partner_ids, self.user_accountman.partner_id + move.partner_id, ) @users('user_account') @warmup @mute_logger('odoo.addons.mail.models.mail_mail', 'odoo.models.unlink') def test_move_composer_multi(self): """ Test with multi mode """ test_moves = self.test_account_moves.with_env(self.env) test_customers = self.test_customers.with_env(self.env) move_template = self.move_template.with_env(self.env) for test_move in test_moves: self.assertFalse(test_move.is_move_sent) composer = self.env['account.move.send']\ .with_context(active_model='account.move', active_ids=test_moves.ids)\ .create({'mail_template_id': move_template.id}) with self.mock_mail_gateway(mail_unlink_sent=False): composer.action_send_and_print(force_synchronous=True) # check results: emails (mailing mode when being in multi) self.assertEqual(len(self._mails), 20, 'Should send an email to each invoice followers (accountman + partner)') for move, customer in zip(test_moves, test_customers): with self.subTest(move=move, customer=customer): _exp_move_name = move.name _exp_report_name = f"{move.name}.pdf" if move.partner_id.lang == 'es_ES': _exp_body_tip = f'SpanishBody for {move.name}' _exp_subject = f'SpanishSubject for {move.name}' else: _exp_body_tip = f'TemplateBody for {move.name}' _exp_subject = f'{self.env.user.company_id.name} Invoice (Ref {move.name})' self.assertEqual(move.partner_id, customer) self.assertMailMail( customer, 'sent', author=self.user_account_other.partner_id, # author: synchronized with email_from of template content=_exp_body_tip, email_values={ 'attachments_info': [ {'name': 'AttFileName_00.txt', 'raw': b'AttContent_00', 'type': 'text/plain'}, {'name': 'AttFileName_01.txt', 'raw': b'AttContent_01', 'type': 'text/plain'}, {'name': _exp_report_name, 'type': 'application/pdf'}, ], 'body_content': _exp_body_tip, 'email_from': self.user_account_other.email_formatted, 'subject': _exp_subject, 'reply_to': formataddr(( f'{move.company_id.name} {_exp_move_name}', f'{self.alias_catchall}@{self.alias_domain}' )), }, fields_values={ 'auto_delete': True, 'email_from': self.user_account_other.email_formatted, 'is_notification': True, # should keep logs by default 'mail_server_id': self.mail_server_default, 'subject': _exp_subject, 'reply_to': formataddr(( f'{move.company_id.name} {_exp_move_name}', f'{self.alias_catchall}@{self.alias_domain}' )), }, ) # composer configuration self.assertEqual(composer.move_ids, test_moves) self.assertTrue(composer.checkbox_send_mail) self.assertFalse(composer.checkbox_download) self.assertEqual(composer.mail_template_id, move_template) # invoice update for test_move in test_moves: self.assertTrue(test_move.is_move_sent) @users('user_account') @warmup @mute_logger('odoo.addons.mail.models.mail_mail', 'odoo.models.unlink') def test_move_composer_single(self): """ Test single mode """ test_move = self.test_account_moves[0].with_env(self.env) test_customer = self.test_customers[0].with_env(self.env) move_template = self.move_template.with_env(self.env) composer = self.env['account.move.send']\ .with_context(active_model='account.move', active_ids=test_move.ids)\ .create({'mail_template_id': move_template.id}) with self.mock_mail_gateway(mail_unlink_sent=False), \ self.mock_mail_app(): composer.action_send_and_print(allow_fallback_pdf=True) self.env.cr.flush() # force tracking message self.assertEqual(len(self._new_msgs), 2, 'Should produce 2 messages: one for posting template, one for tracking') print_msg, track_msg = self._new_msgs[0], self._new_msgs[1] self.assertNotified( print_msg, [{ 'is_read': True, 'partner': test_customer, 'type': 'email', }], ) # print: template-based message self.assertEqual(len(print_msg.attachment_ids), 3) self.assertNotIn(self.attachments, print_msg.attachment_ids, 'Attachments should be duplicated, not just linked') self.assertEqual(print_msg.author_id, self.user_account_other.partner_id, 'Should take invoice_user_id partner') self.assertEqual(print_msg.email_from, self.user_account_other.email_formatted, 'Should take invoice_user_id email') self.assertEqual(print_msg.notified_partner_ids, test_customer + self.user_accountman.partner_id) self.assertEqual(print_msg.subject, f'{self.env.user.company_id.name} Invoice (Ref {test_move.name})') # tracking: is_move_sent self.assertEqual(track_msg.author_id, self.env.user.partner_id) self.assertEqual(track_msg.email_from, self.env.user.email_formatted) self.assertTrue('is_move_sent' in track_msg.tracking_value_ids.field_id.mapped('name')) # sent email self.assertMailMail( test_customer, 'sent', author=self.user_account_other.partner_id, # author: synchronized with email_from of template content=f'TemplateBody for {test_move.name}', email_values={ 'attachments_info': [ {'name': 'AttFileName_00.txt', 'raw': b'AttContent_00', 'type': 'text/plain'}, {'name': 'AttFileName_01.txt', 'raw': b'AttContent_01', 'type': 'text/plain'}, {'name': f'{test_move.name}.pdf', 'type': 'application/pdf'}, ], 'body_content': f'TemplateBody for {test_move.name}', 'email_from': self.user_account_other.email_formatted, 'subject': f'{self.env.user.company_id.name} Invoice (Ref {test_move.name})', 'reply_to': formataddr(( f'{test_move.company_id.name} {test_move.display_name}', f'{self.alias_catchall}@{self.alias_domain}' )), }, fields_values={ 'auto_delete': True, 'email_from': self.user_account_other.email_formatted, 'is_notification': True, # should keep logs by default 'mail_server_id': self.mail_server_default, 'subject': f'{self.env.user.company_id.name} Invoice (Ref {test_move.name})', 'reply_to': formataddr(( f'{test_move.company_id.name} {test_move.display_name}', f'{self.alias_catchall}@{self.alias_domain}' )), }, ) # composer configuration self.assertIn(f'TemplateBody for {test_move.name}', composer.mail_body) self.assertEqual(composer.move_ids, test_move) self.assertTrue(composer.checkbox_send_mail) self.assertFalse(composer.checkbox_download) self.assertEqual(composer.mail_subject, f'{self.env.user.company_id.name} Invoice (Ref {test_move.name})') self.assertEqual(composer.mail_template_id, move_template) # invoice update self.assertTrue(test_move.is_move_sent) @users('user_account') @warmup @mute_logger('odoo.addons.mail.models.mail_mail', 'odoo.models.unlink') def test_move_composer_single_lang(self): """ Test single with another language """ test_move = self.test_account_moves[1].with_env(self.env) test_customer = self.test_customers[1].with_env(self.env) move_template = self.move_template.with_env(self.env) composer = self.env['account.move.send']\ .with_context(active_model='account.move', active_ids=test_move.ids)\ .create({'mail_template_id': move_template.id}) with self.mock_mail_gateway(mail_unlink_sent=False), \ self.mock_mail_app(): composer.action_send_and_print() self.env.cr.flush() # force tracking message self.assertEqual(len(self._new_msgs), 2, 'Should produce 2 messages: one for posting template, one for tracking') print_msg, track_msg = self._new_msgs[0], self._new_msgs[1] self.assertNotified( print_msg, [{ 'is_read': True, 'partner': test_customer, 'type': 'email', }], ) # print: template-based message self.assertEqual(len(print_msg.attachment_ids), 3) self.assertNotIn(self.attachments, print_msg.attachment_ids, 'Attachments should be duplicated, not just linked') self.assertEqual(print_msg.author_id, self.user_account_other.partner_id, 'Should take invoice_user_id partner') self.assertEqual(print_msg.email_from, self.user_account_other.email_formatted, 'Should take invoice_user_id email') self.assertEqual(print_msg.notified_partner_ids, test_customer + self.user_accountman.partner_id) self.assertEqual(print_msg.subject, f'SpanishSubject for {test_move.name}') # tracking: is_move_sent self.assertEqual(track_msg.author_id, self.env.user.partner_id) self.assertEqual(track_msg.email_from, self.env.user.email_formatted) self.assertTrue('is_move_sent' in track_msg.tracking_value_ids.field_id.mapped('name')) # sent email self.assertMailMail( test_customer, 'sent', author=self.user_account_other.partner_id, # author: synchronized with email_from of template content=f'SpanishBody for {test_move.name}', # translated version email_values={ 'attachments_info': [ {'name': 'AttFileName_00.txt', 'raw': b'AttContent_00', 'type': 'text/plain'}, {'name': 'AttFileName_01.txt', 'raw': b'AttContent_01', 'type': 'text/plain'}, {'name': f'{test_move.name}.pdf', 'type': 'application/pdf'}, ], 'body_content': f'SpanishBody for {test_move.name}', # translated version 'email_from': self.user_account_other.email_formatted, 'subject': f'SpanishSubject for {test_move.name}', # translated version 'reply_to': formataddr(( f'{test_move.company_id.name} {test_move.display_name}', f'{self.alias_catchall}@{self.alias_domain}' )), }, fields_values={ 'auto_delete': True, 'email_from': self.user_account_other.email_formatted, 'is_notification': True, # should keep logs by default 'mail_server_id': self.mail_server_default, 'subject': f'SpanishSubject for {test_move.name}', # translated version 'reply_to': formataddr(( f'{test_move.company_id.name} {test_move.display_name}', f'{self.alias_catchall}@{self.alias_domain}' )), }, ) # composer configuration self.assertIn(f'SpanishBody for {test_move.name}', composer.mail_body, 'Should be translated, based on template') self.assertEqual(composer.move_ids, test_move) self.assertTrue(composer.checkbox_send_mail) self.assertFalse(composer.checkbox_download) self.assertEqual(composer.mail_subject, f'SpanishSubject for {test_move.name}', 'Should be translated, based on template') self.assertEqual(composer.mail_template_id, move_template) # invoice update self.assertTrue(test_move.is_move_sent) def test_invoice_sent_to_additional_partner(self): """ Make sure that when an invoice is sent to a partner who is not the invoiced customer, they receive a link containing an access token, allowing them to view the invoice without needing to log in. """ test_move = self.test_account_moves[1].with_env(self.env) move_template = self.move_template.with_env(self.env) # add a follower to the invoice self.partner_b.email = 'partner_b@example.com' test_move.message_subscribe(self.partner_b.ids) additional_partner = self.env['res.partner'].create({ 'name': "Additional Partner", 'email': "additional@example.com", }) composer = self.env['account.move.send']\ .with_context(active_model='account.move', active_ids=test_move.ids)\ .create({'mail_template_id': move_template.id, 'mail_partner_ids': additional_partner.ids}) with self.mock_mail_gateway(mail_unlink_sent=False): composer.action_send_and_print() self.assertMailMail( test_move.partner_id, 'sent', author=self.user_account_other.partner_id, content='access_token=' ) self.assertMailMail( additional_partner, 'sent', author=self.user_account_other.partner_id, content='access_token=' ) follower_mail = self.assertMailMail( self.partner_b, 'sent', author=self.user_account_other.partner_id, ) self.assertNotIn('access_token=', follower_mail.body_html, "The followers should not bet sent the access token by default") @tagged('post_install_l10n', 'post_install', '-at_install') class TestAccountMoveSendCommon(AccountTestInvoicingCommon): @classmethod def setUpClass(cls, chart_template_ref=None): super().setUpClass(chart_template_ref=chart_template_ref) cls.partner_a.email = "partner_a@tsointsoin" cls.partner_b.email = "partner_b@tsointsoin" def _assert_mail_attachments_widget(self, wizard, expected_values_list): self.assertEqual(len(wizard.mail_attachments_widget), len(expected_values_list)) for values, expected_values in zip(wizard.mail_attachments_widget, expected_values_list): try: int(values['id']) check_id_needed = True except ValueError: check_id_needed = False self.assertDictEqual( {k: v for k, v in values.items() if not check_id_needed and k != 'id'}, {k: v for k, v in expected_values.items() if not check_id_needed and k != 'id'}, ) def create_send_and_print(self, invoices, **kwargs): template = self.env.ref(invoices._get_mail_template()) return self.env['account.move.send']\ .with_context(active_model='account.move', active_ids=invoices.ids)\ .create({ 'mail_template_id': template.id, **kwargs, }) def _get_mail_message(self, move): return self.env['mail.message'].search([('model', '=', move._name), ('res_id', '=', move.id)], limit=1) @tagged('post_install_l10n', 'post_install', '-at_install') class TestAccountMoveSend(TestAccountMoveSendCommon): def test_invoice_single(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) self.assertRecordValues(wizard, [{ 'mode': 'invoice_single', 'enable_download': True, 'checkbox_download': True, 'enable_send_mail': True, 'display_mail_composer': True, 'send_mail_readonly': False, 'checkbox_send_mail': True, 'mail_lang': 'en_US', 'mail_partner_ids': wizard.move_ids.partner_id.ids, }]) self.assertFalse(wizard.send_mail_warning_message) self.assertTrue(wizard.mail_subject) self.assertTrue(wizard.mail_body) self._assert_mail_attachments_widget(wizard, [{ 'mimetype': 'application/pdf', 'name': 'INV_2019_00001.pdf', 'placeholder': True, }]) # Process. results = wizard.action_send_and_print() self.assertEqual(results['type'], 'ir.actions.act_url') self.assertFalse(invoice.send_and_print_values) # The PDF has been successfully generated. pdf_report = invoice.invoice_pdf_report_id self.assertTrue(pdf_report) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice._name), ('res_id', '=', invoice.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) # Send it again. The PDF must not be created again. wizard = self.create_send_and_print(invoice) results = wizard.action_send_and_print() self.assertEqual(results['type'], 'ir.actions.act_url') self.assertFalse(invoice.send_and_print_values) self.assertRecordValues(invoice, [{'invoice_pdf_report_id': pdf_report.id}]) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice._name), ('res_id', '=', invoice.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) # Only one PDF linked to the invoice. invoice_attachments = self.env['ir.attachment'].search([ ('name', '=', pdf_report.name), ('res_model', '=', invoice._name), ('res_id', '=', invoice.id), ('res_field', '=', False), ]) self.assertFalse(invoice_attachments) def test_invoice_single_readonly_and_checkbox(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) self.partner_a.email = None wizard = self.create_send_and_print(invoice) self.assertRecordValues(wizard, [{ 'send_mail_readonly': True, 'checkbox_send_mail': False, }]) self.partner_a.email = "turlututu@tsointsoin" wizard = self.create_send_and_print(invoice) self.assertRecordValues(wizard, [{ 'send_mail_readonly': False, 'checkbox_send_mail': True, }]) def test_invoice_multi(self): invoice1 = self.init_invoice("out_invoice", partner=self.partner_a, amounts=[1000], post=True) invoice2 = self.init_invoice("out_invoice", partner=self.partner_b, amounts=[1000], post=True) wizard = self.create_send_and_print(invoice1 + invoice2) wizard_values = { 'mode': 'invoice_multi', 'enable_download': True, 'checkbox_download': True, 'enable_send_mail': True, 'display_mail_composer': False, 'send_mail_readonly': False, 'checkbox_send_mail': True, 'mail_lang': 'en_US', 'mail_partner_ids': [], 'mail_subject': False, 'mail_body': False, 'mail_attachments_widget': False, } self.assertRecordValues(wizard, [wizard_values]) wizard.checkbox_download = False # We don't want to download the zip (since then we process synchronously) # Process. results = wizard.action_send_and_print() self.assertEqual(results['type'], 'ir.actions.client') self.assertEqual(results['params']['next']['type'], 'ir.actions.act_window_close') self.assertRecordValues(wizard, [{'mode': 'invoice_multi'}]) # Awaiting the CRON. self.assertFalse(invoice1.invoice_pdf_report_id) self.assertFalse(invoice2.invoice_pdf_report_id) self.assertTrue(invoice1.is_being_sent) self.assertTrue(invoice2.is_being_sent) # Run the CRON. wizard.action_send_and_print(force_synchronous=True) self.assertTrue(invoice1.invoice_pdf_report_id) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice1._name), ('res_id', '=', invoice1.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) self.assertTrue(invoice2.invoice_pdf_report_id) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice2._name), ('res_id', '=', invoice2.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) self.assertFalse(invoice1.is_being_sent) self.assertFalse(invoice2.is_being_sent) # Mix already sent invoice with a new one. invoice3 = self.init_invoice("out_invoice", partner=self.partner_b, amounts=[1000], post=True) wizard = self.create_send_and_print(invoice1 + invoice2 + invoice3) self.assertRecordValues(wizard, [wizard_values]) wizard.action_send_and_print(force_synchronous=True) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice1._name), ('res_id', '=', invoice1.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice2._name), ('res_id', '=', invoice2.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) self.assertTrue(invoice1.invoice_pdf_report_id) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice3._name), ('res_id', '=', invoice3.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) def test_invoice_multi_with_download(self): invoice1 = self.init_invoice("out_invoice", partner=self.partner_a, amounts=[1000], post=True) invoice2 = self.init_invoice("out_invoice", partner=self.partner_b, amounts=[1000], post=True) wizard = self.create_send_and_print(invoice1 + invoice2) wizard_values = { 'mode': 'invoice_multi', 'enable_download': True, 'checkbox_download': True, 'enable_send_mail': True, 'display_mail_composer': False, 'send_mail_readonly': False, 'checkbox_send_mail': True, 'mail_lang': 'en_US', 'mail_partner_ids': [], 'mail_subject': False, 'mail_body': False, 'mail_attachments_widget': False, } self.assertRecordValues(wizard, [wizard_values]) # Process. results = wizard.action_send_and_print() self.assertEqual(results['type'], 'ir.actions.act_url') self.assertEqual((invoice1 + invoice2).mapped('send_and_print_values'), [False, False]) self.assertTrue(invoice1.invoice_pdf_report_id) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice1._name), ('res_id', '=', invoice1.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) self.assertTrue(invoice2.invoice_pdf_report_id) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice2._name), ('res_id', '=', invoice2.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) self.assertFalse(invoice1.is_being_sent) self.assertFalse(invoice2.is_being_sent) # Mix already sent invoice with a new one. invoice3 = self.init_invoice("out_invoice", partner=self.partner_b, amounts=[1000], post=True) wizard = self.create_send_and_print(invoice1 + invoice2 + invoice3) self.assertRecordValues(wizard, [wizard_values]) wizard.action_send_and_print(force_synchronous=True) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice1._name), ('res_id', '=', invoice1.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice2._name), ('res_id', '=', invoice2.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) self.assertTrue(invoice1.invoice_pdf_report_id) invoice_attachments = self.env['ir.attachment'].search([ ('res_model', '=', invoice3._name), ('res_id', '=', invoice3.id), ('res_field', '=', 'invoice_pdf_report_file'), ]) self.assertEqual(len(invoice_attachments), 1) def test_invoice_multi_readonly_checkbox_warning_message(self): invoice1 = self.init_invoice("out_invoice", partner=self.partner_a, amounts=[1000], post=True) invoice2 = self.init_invoice("out_invoice", partner=self.partner_b, amounts=[1000], post=True) self.partner_a.email = None self.partner_b.email = None wizard = self.create_send_and_print(invoice1 + invoice2) self.assertTrue(wizard.send_mail_warning_message) self.assertRecordValues(wizard, [{ 'send_mail_readonly': True, 'checkbox_send_mail': False, }]) self.partner_a.email = "turlututu@tsointsoin" wizard = self.create_send_and_print(invoice1 + invoice2) self.assertTrue(wizard.send_mail_warning_message) self.assertRecordValues(wizard, [{ 'send_mail_readonly': False, 'checkbox_send_mail': True, }]) self.partner_b.email = "turlututu@tsointsoin" wizard = self.create_send_and_print(invoice1 + invoice2) self.assertFalse(wizard.send_mail_warning_message) self.assertRecordValues(wizard, [{ 'send_mail_readonly': False, 'checkbox_send_mail': True, }]) def test_invoice_mail_attachments_widget(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) # Add a new attachment on the mail_template. template = self.env.ref(invoice._get_mail_template()) extra_attachment = self.env['ir.attachment'].create({'name': "extra_attachment", 'raw': b'bar'}) template.attachment_ids = [Command.link(extra_attachment.id)] wizard = self.create_send_and_print(invoice) pdf_report_values = { 'mimetype': 'application/pdf', 'name': 'INV_2019_00001.pdf', 'placeholder': True, } extra_attachment_values = { 'mimetype': 'application/octet-stream', 'name': extra_attachment.name, 'placeholder': False, 'mail_template_id': template.id, } self._assert_mail_attachments_widget(wizard, [pdf_report_values, extra_attachment_values]) # Add a new attachment manually. manual_attachment = self.env['ir.attachment'].create({'name': "manual_attachment", 'raw': b'foo'}) manual_attachment_values = { 'id': manual_attachment.id, 'name': manual_attachment.name, 'mimetype': manual_attachment.mimetype, 'placeholder': False, 'manual': True, } wizard.mail_attachments_widget = wizard.mail_attachments_widget + [manual_attachment_values] # Add an attachment to a new mail_template and change it. new_mail_template = wizard.mail_template_id.copy() extra_attachment2 = self.env['ir.attachment'].create({'name': "extra_attachment2", 'raw': b'bar'}) new_mail_template.attachment_ids = [Command.set(extra_attachment2.ids)] extra_attachment2_values = { 'mimetype': 'application/octet-stream', 'name': extra_attachment2.name, 'placeholder': False, 'mail_template_id': new_mail_template.id, } wizard.mail_template_id = new_mail_template self._assert_mail_attachments_widget(wizard, [ pdf_report_values, extra_attachment2_values, manual_attachment_values, ]) # Send. wizard.action_send_and_print() message = self._get_mail_message(invoice) self.assertRecordValues(message.attachment_ids.sorted('name'), [ { 'name': invoice.invoice_pdf_report_id.name, 'datas': invoice.invoice_pdf_report_id.datas, }, { 'name': extra_attachment2.name, 'datas': extra_attachment2.datas, }, { 'name': manual_attachment.name, 'datas': manual_attachment.datas, }, ]) # Resend. wizard = self.create_send_and_print(invoice) pdf_report_values['id'] = invoice.invoice_pdf_report_id.id self._assert_mail_attachments_widget(wizard, [ pdf_report_values, extra_attachment_values, ]) # Switch the template. wizard.mail_template_id = new_mail_template self._assert_mail_attachments_widget(wizard, [ pdf_report_values, extra_attachment2_values, ]) # Send. wizard.action_send_and_print() message = self._get_mail_message(invoice) self.assertRecordValues(message.attachment_ids.sorted('name'), [ { 'name': invoice.invoice_pdf_report_id.name, 'datas': invoice.invoice_pdf_report_id.datas, }, { 'name': extra_attachment2.name, 'datas': extra_attachment2.datas, }, ]) # Manually remove the attachment and check the mail's attachments are not removed. invoice_pdf_report_name = invoice.invoice_pdf_report_id.name invoice_pdf_report_datas = invoice.invoice_pdf_report_id.datas invoice.invoice_pdf_report_id.unlink() self.assertRecordValues(message.attachment_ids.sorted('name'), [ { 'name': invoice_pdf_report_name, 'datas': invoice_pdf_report_datas, }, { 'name': extra_attachment2.name, 'datas': extra_attachment2.datas, }, ]) def test_invoice_web_service_after_pdf_rendering(self): """ Test the ir.attachment for the PDF is not generated when the web service is called after the PDF generation but performing a cr.commit even in case of error. """ invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) def call_web_service_after_invoice_pdf_render(record, invoices_data): for invoice_data in invoices_data.values(): invoice_data['error'] = "turlututu" with patch.object(type(wizard), '_call_web_service_after_invoice_pdf_render', call_web_service_after_invoice_pdf_render): try: wizard.action_send_and_print(allow_fallback_pdf=False) except UserError: # Prevent a rollback in case of UserError because we can't commit in this test. # Instead, ignore the raised error. pass # The PDF is not generated in case of error. self.assertFalse(invoice.invoice_pdf_report_id) def test_proforma_pdf(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) def _hook_invoice_document_before_pdf_report_render(self, invoice, invoice_data): invoice_data['error'] = 'test_proforma_pdf' # Process. with patch.object(type(wizard), '_hook_invoice_document_before_pdf_report_render', _hook_invoice_document_before_pdf_report_render): results = wizard.action_send_and_print(allow_fallback_pdf=True) self.assertEqual(results['type'], 'ir.actions.act_url') self.assertFalse(invoice.send_and_print_values) # The PDF is not generated but a proforma. self.assertFalse(invoice.invoice_pdf_report_id) def test_error_but_continue(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) def _hook_invoice_document_before_pdf_report_render(self, invoice, invoice_data): invoice_data['error'] = 'prout' invoice_data['error_but_continue'] = True # Process. with patch.object(type(wizard), '_hook_invoice_document_before_pdf_report_render', _hook_invoice_document_before_pdf_report_render): results = wizard.action_send_and_print(allow_fallback_pdf=True) self.assertEqual(results['type'], 'ir.actions.act_url') self.assertFalse(invoice.send_and_print_values) # The PDF is generated even in case of error, but invoice_pdf_report_id is not set self.assertFalse(invoice.invoice_pdf_report_id) self.assertTrue(invoice.message_main_attachment_id) def test_with_empty_mail_template_single(self): """ Test you can use the send & print wizard without any mail template if and only if you are in single mode. """ self.partner_a.email = "turlututu@tsointsoin" invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) custom_subject = "turlututu" wizard = self.create_send_and_print(invoice, mail_template_id=None, mail_subject=custom_subject) wizard.action_send_and_print(allow_fallback_pdf=True) message = self.env['mail.message'].search([('model', '=', invoice._name), ('res_id', '=', invoice.id)], limit=1) self.assertRecordValues(message, [{'subject': custom_subject}]) def test_with_empty_mail_template_multi(self): """ Test shouldn't be able to send email without mail template in multi mode. """ self.partner_a.email = "turlututu@tsointsoin" invoice_1 = self.init_invoice("out_invoice", amounts=[1000], post=True) invoice_2 = self.init_invoice("out_invoice", amounts=[1000], post=True) custom_subject = "turlututu" wizard = self.create_send_and_print((invoice_1 + invoice_2), mail_template_id=None, mail_subject=custom_subject) with self.assertRaises(UserError): wizard.action_send_and_print(allow_fallback_pdf=True) def test_with_draft_invoices(self): """ Use Send & Print wizard on draft invoice(s) should raise an error. """ invoice_posted = self.init_invoice("out_invoice", amounts=[1000], post=True) invoice_draft = self.init_invoice("out_invoice", amounts=[1000], post=False) with self.assertRaises(UserError): self.create_send_and_print(invoice_draft) with self.assertRaises(UserError): self.create_send_and_print(invoice_posted + invoice_draft) def test_link_pdf_webservice_fails_after(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) def _call_web_service_after_invoice_pdf_render(self, invoices_data): for move_data in invoices_data.values(): move_data['error'] = 'service_failed_after' # Process. with patch( 'odoo.addons.account.wizard.account_move_send.AccountMoveSend._call_web_service_after_invoice_pdf_render', _call_web_service_after_invoice_pdf_render, ): wizard.action_send_and_print(allow_fallback_pdf=True) # The PDF is generated and linked self.assertTrue(invoice.invoice_pdf_report_id) # Not a proforma self.assertFalse(self.env['ir.attachment'].search([ ('name', '=', invoice._get_invoice_proforma_pdf_report_filename()), ])) def test_link_pdf_webservice_fails_before(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) def _call_web_service_before_invoice_pdf_render(self, invoices_data): for move_data in invoices_data.values(): move_data['error'] = 'service_failed_before' # Process. with patch( 'odoo.addons.account.wizard.account_move_send.AccountMoveSend._call_web_service_before_invoice_pdf_render', _call_web_service_before_invoice_pdf_render, ): wizard.action_send_and_print(allow_fallback_pdf=True) # The PDF is not generated but a proforma. self.assertFalse(invoice.invoice_pdf_report_id) self.assertTrue(self.env['ir.attachment'].search([ ('name', '=', invoice._get_invoice_proforma_pdf_report_filename()), ])) def test_send_and_print_cron(self): """ Test the cron for generating """ invoice_1_1 = self.init_invoice("out_invoice", amounts=[1000], post=True, company=self.company_data['company']) invoice_1_2 = self.init_invoice("out_invoice", amounts=[1000], post=True, company=self.company_data['company']) wizard = self.create_send_and_print(invoice_1_1 + invoice_1_2) wizard.checkbox_download = False wizard.action_send_and_print() # saves value on moves to be sent asynchronously invoice_2_1 = self.init_invoice("out_invoice", amounts=[1000], post=True, company=self.company_data_2['company']) invoice_2_2 = self.init_invoice("out_invoice", amounts=[1000], post=True, company=self.company_data_2['company']) wizard_2 = self.create_send_and_print(invoice_2_1 + invoice_2_2) wizard_2.checkbox_download = False wizard_2.action_send_and_print() invoices = invoice_1_1 + invoice_1_2 + invoice_2_1 + invoice_2_2 self.assertFalse(invoices.invoice_pdf_report_id) self.assertEqual(invoices.mapped(lambda inv: bool(inv.send_and_print_values)), [True] * len(invoices)) self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() # force processing self.assertTrue(all(invoice.invoice_pdf_report_id for invoice in invoices)) self.assertTrue(all(not invoice.send_and_print_values for invoice in invoices)) def test_cron_notifications(self): invoices_success = ( self.init_invoice("out_invoice", amounts=[1000], post=True) + self.init_invoice("out_invoice", amounts=[1000], post=True) ) invoices_error = ( self.init_invoice("out_invoice", amounts=[1000], post=True) + self.init_invoice("out_invoice", amounts=[1000], post=True) ) sp_partner_1 = self.env.user.partner_id wizard_partner_1 = self.create_send_and_print(invoices_success) wizard_partner_1.checkbox_download = False wizard_partner_1.action_send_and_print() sp_partner_2 = self.env['res.partner'].create({'name': 'Partner 2', 'email': 'test@test.odoo.com'}) self.env.user.partner_id = sp_partner_2 wizard_partner_2 = self.create_send_and_print(invoices_error) wizard_partner_2.checkbox_download = False wizard_partner_2.action_send_and_print() def _hook_invoice_document_before_pdf_report_render(self, invoice, invoice_data): if invoice.id in invoices_error.ids: invoice_data['error'] = 'blblblbl' self.assertTrue(all(invoice.send_and_print_values for invoice in invoices_success + invoices_error)) self.assertEqual(invoices_success[0].send_and_print_values.get('sp_partner_id'), sp_partner_1.id) self.assertEqual(invoices_error[0].send_and_print_values.get('sp_partner_id'), sp_partner_2.id) with patch( 'odoo.addons.account.wizard.account_move_send.AccountMoveSend._hook_invoice_document_before_pdf_report_render', _hook_invoice_document_before_pdf_report_render, ): self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() # force processing bus_1 = self.env['bus.bus'].sudo().search( [('channel', 'like', f'"res.partner",{sp_partner_1.id}')], order='id desc', limit=1, ) self.assertEqual(json.loads(bus_1.message)['payload']['type'], 'success') self.assertEqual(json.loads(bus_1.message)['payload']['action_button']['res_ids'], invoices_success.ids) bus_2 = self.env['bus.bus'].sudo().search( [('channel', 'like', f'"res.partner",{sp_partner_2.id}')], order='id desc', limit=1, ) self.assertEqual(json.loads(bus_2.message)['payload']['type'], 'warning') self.assertEqual(json.loads(bus_2.message)['payload']['action_button']['res_ids'], invoices_error.ids)