# -*- 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 import users, warmup, 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.partner_a.email = 'turlututu@tsointsoin' cls.user_accountman = cls.env.user # main account setup shadows users, better save it cls.company_main = cls.company_data['company'] 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': [ Command.set(cls.company_data['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', 'invoice_edi_format': False, '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, ) @classmethod def default_env_context(cls): # OVERRIDE return {} 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) with self.mock_mail_gateway(mail_unlink_sent=False): self.env['account.move.send']._generate_and_send_invoices( test_moves, sending_methods=['email'], mail_template=move_template, ) # 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}' )), }, ) # 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.wizard'].with_context(active_model='account.move', active_ids=test_move.ids).create({ 'sending_methods': ['email'], '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_id, test_move) self.assertTrue('email' in composer.sending_methods) 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.wizard'].with_context(active_model='account.move', active_ids=test_move.ids).create({ 'sending_methods': ['email'], '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_id, test_move) self.assertTrue('email' in composer.sending_methods) 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) @users('user_account') @warmup @mute_logger('odoo.addons.mail.models.mail_mail', 'odoo.models.unlink') def test_move_composer_with_dynamic_reports(self): """ It makes sure that when an invoice is sent using a template that has additional dynamic reports, those extra reports are also generated and sent by mail along side the invoice PDF and the other attachments that were manually added. """ 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) extra_dynamic_report = self.env.ref('account.action_account_original_vendor_bill') move_template.report_template_ids += extra_dynamic_report composer = self.env['account.move.send.wizard'].with_context(active_model='account.move', active_ids=test_move.ids).create({ 'sending_methods': ['email'], '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.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'}, {'name': f'{extra_dynamic_report.name.lower()}_{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}' )), }, ) 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) additional_partner = self.env['res.partner'].create({ 'name': "Additional Partner", 'email': "additional@example.com", }) composer = self.env['account.move.send.wizard'].with_context(active_model='account.move', active_ids=test_move.ids).create({ 'sending_methods': ['email'], '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=', ) class TestAccountMoveSendCommon(AccountTestInvoicingCommon): @classmethod def setUpClass(cls): super().setUpClass() 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): wizard_model = 'account.move.send.wizard' if len(invoices) == 1 else 'account.move.send.batch.wizard' return self.env[wizard_model]\ .with_context(active_model='account.move', active_ids=invoices.ids)\ .create(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): @classmethod def setUpClass(cls): super().setUpClass() cls.company_data_2 = cls.setup_other_company() (cls.partner_a + cls.partner_b).write({ 'invoice_sending_method': 'email', 'email': "turlututu@tsointsoin", }) def test_invoice_single(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice, sending_methods=['email', 'manual']) self.assertRecordValues(wizard, [{ 'move_id': invoice.id, 'sending_methods': ['email', 'manual'], 'extra_edis': False, 'extra_edi_checkboxes': False, 'pdf_report_id': wizard._get_default_pdf_report_id(invoice).id, 'display_pdf_report_id': False, 'mail_template_id': wizard._get_default_mail_template_id(invoice).id, 'mail_lang': 'en_US', 'mail_partner_ids': wizard.move_id.partner_id.ids, }]) self.assertFalse(wizard.alerts) self.assertTrue(wizard.mail_subject) self.assertTrue(wizard.mail_body) self._assert_mail_attachments_widget(wizard, [{ 'mimetype': 'application/pdf', 'name': invoice._get_invoice_report_filename(), 'placeholder': True, }]) # Process. results = wizard.action_send_and_print() self.assertEqual(results['type'], 'ir.actions.act_url') self.assertFalse(invoice.sending_data) # 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) self.assertTrue(self._get_mail_message(invoice)) # Send it again. The PDF must not be created again. wizard = self.create_send_and_print(invoice, sending_methods=['email', 'manual']) with patch('odoo.addons.account.models.account_move_send.AccountMoveSend._hook_invoice_document_after_pdf_report_render') as mocked_method: results = wizard.action_send_and_print() mocked_method.assert_not_called() self.assertEqual(results['type'], 'ir.actions.act_url') self.assertFalse(invoice.sending_data) 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) self.assertTrue(self._get_mail_message(invoice)) # 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_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) self.partner_a.invoice_sending_method = 'email' self.partner_b.invoice_sending_method = 'manual' wizard = self.create_send_and_print(invoice1 + invoice2) self.assertEqual(wizard.move_ids, invoice1 + invoice2) self.assertFalse(wizard.alerts) self.assertEqual(wizard.summary_data, { 'manual': {'count': 1, 'label': 'Manually'}, 'email': {'count': 1, 'label': 'by Email'}, }) # 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') # 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. self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() 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.assertTrue(self._get_mail_message(invoice1)) 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.assertEqual(wizard.move_ids, invoice1 + invoice2 + invoice3) self.assertEqual(wizard.summary_data, { 'manual': {'count': 2, 'label': 'Manually'}, 'email': {'count': 1, 'label': 'by Email'}, }) wizard.action_send_and_print() self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() 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_email_missing(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.assertTrue(bool(self.partner_b.email)) wizard = self.create_send_and_print(invoice1 + invoice2) self.assertEqual(wizard.summary_data, { 'email': {'count': 1, 'label': 'by Email'}, # Only one will be actually sent by email }) self.assertTrue('account_missing_email' in wizard.alerts) self.assertEqual(wizard.alerts['account_missing_email']['level'], 'warning') wizard.action_send_and_print() self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() # invoices are generated, but only partner_b got an email, without raising any errors self.assertTrue(invoice1.invoice_pdf_report_id) self.assertFalse(self._get_mail_message(invoice1)) self.assertTrue(invoice2.invoice_pdf_report_id) self.assertTrue(self._get_mail_message(invoice2)) def test_invoice_multi_with_edi(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.invoice_sending_method = 'email' self.partner_b.invoice_sending_method = 'manual' def get_default_extra_edis(self, move): if move == invoice1: return {'edi1'} return {'edi1', 'edi2'} def get_all_extra_edis(self): return { 'edi1': {'label': 'EDI 1'}, 'edi2': {'label': 'EDI 2'}, } with ( patch('odoo.addons.account.models.account_move_send.AccountMoveSend._get_default_extra_edis', get_default_extra_edis), patch('odoo.addons.account.models.account_move_send.AccountMoveSend._get_all_extra_edis', get_all_extra_edis) ): wizard = self.create_send_and_print(invoice1 + invoice2) self.assertEqual(wizard.summary_data, { 'edi1': {'count': 2, 'label': 'by EDI 1'}, 'edi2': {'count': 1, 'label': 'by EDI 2'}, 'manual': {'count': 1, 'label': 'Manually'}, 'email': {'count': 1, 'label': 'by Email'}, }) 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 = 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, sending_methods=['email'], mail_template_id=template.id ) 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( 'odoo.addons.account.models.account_move_send.AccountMoveSend._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( 'odoo.addons.account.models.account_move_send.AccountMoveSend._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_window_close') self.assertFalse(invoice.sending_data) # The PDF is not generated but a proforma. self.assertFalse(invoice.invoice_pdf_report_id) self.assertEqual(invoice.message_main_attachment_id.name, 'INV_2019_00001_proforma.pdf') 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( 'odoo.addons.account.models.account_move_send.AccountMoveSend._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_window_close') self.assertFalse(invoice.sending_data) # 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.assertEqual(invoice.message_main_attachment_id.name, 'INV_2019_00001_proforma.pdf') 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. """ invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) custom_subject = "turlututu" wizard = self.create_send_and_print(invoice) wizard.mail_template_id = None wizard.mail_subject = custom_subject wizard.action_send_and_print(allow_fallback_pdf=True) message = self._get_mail_message(invoice) 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. """ invoice_1 = self.init_invoice("out_invoice", amounts=[1000], partner=self.partner_a, post=True) invoice_2 = self.init_invoice("out_invoice", amounts=[1000], partner=self.partner_a, post=True) self.assertRecordValues(self.partner_a, [{ 'email': 'turlututu@tsointsoin', 'invoice_sending_method': 'email', }]) wizard = self.create_send_and_print(invoice_1 + invoice_2) wizard.action_send_and_print() self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() # generation defaulted on the generic mail_template and processed successfully self.assertTrue(invoice_1.invoice_pdf_report_id) self.assertTrue(invoice_2.invoice_pdf_report_id) 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.models.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.models.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.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.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.sending_data)), [True] * len(invoices)) self.env.ref('account.ir_cron_account_move_send').method_direct_trigger() self.assertTrue(all(invoice.invoice_pdf_report_id for invoice in invoices)) self.assertTrue(all(not invoice.sending_data 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.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.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.sending_data for invoice in invoices_success + invoices_error)) self.assertTrue(all(invoice.sending_data.get('author_partner_id') == sp_partner_1.id for invoice in invoices_success)) self.assertTrue(all(invoice.sending_data.get('author_partner_id') == sp_partner_2.id for invoice in invoices_error)) with patch( 'odoo.addons.account.models.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() self.env.cr.precommit.run() # trigger the creation of bus.bus records bus_1 = self.env['bus.bus'].sudo().search( [('channel', 'like', f'"res.partner",{sp_partner_1.id}')], order='id desc', limit=1, ) payload_1 = json.loads(bus_1.message)['payload'] self.assertEqual(payload_1['type'], 'success') self.assertEqual(sorted(payload_1['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, ) payload_2 = json.loads(bus_2.message)['payload'] self.assertEqual(payload_2['type'], 'warning') self.assertEqual(sorted(payload_2['action_button']['res_ids']), invoices_error.ids) def test_is_move_sent_state(self): # Post a move, nothing sent yet invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) self.assertFalse(invoice.is_move_sent) # Send via send & print wizard = self.create_send_and_print(invoice) wizard.action_send_and_print() self.assertTrue(invoice.is_move_sent) # Revert move to draft invoice.button_draft() self.assertTrue(invoice.is_move_sent) # Unlink PDF pdf_report = invoice.invoice_pdf_report_id self.assertTrue(pdf_report) invoice.invoice_pdf_report_id.unlink() self.assertTrue(invoice.is_move_sent) def test_no_sending_method_selected(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) self.assertFalse(invoice.invoice_pdf_report_id) wizard = self.create_send_and_print(invoice, sending_methods=[]) self.assertFalse(wizard.sending_methods) wizard.action_send_and_print() self.assertTrue(invoice.is_move_sent) self.assertTrue(invoice.invoice_pdf_report_id) def test_get_sending_settings(self): invoice = self.init_invoice("out_invoice", amounts=[1000], post=True) wizard = self.create_send_and_print(invoice) expected_results = { 'sending_methods': ['email'], 'invoice_edi_format': False, 'extra_edis': [], 'pdf_report': self.env.ref('account.account_invoices'), 'author_user_id': self.env.user.id, 'author_partner_id': self.env.user.partner_id.id, 'mail_template': self.env.ref('account.email_template_edi_invoice'), 'mail_lang': 'en_US', 'mail_body': wizard.mail_body, 'mail_subject': 'company_1_data Invoice (Ref INV/2019/00001)', 'mail_partner_ids': invoice.partner_id.ids, 'mail_attachments_widget': [{'id': 'placeholder_INV_2019_00001.pdf', 'name': 'INV_2019_00001.pdf', 'mimetype': 'application/pdf', 'placeholder': True}], } results = wizard._get_sending_settings() self.assertDictEqual(results, expected_results)