# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. from odoo import fields, Command from odoo.tests.common import TransactionCase, HttpCase, tagged, Form import json import time import base64 from lxml import etree from unittest import SkipTest def instantiate_accountman(cls): cls.user = cls.env['res.users'].create({ 'name': 'Because I am accountman!', 'login': 'accountman', 'password': 'accountman', 'groups_id': [ Command.set(cls.env.user.groups_id.ids), Command.link(cls.env.ref('account.group_account_manager').id), Command.link(cls.env.ref('account.group_account_user').id), ], }) cls.user.partner_id.email = 'accountman@test.com' # Shadow the current environment/cursor with one having the report user. # This is mandatory to test access rights. cls.env = cls.env(user=cls.user) cls.cr = cls.env.cr class AccountTestInvoicingCommon(TransactionCase): @classmethod def safe_copy(cls, record): return record and record.copy() @classmethod def copy_account(cls, account, default=None): suffix_nb = 1 while True: new_code = '%s.%s' % (account.code, suffix_nb) if account.search_count([('company_id', '=', account.company_id.id), ('code', '=', new_code)]): suffix_nb += 1 else: return account.copy(default={**(default or {}), 'code': new_code}) @classmethod def setUpClass(cls, chart_template_ref=None): super().setUpClass() cls.env.ref('base.main_company').currency_id = cls.env.ref('base.USD') instantiate_accountman(cls) assert 'post_install' in cls.test_tags, 'This test requires a CoA to be installed, it should be tagged "post_install"' if chart_template_ref: template_vals = cls.env['account.chart.template']._get_chart_template_mapping()[chart_template_ref] template_module = cls.env.ref(f"base.module_{template_vals['module']}") if template_module.state != 'installed': raise SkipTest(f"Module required for the test is not installed ({template_module.name})") cls.company_data_2 = cls.setup_company_data('company_2_data', chart_template=chart_template_ref) cls.company_data = cls.setup_company_data('company_1_data', chart_template=chart_template_ref) cls.user.write({ 'company_ids': [Command.set((cls.company_data['company'] + cls.company_data_2['company']).ids)], 'company_id': cls.company_data['company'].id, }) cls.currency_data = cls.setup_multi_currency_data() # ==== Taxes ==== cls.tax_sale_a = cls.company_data['default_tax_sale'] cls.tax_sale_b = cls.safe_copy(cls.company_data['default_tax_sale']) cls.tax_purchase_a = cls.company_data['default_tax_purchase'] cls.tax_purchase_b = cls.safe_copy(cls.company_data['default_tax_purchase']) cls.tax_armageddon = cls.setup_armageddon_tax('complex_tax', cls.company_data) # ==== Products ==== cls.product_a = cls.env['product.product'].create({ 'name': 'product_a', 'uom_id': cls.env.ref('uom.product_uom_unit').id, 'lst_price': 1000.0, 'standard_price': 800.0, 'property_account_income_id': cls.company_data['default_account_revenue'].id, 'property_account_expense_id': cls.company_data['default_account_expense'].id, 'taxes_id': [Command.set(cls.tax_sale_a.ids)], 'supplier_taxes_id': [Command.set(cls.tax_purchase_a.ids)], }) cls.product_b = cls.env['product.product'].create({ 'name': 'product_b', 'uom_id': cls.env.ref('uom.product_uom_dozen').id, 'lst_price': 200.0, 'standard_price': 160.0, 'property_account_income_id': cls.copy_account(cls.company_data['default_account_revenue']).id, 'property_account_expense_id': cls.copy_account(cls.company_data['default_account_expense']).id, 'taxes_id': [Command.set((cls.tax_sale_a + cls.tax_sale_b).ids)], 'supplier_taxes_id': [Command.set((cls.tax_purchase_a + cls.tax_purchase_b).ids)], }) # ==== Fiscal positions ==== cls.fiscal_pos_a = cls.env['account.fiscal.position'].create({ 'name': 'fiscal_pos_a', 'tax_ids': ([(0, None, { 'tax_src_id': cls.tax_sale_a.id, 'tax_dest_id': cls.tax_sale_b.id, })] if cls.tax_sale_b else []) + ([(0, None, { 'tax_src_id': cls.tax_purchase_a.id, 'tax_dest_id': cls.tax_purchase_b.id, })] if cls.tax_purchase_b else []), 'account_ids': [ (0, None, { 'account_src_id': cls.product_a.property_account_income_id.id, 'account_dest_id': cls.product_b.property_account_income_id.id, }), (0, None, { 'account_src_id': cls.product_a.property_account_expense_id.id, 'account_dest_id': cls.product_b.property_account_expense_id.id, }), ], }) # ==== Payment terms ==== cls.pay_terms_a = cls.env.ref('account.account_payment_term_immediate') cls.pay_terms_b = cls.env['account.payment.term'].create({ 'name': '30% Advance End of Following Month', 'note': 'Payment terms: 30% Advance End of Following Month', 'line_ids': [ (0, 0, { 'value': 'percent', 'value_amount': 30.0, 'nb_days': 0, }), (0, 0, { 'value': 'percent', 'value_amount': 70.0, 'delay_type': 'days_after_end_of_next_month', 'nb_days': 0, }), ], }) # ==== Partners ==== cls.partner_a = cls.env['res.partner'].create({ 'name': 'partner_a', 'property_payment_term_id': cls.pay_terms_a.id, 'property_supplier_payment_term_id': cls.pay_terms_a.id, 'property_account_receivable_id': cls.company_data['default_account_receivable'].id, 'property_account_payable_id': cls.company_data['default_account_payable'].id, 'company_id': False, }) cls.partner_b = cls.env['res.partner'].create({ 'name': 'partner_b', 'property_payment_term_id': cls.pay_terms_b.id, 'property_supplier_payment_term_id': cls.pay_terms_b.id, 'property_account_position_id': cls.fiscal_pos_a.id, 'property_account_receivable_id': cls.company_data['default_account_receivable'].copy().id, 'property_account_payable_id': cls.company_data['default_account_payable'].copy().id, 'company_id': False, }) # ==== Cash rounding ==== cls.cash_rounding_a = cls.env['account.cash.rounding'].create({ 'name': 'add_invoice_line', 'rounding': 0.05, 'strategy': 'add_invoice_line', 'profit_account_id': cls.company_data['default_account_revenue'].copy().id, 'loss_account_id': cls.company_data['default_account_expense'].copy().id, 'rounding_method': 'UP', }) cls.cash_rounding_b = cls.env['account.cash.rounding'].create({ 'name': 'biggest_tax', 'rounding': 0.05, 'strategy': 'biggest_tax', 'rounding_method': 'DOWN', }) # ==== Payment methods ==== bank_journal = cls.company_data['default_journal_bank'] cls.inbound_payment_method_line = bank_journal.inbound_payment_method_line_ids[0] cls.outbound_payment_method_line = bank_journal.outbound_payment_method_line_ids[0] @classmethod def change_company_country(cls, company, country): company.country_id = country company.account_fiscal_country_id = country for model in ('account.tax', 'account.tax.group'): cls.env.add_to_compute( cls.env[model]._fields['country_id'], cls.env[model].search([('company_id', '=', company.id)]), ) @classmethod def setup_company_data(cls, company_name, chart_template=None, **kwargs): ''' Create a new company having the name passed as parameter. A chart of accounts will be installed to this company: the same as the current company one. The current user will get access to this company. :param chart_template: The chart template to be used on this new company. :param company_name: The name of the company. :return: A dictionary will be returned containing all relevant accounting data for testing. ''' company = cls.env['res.company'].create({ 'name': company_name, **kwargs, }) cls.env.user.company_ids |= company # Install the chart template chart_template = chart_template or cls.env['account.chart.template']._guess_chart_template(company.country_id) cls.env['account.chart.template'].try_loading(chart_template, company=company, install_demo=False) if not company.account_fiscal_country_id: company.account_fiscal_country_id = cls.env.ref('base.us') # The currency could be different after the installation of the chart template. if kwargs.get('currency_id'): company.write({'currency_id': kwargs['currency_id']}) return { 'company': company, 'currency': company.currency_id, 'default_account_revenue': cls.env['account.account'].search([ ('company_id', '=', company.id), ('account_type', '=', 'income'), ('id', '!=', company.account_journal_early_pay_discount_gain_account_id.id) ], limit=1), 'default_account_expense': cls.env['account.account'].search([ ('company_id', '=', company.id), ('account_type', '=', 'expense'), ('id', '!=', company.account_journal_early_pay_discount_loss_account_id.id) ], limit=1), 'default_account_receivable': cls.env['ir.property'].with_company(company)._get( 'property_account_receivable_id', 'res.partner' ), 'default_account_payable': cls.env['account.account'].search([ ('company_id', '=', company.id), ('account_type', '=', 'liability_payable') ], limit=1), 'default_account_assets': cls.env['account.account'].search([ ('company_id', '=', company.id), ('account_type', '=', 'asset_fixed') ], limit=1), 'default_account_deferred_expense': cls.env['account.account'].search([ ('company_id', '=', company.id), ('account_type', '=', 'asset_current') ], limit=1), 'default_account_deferred_revenue': cls.env['account.account'].search([ ('company_id', '=', company.id), ('account_type', '=', 'liability_current') ], limit=1), 'default_account_tax_sale': company.account_sale_tax_id.mapped('invoice_repartition_line_ids.account_id'), 'default_account_tax_purchase': company.account_purchase_tax_id.mapped('invoice_repartition_line_ids.account_id'), 'default_journal_misc': cls.env['account.journal'].search([ ('company_id', '=', company.id), ('type', '=', 'general') ], limit=1), 'default_journal_sale': cls.env['account.journal'].search([ ('company_id', '=', company.id), ('type', '=', 'sale') ], limit=1), 'default_journal_purchase': cls.env['account.journal'].search([ ('company_id', '=', company.id), ('type', '=', 'purchase') ], limit=1), 'default_journal_bank': cls.env['account.journal'].search([ ('company_id', '=', company.id), ('type', '=', 'bank') ], limit=1), 'default_journal_cash': cls.env['account.journal'].search([ ('company_id', '=', company.id), ('type', '=', 'cash') ], limit=1), 'default_tax_sale': company.account_sale_tax_id, 'default_tax_purchase': company.account_purchase_tax_id, } @classmethod def setup_multi_currency_data(cls, default_values=None, rate2016=3.0, rate2017=2.0): default_values = default_values or {} foreign_currency = cls.env['res.currency'].create({ 'name': 'Gold Coin', 'symbol': '☺', 'rounding': 0.001, 'position': 'after', 'currency_unit_label': 'Gold', 'currency_subunit_label': 'Silver', **default_values, }) rate1 = cls.env['res.currency.rate'].create({ 'name': '2016-01-01', 'rate': rate2016, 'currency_id': foreign_currency.id, 'company_id': cls.env.company.id, }) rate2 = cls.env['res.currency.rate'].create({ 'name': '2017-01-01', 'rate': rate2017, 'currency_id': foreign_currency.id, 'company_id': cls.env.company.id, }) return { 'currency': foreign_currency, 'rates': rate1 + rate2, } @classmethod def _instantiate_basic_test_tax_group(cls, company=None, country=None): company = company or cls.env.company vals = { 'name': 'Test tax group', 'company_id': company.id, 'tax_receivable_account_id': cls.company_data['default_account_receivable'].sudo().copy({'company_id': company.id}).id, 'tax_payable_account_id': cls.company_data['default_account_payable'].sudo().copy({'company_id': company.id}).id, } if country: vals['country_id'] = country.id return cls.env['account.tax.group'].sudo().create(vals) @classmethod def setup_armageddon_tax(cls, tax_name, company_data): return cls.env['account.tax'].create({ 'name': '%s (group)' % tax_name, 'amount_type': 'group', 'amount': 0.0, 'country_id': company_data['company'].account_fiscal_country_id.id, 'children_tax_ids': [ (0, 0, { 'name': '%s (child 1)' % tax_name, 'amount_type': 'percent', 'amount': 20.0, 'country_id': company_data['company'].account_fiscal_country_id.id, 'price_include': True, 'include_base_amount': True, 'tax_exigibility': 'on_invoice', 'invoice_repartition_line_ids': [ (0, 0, { 'repartition_type': 'base', }), (0, 0, { 'factor_percent': 40, 'repartition_type': 'tax', 'account_id': company_data['default_account_tax_sale'].id, }), (0, 0, { 'factor_percent': 60, 'repartition_type': 'tax', # /!\ No account set. }), ], 'refund_repartition_line_ids': [ (0, 0, { 'repartition_type': 'base', }), (0, 0, { 'factor_percent': 40, 'repartition_type': 'tax', 'account_id': company_data['default_account_tax_sale'].id, }), (0, 0, { 'factor_percent': 60, 'repartition_type': 'tax', # /!\ No account set. }), ], }), (0, 0, { 'name': '%s (child 2)' % tax_name, 'amount_type': 'percent', 'amount': 10.0, 'country_id': company_data['company'].account_fiscal_country_id.id, 'tax_exigibility': 'on_payment', 'cash_basis_transition_account_id': cls.safe_copy(company_data['default_account_tax_sale']).id, 'invoice_repartition_line_ids': [ (0, 0, { 'repartition_type': 'base', }), (0, 0, { 'repartition_type': 'tax', 'account_id': company_data['default_account_tax_sale'].id, }), ], 'refund_repartition_line_ids': [ (0, 0, { 'repartition_type': 'base', }), (0, 0, { 'repartition_type': 'tax', 'account_id': company_data['default_account_tax_sale'].id, }), ], }), ], }) @classmethod def init_invoice(cls, move_type, partner=None, invoice_date=None, post=False, products=None, amounts=None, taxes=None, company=False, currency=None): products = [] if products is None else products amounts = [] if amounts is None else amounts move_form = Form(cls.env['account.move'] \ .with_company(company or cls.env.company) \ .with_context(default_move_type=move_type)) move_form.invoice_date = invoice_date or fields.Date.from_string('2019-01-01') # According to the state or type of the invoice, the date field is sometimes visible or not # Besides, the date field can be put multiple times in the view # "invisible": "['|', ('state', '!=', 'draft'), ('auto_post', '!=', 'at_date')]" # "invisible": ['|', '|', ('state', '!=', 'draft'), ('auto_post', '=', 'no'), ('auto_post', '=', 'at_date')] # "invisible": "['&', ('move_type', 'in', ['out_invoice', 'out_refund', 'out_receipt']), ('quick_edit_mode', '=', False)]" # :TestAccountMoveOutInvoiceOnchanges, :TestAccountMoveOutRefundOnchanges, .test_00_debit_note_out_invoice, :TestAccountEdi if not move_form._get_modifier('date', 'invisible'): move_form.date = move_form.invoice_date move_form.partner_id = partner or cls.partner_a if currency: move_form.currency_id = currency for product in (products or []): with move_form.invoice_line_ids.new() as line_form: line_form.product_id = product if taxes is not None: line_form.tax_ids.clear() for tax in taxes: line_form.tax_ids.add(tax) for amount in (amounts or []): with move_form.invoice_line_ids.new() as line_form: line_form.name = "test line" line_form.price_unit = amount if taxes is not None: line_form.tax_ids.clear() for tax in taxes: line_form.tax_ids.add(tax) rslt = move_form.save() if post: rslt.action_post() return rslt def create_line_for_reconciliation(self, balance, amount_currency, currency, move_date, account_1=None, partner=None): write_off_account_to_be_reconciled = account_1 if account_1 else self.receivable_account move = self.env['account.move'].create({ 'move_type': 'entry', 'date': move_date, 'line_ids': [ Command.create({ 'debit': balance if balance > 0.0 else 0.0, 'credit': -balance if balance < 0.0 else 0.0, 'amount_currency': amount_currency, 'account_id': write_off_account_to_be_reconciled.id, 'currency_id': currency.id, 'partner_id': partner.id if partner else None, }), Command.create({ 'debit': -balance if balance < 0.0 else 0.0, 'credit': balance if balance > 0.0 else 0.0, 'amount_currency': -amount_currency, 'account_id': self.company_data['default_account_revenue'].id, 'currency_id': currency.id, 'partner_id': partner.id if partner else None, }), ], }) move.action_post() line = move.line_ids.filtered(lambda x: x.account_id == write_off_account_to_be_reconciled) self.assertRecordValues(line, [{ 'amount_residual': balance, 'amount_residual_currency': amount_currency, 'reconciled': False, }]) return line def assertInvoiceValues(self, move, expected_lines_values, expected_move_values): def sort_lines(lines): return lines.sorted(lambda line: (line.sequence, not bool(line.tax_line_id), line.name or '', line.balance)) self.assertRecordValues(sort_lines(move.line_ids.sorted()), expected_lines_values) self.assertRecordValues(move, [expected_move_values]) def assert_invoice_outstanding_to_reconcile_widget(self, invoice, expected_amounts): """ Check the outstanding widget before the reconciliation. :param invoice: An invoice. :param expected_amounts: A map -> """ invoice.invalidate_recordset(['invoice_outstanding_credits_debits_widget']) widget_vals = invoice.invoice_outstanding_credits_debits_widget if widget_vals: current_amounts = {vals['move_id']: vals['amount'] for vals in widget_vals['content']} else: current_amounts = {} self.assertDictEqual(current_amounts, expected_amounts) def assert_invoice_outstanding_reconciled_widget(self, invoice, expected_amounts): """ Check the outstanding widget after the reconciliation. :param invoice: An invoice. :param expected_amounts: A map -> """ invoice.invalidate_recordset(['invoice_payments_widget']) widget_vals = invoice.invoice_payments_widget if widget_vals: current_amounts = {vals['move_id']: vals['amount'] for vals in widget_vals['content']} else: current_amounts = {} self.assertDictEqual(current_amounts, expected_amounts) #################################################### # Xml Comparison #################################################### def _turn_node_as_dict_hierarchy(self, node, path=''): ''' Turn the node as a python dictionary to be compared later with another one. Allow to ignore the management of namespaces. :param node: A node inside an xml tree. :param path: The optional path of tags for recursive call. :return: A python dictionary. ''' tag_split = node.tag.split('}') tag_wo_ns = tag_split[-1] attrib_wo_ns = {k: v for k, v in node.attrib.items() if '}' not in k} full_path = f'{path}/{tag_wo_ns}' return { 'tag': tag_wo_ns, 'full_path': full_path, 'namespace': None if len(tag_split) < 2 else tag_split[0], 'text': (node.text or '').strip(), 'attrib': attrib_wo_ns, 'children': [ self._turn_node_as_dict_hierarchy(child_node, path=path) for child_node in node.getchildren() ], } def assertXmlTreeEqual(self, xml_tree, expected_xml_tree): ''' Compare two lxml.etree. :param xml_tree: The current tree. :param expected_xml_tree: The expected tree. ''' def assertNodeDictEqual(node_dict, expected_node_dict): ''' Compare nodes created by the `_turn_node_as_dict_hierarchy` method. :param node_dict: The node to compare with. :param expected_node_dict: The expected node. ''' # Check tag. self.assertEqual(node_dict['tag'], expected_node_dict['tag']) # Check attributes. node_dict_attrib = {k: '___ignore___' if expected_node_dict['attrib'].get(k) == '___ignore___' else v for k, v in node_dict['attrib'].items()} expected_node_dict_attrib = {k: v for k, v in expected_node_dict['attrib'].items() if v != '___remove___'} self.assertDictEqual( node_dict_attrib, expected_node_dict_attrib, f"Element attributes are different for node {node_dict['full_path']}", ) # Check text. if expected_node_dict['text'] != '___ignore___': self.assertEqual( node_dict['text'], expected_node_dict['text'], f"Element text are different for node {node_dict['full_path']}", ) # Check children. self.assertEqual( [child['tag'] for child in node_dict['children']], [child['tag'] for child in expected_node_dict['children']], f"Number of children elements for node {node_dict['full_path']} is different.", ) for child_node_dict, expected_child_node_dict in zip(node_dict['children'], expected_node_dict['children']): assertNodeDictEqual(child_node_dict, expected_child_node_dict) assertNodeDictEqual( self._turn_node_as_dict_hierarchy(xml_tree), self._turn_node_as_dict_hierarchy(expected_xml_tree), ) def with_applied_xpath(self, xml_tree, xpath): ''' Applies the xpath to the xml_tree passed as parameter. :param xml_tree: An instance of etree. :param xpath: The xpath to apply as a string. :return: The resulting etree after applying the xpaths. ''' diff_xml_tree = etree.fromstring('%s' % xpath) return self.env['ir.ui.view'].apply_inheritance_specs(xml_tree, diff_xml_tree) def get_xml_tree_from_attachment(self, attachment): ''' Extract an instance of etree from an ir.attachment. :param attachment: An ir.attachment. :return: An instance of etree. ''' return etree.fromstring(base64.b64decode(attachment.with_context(bin_size=False).datas)) def get_xml_tree_from_string(self, xml_tree_str): ''' Convert the string passed as parameter to an instance of etree. :param xml_tree_str: A string representing an xml. :return: An instance of etree. ''' return etree.fromstring(xml_tree_str) class AccountTestInvoicingHttpCommon(AccountTestInvoicingCommon, HttpCase): pass