fix flake8

Author: alessandrocamilli

l10n_it_withholding_tax/tests/__init__.py

@@ -0,0 +1 @@
+from . import test_withholding_tax

l10n_it_withholding_tax/tests/test_withholding_tax.py

@@ -0,0 +1,136 @@
+
+from odoo.tests.common import TransactionCase
+import time
+
+
+class TestWithholdingTax(TransactionCase):
+
+    def setUp(self):
+        super(TestWithholdingTax, self).setUp()
+
+        # Accounts
+        type_payable = self.env.ref('account.data_account_type_payable')
+        type_receivable = self.env.ref('account.data_account_type_receivable')
+        self.wt_account_payable = self.env['account.account'].create({
+            'name': 'Debiti per ritenute da versare',
+            'code': 'WT_001',
+            'user_type_id': type_payable.id,
+            'reconcile': True,
+        })
+        self.wt_account_receivable = self.env['account.account'].create({
+            'name': 'Crediti per ritenute subite',
+            'code': 'WT_002',
+            'user_type_id': type_receivable.id,
+            'reconcile': True,
+        })
+
+        # Journals
+        self.journal_misc = self.env['account.journal'].search(
+            [('type', '=', 'general')])[0]
+        self.journal_bank = self.env['account.journal'].create(
+            {'name': 'Bank', 'type': 'bank', 'code': 'BNK67'})
+
+        # Payments
+        vals_payment = {
+            'name': "",
+            'line_ids': [(0, 0, {'value': 'balance', 'days': 15})]
+            }
+        self.payment_term_15 = self.env['account.payment.term'].create(
+            vals_payment)
+
+        # Withholding tax
+        wt_vals = {
+            'name': 'Code 1040',
+            'code': '1040',
+            'certification': True,
+            'account_receivable_id': self.wt_account_receivable.id,
+            'account_payable_id': self.wt_account_payable.id,
+            'journal_id': self.journal_misc.id,
+            'payment_term': self.payment_term_15.id,
+            'rate_ids': [(0, 0, {
+                    'tax': 20,
+                    'base': 1,
+                })]
+            }
+        self.wt1040 = self.env['withholding.tax'].create(wt_vals)
+
+        # Supplier Invoice with WT
+        invoice_vals = [
+            (0, 0, {
+                'quantity': 1.0,
+                'account_id': self.env['account.account'].search(
+                    [('user_type_id', '=', self.env.ref(
+                        'account.data_account_type_expenses').id)],
+                    limit=1).id,
+                'name': 'Advice',
+                'price_unit': 1000.00,
+                'invoice_line_tax_wt_ids': [(6, 0, [self.wt1040.id])]
+                })]
+        self.invoice = self.env['account.invoice'].create({
+            'name': "Test Supplier Invoice WT",
+            'journal_id': self.env['account.journal'].search(
+                [('type', '=', 'purchase')])[0].id,
+            'partner_id': self.env.ref('base.res_partner_12').id,
+            'account_id': self.env['account.account'].search(
+                [('user_type_id', '=', self.env.ref(
+                    'account.data_account_type_receivable').id)],
+                limit=1, order='id').id,
+            'invoice_line_ids': invoice_vals,
+        })
+        self.invoice._onchange_invoice_line_wt_ids()
+        self.invoice.action_invoice_open()
+
+    def test_creation_withholding_tax(self):
+        domain = [('name', '=', 'Code 1040')]
+        wts = self.env['withholding.tax'].search(domain)
+        self.assertEqual(len(wts), 1, msg="Withholding tax was not created")
+
+    def test_invoice_values(self):
+        self.assertEqual(
+            self.invoice.withholding_tax_amount, 200, msg='Invoice WT amount')
+        self.assertEqual(
+            self.invoice.amount_net_pay, 800, msg='Invoice WT amount net pay')
+
+    def test_creation_statement(self):
+        domain = [('invoice_id', '=', self.invoice.id),
+                  ('withholding_tax_id', '=', self.wt1040.id)]
+        wt_statement = self.env['withholding.tax.statement'].search(domain)
+        self.assertEqual(
+            len(wt_statement), 1, msg='WT statement was not created')
+        self.assertEqual(
+            wt_statement.base, 1000, msg='WT statement Base amount')
+        self.assertEqual(
+            wt_statement.amount, 0, msg='WT statement amount applied')
+        self.assertEqual(
+            wt_statement.amount_paid, 0, msg='WT statement Base paid')
+
+    def test_invoice_payment(self):
+        ctx = {
+            'active_model': 'account.invoice',
+            'active_ids': [self.invoice.id],
+            }
+        register_payments = self.env['account.register.payments']\
+            .with_context(ctx).create({
+                'payment_date': time.strftime('%Y') + '-07-15',
+                'amount': 800,
+                'journal_id': self.journal_bank.id,
+                'payment_method_id': self.env.ref(
+                    "account.account_payment_method_manual_out").id,
+        })
+        register_payments.create_payments()
+
+        # WT payment generation
+        self.assertEqual(
+            len(self.invoice.payment_move_line_ids), 2,
+            msg='Missing WT payment')
+
+        # WT amount in account move
+        wt_line = self.invoice.payment_move_line_ids.filtered(
+            lambda r: r.account_id.id == self.wt_account_payable.id)
+        self.assertEqual(wt_line.debit, 200, msg='WT amount payment')
+
+        # WT aomunt applied in statement
+        domain = [('invoice_id', '=', self.invoice.id),
+                  ('withholding_tax_id', '=', self.wt1040.id)]
+        wt_statement = self.env['withholding.tax.statement'].search(domain)
+        self.assertEqual(wt_statement.amount, 200)