Add unittest for SEPA direct debit

account_banking_sepa_credit_transfer/tests/test_sct.py

@@ -26,10 +26,10 @@ class TestSCT(AccountingTestCase):
         company = self.env.ref('base.main_company')
         self.partner_agrolait = self.env.ref('base.res_partner_2')
         self.partner_c2c = self.env.ref('base.res_partner_12')
-        self.account_revenue = self.account_model.search([(
+        self.account_expense = self.account_model.search([(
             'user_type_id',
             '=',
-            self.env.ref('account.data_account_type_revenue').id)], limit=1)
+            self.env.ref('account.data_account_type_expenses').id)], limit=1)
         self.account_payable = self.account_model.search([(
             'user_type_id',
             '=',
@@ -54,12 +54,22 @@ class TestSCT(AccountingTestCase):
             })
         eur_currency_id = self.env.ref('base.EUR').id
         company.currency_id = eur_currency_id
-        invoice1 = self.create_invoice(self.partner_agrolait.id, 42.0, 'F1341')
-        invoice2 = self.create_invoice(self.partner_agrolait.id, 12.0, 'F1342')
+        invoice1 = self.create_invoice(
+            self.partner_agrolait.id,
+            'account_payment_mode.res_partner_2_iban', 42.0, 'F1341')
+        invoice2 = self.create_invoice(
+            self.partner_agrolait.id,
+            'account_payment_mode.res_partner_2_iban', 12.0, 'F1342')
         invoice3 = self.create_invoice(
-            self.partner_agrolait.id, 5.0, 'A1301', 'in_refund')
-        invoice4 = self.create_invoice(self.partner_c2c.id, 11.0, 'I1642')
-        invoice5 = self.create_invoice(self.partner_c2c.id, 41.0, 'I1643')
+            self.partner_agrolait.id,
+            'account_payment_mode.res_partner_2_iban', 5.0, 'A1301',
+            'in_refund')
+        invoice4 = self.create_invoice(
+            self.partner_c2c.id,
+            'account_payment_mode.res_partner_12_iban', 11.0, 'I1642')
+        invoice5 = self.create_invoice(
+            self.partner_c2c.id,
+            'account_payment_mode.res_partner_12_iban', 41.0, 'I1643')
         for inv in [invoice1, invoice2, invoice3, invoice4, invoice5]:
             action = inv.create_account_payment_line()
         self.assertEquals(action['res_model'], 'account.payment.order')
@@ -71,7 +81,8 @@ class TestSCT(AccountingTestCase):
         self.assertEquals(
             self.payment_order.journal_id, self.bank_journal)
         pay_lines = self.payment_line_model.search([
-            ('partner_id', '=', self.partner_agrolait.id)])
+            ('partner_id', '=', self.partner_agrolait.id),
+            ('order_id', '=', self.payment_order.id)])
         self.assertEquals(len(pay_lines), 3)
         agrolait_pay_line1 = pay_lines[0]
         accpre = self.env['decimal.precision'].precision_get('Account')
@@ -129,7 +140,8 @@ class TestSCT(AccountingTestCase):
         return
 
     def create_invoice(
-            self, partner_id, price_unit, reference, type='in_invoice'):
+            self, partner_id, partner_bank_xmlid, price_unit, reference,
+            type='in_invoice'):
         invoice = self.invoice_model.create({
             'partner_id': partner_id,
             'reference_type': 'none',
@@ -140,15 +152,14 @@ class TestSCT(AccountingTestCase):
             'type': type,
             'date_invoice': time.strftime('%Y-%m-%d'),
             'payment_mode_id': self.payment_mode.id,
-            'partner_bank_id':
-            self.env.ref('account_payment_mode.res_partner_2_iban').id,
+            'partner_bank_id': self.env.ref(partner_bank_xmlid).id,
             })
         self.invoice_line_model.create({
             'invoice_id': invoice.id,
             'price_unit': price_unit,
             'quantity': 1,
             'name': 'Great service',
-            'account_id': self.account_revenue.id,
+            'account_id': self.account_expense.id,
             })
         invoice.signal_workflow('invoice_open')
         return invoice

account_banking_sepa_direct_debit/tests/__init__.py

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+
+from . import test_sdd

account_banking_sepa_direct_debit/tests/test_sdd.py

@@ -0,0 +1,167 @@
+# -*- coding: utf-8 -*-
+# © 2016 Akretion (Alexis de Lattre <alexis.delattre@akretion.com>)
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from openerp.addons.account.tests.account_test_classes\
+    import AccountingTestCase
+from openerp.tools import float_compare
+import time
+from lxml import etree
+
+
+class TestSDD(AccountingTestCase):
+
+    def test_sdd(self):
+        self.company = self.env['res.company']
+        self.account_model = self.env['account.account']
+        self.move_model = self.env['account.move']
+        self.journal_model = self.env['account.journal']
+        self.payment_order_model = self.env['account.payment.order']
+        self.payment_line_model = self.env['account.payment.line']
+        self.mandate_model = self.env['account.banking.mandate']
+        self.bank_line_model = self.env['bank.payment.line']
+        self.partner_bank_model = self.env['res.partner.bank']
+        self.attachment_model = self.env['ir.attachment']
+        self.invoice_model = self.env['account.invoice']
+        self.invoice_line_model = self.env['account.invoice.line']
+        company = self.env.ref('base.main_company')
+        self.partner_agrolait = self.env.ref('base.res_partner_2')
+        self.partner_c2c = self.env.ref('base.res_partner_12')
+        self.account_revenue = self.account_model.search([(
+            'user_type_id',
+            '=',
+            self.env.ref('account.data_account_type_revenue').id)], limit=1)
+        self.account_receivable = self.account_model.search([(
+            'user_type_id',
+            '=',
+            self.env.ref('account.data_account_type_receivable').id)], limit=1)
+        # create journal
+        self.bank_journal = self.journal_model.create({
+            'name': 'Company Bank journal',
+            'type': 'bank',
+            'code': 'BNKFC',
+            'bank_account_id':
+            self.env.ref('account_payment_mode.main_company_iban').id,
+            'bank_id':
+            self.env.ref('account_payment_mode.bank_la_banque_postale').id,
+            })
+        # update payment mode
+        self.payment_mode = self.env.ref(
+            'account_banking_sepa_direct_debit.'
+            'payment_mode_inbound_sepa_dd1')
+        self.payment_mode.write({
+            'bank_account_link': 'fixed',
+            'fixed_journal_id': self.bank_journal.id,
+            })
+        eur_currency_id = self.env.ref('base.EUR').id
+        company.currency_id = eur_currency_id
+        self.env.ref(
+            'account_banking_sepa_direct_debit.res_partner_2_mandate').\
+            recurrent_sequence_type = 'first'
+        invoice1 = self.create_invoice(
+            self.partner_agrolait.id,
+            'account_banking_sepa_direct_debit.res_partner_2_mandate', 42.0)
+        invoice2 = self.create_invoice(
+            self.partner_c2c.id,
+            'account_banking_sepa_direct_debit.res_partner_12_mandate', 11.0)
+        for inv in [invoice1, invoice2]:
+            action = inv.create_account_payment_line()
+        self.assertEquals(action['res_model'], 'account.payment.order')
+        self.payment_order = self.payment_order_model.browse(action['res_id'])
+        self.assertEquals(
+            self.payment_order.payment_type, 'inbound')
+        self.assertEquals(
+            self.payment_order.payment_mode_id, self.payment_mode)
+        self.assertEquals(
+            self.payment_order.journal_id, self.bank_journal)
+        # Check payment line
+        pay_lines = self.payment_line_model.search([
+            ('partner_id', '=', self.partner_agrolait.id),
+            ('order_id', '=', self.payment_order.id)])
+        self.assertEquals(len(pay_lines), 1)
+        agrolait_pay_line1 = pay_lines[0]
+        accpre = self.env['decimal.precision'].precision_get('Account')
+        self.assertEquals(agrolait_pay_line1.currency_id.id, eur_currency_id)
+        self.assertEquals(
+            agrolait_pay_line1.mandate_id, invoice1.mandate_id)
+        self.assertEquals(
+            agrolait_pay_line1.partner_bank_id,
+            invoice1.mandate_id.partner_bank_id)
+        self.assertEquals(float_compare(
+            agrolait_pay_line1.amount_currency, 42, precision_digits=accpre),
+            0)
+        self.assertEquals(agrolait_pay_line1.communication_type, 'normal')
+        self.assertEquals(agrolait_pay_line1.communication, invoice1.number)
+        self.payment_order.draft2open()
+        self.assertEquals(self.payment_order.state, 'open')
+        self.assertEquals(self.payment_order.sepa, True)
+        # Check bank payment line
+        bank_lines = self.bank_line_model.search([
+            ('partner_id', '=', self.partner_agrolait.id)])
+        self.assertEquals(len(bank_lines), 1)
+        agrolait_bank_line = bank_lines[0]
+        self.assertEquals(agrolait_bank_line.currency_id.id, eur_currency_id)
+        self.assertEquals(float_compare(
+            agrolait_bank_line.amount_currency, 42.0, precision_digits=accpre),
+            0)
+        self.assertEquals(agrolait_bank_line.communication_type, 'normal')
+        self.assertEquals(
+            agrolait_bank_line.communication, invoice1.number)
+        self.assertEquals(
+            agrolait_bank_line.mandate_id, invoice1.mandate_id)
+        self.assertEquals(
+            agrolait_bank_line.partner_bank_id,
+            invoice1.mandate_id.partner_bank_id)
+        action = self.payment_order.open2generated()
+        self.assertEquals(self.payment_order.state, 'generated')
+        self.assertEquals(action['res_model'], 'ir.attachment')
+        attachment = self.attachment_model.browse(action['res_id'])
+        self.assertEquals(attachment.datas_fname[-4:], '.xml')
+        xml_file = attachment.datas.decode('base64')
+        xml_root = etree.fromstring(xml_file)
+        # print "xml_file=", etree.tostring(xml_root, pretty_print=True)
+        namespaces = xml_root.nsmap
+        namespaces['p'] = xml_root.nsmap[None]
+        namespaces.pop(None)
+        pay_method_xpath = xml_root.xpath(
+            '//p:PmtInf/p:PmtMtd', namespaces=namespaces)
+        self.assertEquals(pay_method_xpath[0].text, 'DD')
+        sepa_xpath = xml_root.xpath(
+            '//p:PmtInf/p:PmtTpInf/p:SvcLvl/p:Cd', namespaces=namespaces)
+        self.assertEquals(sepa_xpath[0].text, 'SEPA')
+        debtor_acc_xpath = xml_root.xpath(
+            '//p:PmtInf/p:CdtrAcct/p:Id/p:IBAN', namespaces=namespaces)
+        self.assertEquals(
+            debtor_acc_xpath[0].text,
+            self.payment_order.company_partner_bank_id.sanitized_acc_number)
+        self.payment_order.generated2uploaded()
+        self.assertEquals(self.payment_order.state, 'uploaded')
+        for inv in [invoice1, invoice2]:
+            self.assertEquals(inv.state, 'paid')
+        self.assertEquals(self.env.ref(
+            'account_banking_sepa_direct_debit.res_partner_2_mandate').
+            recurrent_sequence_type, 'recurring')
+        return
+
+    def create_invoice(
+            self, partner_id, mandate_xmlid, price_unit, type='out_invoice'):
+        invoice = self.invoice_model.create({
+            'partner_id': partner_id,
+            'reference_type': 'none',
+            'currency_id': self.env.ref('base.EUR').id,
+            'name': 'test',
+            'account_id': self.account_receivable.id,
+            'type': type,
+            'date_invoice': time.strftime('%Y-%m-%d'),
+            'payment_mode_id': self.payment_mode.id,
+            'mandate_id': self.env.ref(mandate_xmlid).id,
+            })
+        self.invoice_line_model.create({
+            'invoice_id': invoice.id,
+            'price_unit': price_unit,
+            'quantity': 1,
+            'name': 'Great service',
+            'account_id': self.account_revenue.id,
+            })
+        invoice.signal_workflow('invoice_open')
+        return invoice