[FIX+IMP] account_tax_balance:

* Tests
* PEP8
* Use invoice._convert_to_write(invoice._cache). This way, the onchange will be inheritable and will add here also the added values
* better get_context_values
* unify method compute_balance
* open move lines linked to balance

account_tax_balance/README.rst

@@ -12,7 +12,7 @@ It depends on date_range module and exposes 'compute' methods that can be called
 Usage
 =====
 
-Accounting --> Reporting --> Open Tax Balances
+Accounting --> Reporting --> Taxes Balance
 
 Select the company, the date range, the target moves and 'open taxes'
 
@@ -42,6 +42,7 @@ Contributors
 ------------
 
 * Lorenzo Battistini <lorenzo.battistini@agilebg.com>
+* Giovanni Capalbo <giovanni@therp.nl>
 
 Maintainer
 ----------

account_tax_balance/__openerp__.py

@@ -7,7 +7,8 @@
     "version": "9.0.1.0.0",
     "category": "Accounting & Finance",
     "website": "https://www.agilebg.com/",
-    "author": "Agile Business Group, Odoo Community Association (OCA)",
+    "author": "Agile Business Group, Therp BV, "
+              "Odoo Community Association (OCA)",
     "license": "AGPL-3",
     "application": False,
     "installable": True,

account_tax_balance/models/account_tax.py

@@ -2,7 +2,7 @@
 # © 2016 Lorenzo Battistini - Agile Business Group
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-from openerp import models, fields
+from openerp import models, fields, api
 
 
 class AccountTax(models.Model):
@@ -13,31 +13,18 @@ class AccountTax(models.Model):
         string="Base Balance", compute="_compute_balance")
 
     def get_context_values(self):
-        if not self.env.context.get('from_date'):
-            from_date = fields.Date.context_today(self)
-        else:
-            from_date = self.env.context['from_date']
-        if not self.env.context.get('to_date'):
-            to_date = fields.Date.context_today(self)
-        else:
-            to_date = self.env.context['to_date']
-        if not self.env.context.get('target_move'):
-            target_move = 'posted'
-        else:
-            target_move = self.env.context['target_move']
-        if not self.env.context.get('company_id'):
-            company_id = self.env.user.company_id.id
-        else:
-            company_id = self.env.context['company_id']
-        return from_date, to_date, company_id, target_move
+        context = self.env.context
+        return (
+            context.get('from_date', fields.Date.context_today(self)),
+            context.get('to_date', fields.Date.context_today(self)),
+            context.get('company_id', self.env.user.company_id.id),
+            context.get('target_move', 'posted')
+        )
 
     def _compute_balance(self):
-        from_date, to_date, company_id, target_move = self.get_context_values()
         for tax in self:
-            tax.balance = tax.compute_balance(
-                from_date, to_date, company_id, target_move)
-            tax.base_balance = tax.compute_base_balance(
-                from_date, to_date, company_id, target_move)
+            tax.balance = tax.compute_balance(tax_or_base='tax')
+            tax.base_balance = tax.compute_balance(tax_or_base='base')
 
     def get_target_state_list(self, target_move="posted"):
         if target_move == 'posted':
@@ -48,39 +35,63 @@ class AccountTax(models.Model):
             state = []
         return state
 
-    def get_move_line_domain(self, from_date, to_date, company_id):
+    def get_move_line_partial_domain(self, from_date, to_date, company_id):
         return [
             ('date', '<=', to_date),
             ('date', '>=', from_date),
             ('company_id', '=', company_id),
         ]
 
-    def compute_balance(
-        self, from_date, to_date, company_id, target_move="posted"
-    ):
+    def compute_balance(self, tax_or_base='tax'):
         self.ensure_one()
-        move_line_model = self.env['account.move.line']
-        state_list = self.get_target_state_list(target_move)
-        domain = self.get_move_line_domain(from_date, to_date, company_id)
-        domain.extend([
+        move_lines = self.get_move_lines_domain(tax_or_base=tax_or_base)
+        # balance is debit - credit whereas on tax return you want to see what
+        # vat has to be paid so:
+        # VAT on sales (credit) - VAT on purchases (debit).
+        total = -sum([l.balance for l in move_lines])
+        return total
+
+    def get_balance_domain(self, state_list):
+        return [
             ('move_id.state', 'in', state_list),
             ('tax_line_id', '=', self.id),
-        ])
-        move_lines = move_line_model.search(domain)
-        total = sum([l.balance for l in move_lines])
-        return total
+        ]
 
-    def compute_base_balance(
-        self, from_date, to_date, company_id, target_move="posted"
-    ):
-        self.ensure_one()
-        move_line_model = self.env['account.move.line']
-        state_list = self.get_target_state_list(target_move)
-        domain = self.get_move_line_domain(from_date, to_date, company_id)
-        domain.extend([
+    def get_base_balance_domain(self, state_list):
+        return [
             ('move_id.state', 'in', state_list),
             ('tax_ids', 'in', self.id),
-        ])
-        move_lines = move_line_model.search(domain)
-        total = sum([l.balance for l in move_lines])
-        return total
+        ]
+
+    def get_move_lines_domain(self, tax_or_base='tax'):
+        move_line_model = self.env['account.move.line']
+        from_date, to_date, company_id, target_move = self.get_context_values()
+        state_list = self.get_target_state_list(target_move)
+        domain = self.get_move_line_partial_domain(
+            from_date, to_date, company_id)
+        balance_domain = []
+        if tax_or_base == 'tax':
+            balance_domain = self.get_balance_domain(state_list)
+        elif tax_or_base == 'base':
+            balance_domain = self.get_base_balance_domain(state_list)
+        domain.extend(balance_domain)
+        return move_line_model.search(domain)
+
+    def get_lines_action(self, tax_or_base='tax'):
+        move_lines = self.get_move_lines_domain(tax_or_base=tax_or_base)
+        move_line_ids = [l.id for l in move_lines]
+        action = self.env.ref('account.action_account_moves_all_tree')
+        vals = action.read()[0]
+        vals['context'] = {}
+        vals['domain'] = [('id', 'in', move_line_ids)]
+        return vals
+
+    @api.multi
+    def view_tax_lines(self):
+        self.ensure_one()
+        return self.get_lines_action(tax_or_base='tax')
+
+    @api.multi
+    def view_base_lines(self):
+        self.ensure_one()
+        return self.get_lines_action(tax_or_base='base')

account_tax_balance/tests/__init__.py

@@ -1,3 +1,6 @@
-#Accounting tests extending AccountingTestCase
+# -*- coding: utf-8 -*-
+# © 2016 Lorenzo Battistini - Agile Business Group
+# © 2016 Giovanni Capalbo
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 from . import test_account_tax_balance

account_tax_balance/tests/test_account_tax_balance.py

@@ -1,51 +1,54 @@
-#from openerp.addons.account.tests.account_test_users import AccountTestUsers
+# -*- coding: utf-8 -*-
+# © 2016 Lorenzo Battistini - Agile Business Group
+# © 2016 Giovanni Capalbo <giovanni@therp.nl>
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
 from openerp.tests.common import TransactionCase
-from openerp.tools import float_compare
+from datetime import datetime
+from dateutil.rrule import MONTHLY
 
 
 class TestAccountTaxBalance(TransactionCase):
 
     def setUp(self):
         super(TestAccountTaxBalance, self).setUp()
-        self.fixed_tax = self.tax_model.create({
-            'name': "Fixed tax",
-            'amount_type': 'fixed',
-            'amount': 10.0,
-            'sequence': 1,
-        })
-        self.fixed_tax_bis = self.tax_model.create({
-            'name': "Fixed tax bis",
-            'amount_type': 'fixed',
-            'amount': 15,
-            'sequence': 2,
-        })
-        self.percent_tax = self.tax_model.create({
-            'name': "Percent tax",
-            'amount_type': 'percent',
-            'amount': 10.0,
-            'sequence': 3,
-        })
-        self.bank_journal = self.env['account.journal'].search([('type', '=', 'bank'), ('company_id', '=', self.account_manager.company_id.id)])[0]
-        self.bank_account = self.bank_journal.default_debit_account_id
-        self.expense_account = self.env['account.account'].search([('user_type_id.type', '=', 'payable')], limit=1) #Should be done by onchange later
-
+        self.range_type = self.env['date.range.type'].create(
+            {'name': 'Fiscal year',
+             'company_id': False,
+             'allow_overlap': False})
+        self.range_generator = self.env['date.range.generator']
+        self.current_year = datetime.now().year
+        self.current_month = datetime.now().month
+        range_generator = self.range_generator.create({
+            'date_start': '%s-01-01' % self.current_year,
+            'name_prefix': '%s-' % self.current_year,
+            'type_id': self.range_type.id,
+            'duration_count': 1,
+            'unit_of_time': MONTHLY,
+            'count': 12})
+        range_generator.action_apply()
+        self.range = self.env['date.range']
 
     def test_tax_balance(self):
-        company_id = self.env['res.users'].browse(self.env.uid).company_id.id
+        tax_account_id = self.env['account.account'].search(
+            [('name', '=', 'Tax Paid')], limit=1).id
         tax = self.env['account.tax'].create({
             'name': 'Tax 10.0',
             'amount': 10.0,
             'amount_type': 'fixed',
+            'account_id': tax_account_id,
         })
-        analytic_account = self.env['account.analytic.account'].create({
-            'name': 'test account',
-        })
-        invoice_account = self.env['account.account'].search([('user_type_id', '=', self.env.ref('account.data_account_type_receivable').id)], limit=1).id
-        invoice_line_account = self.env['account.account'].search([('user_type_id', '=', self.env.ref('account.data_account_type_expenses').id)], limit=1).id
+        invoice_account_id = self.env['account.account'].search(
+            [('user_type_id', '=', self.env.ref(
+                'account.data_account_type_receivable'
+            ).id)], limit=1).id
+        invoice_line_account_id = self.env['account.account'].search(
+            [('user_type_id', '=', self.env.ref(
+                'account.data_account_type_expenses').id)], limit=1).id
         invoice = self.env['account.invoice'].create({
             'partner_id': self.env.ref('base.res_partner_2').id,
-            'account_id': invoice_account,
-            'type': 'in_invoice',
+            'account_id': invoice_account_id,
+            'type': 'out_invoice',
         })
 
         self.env['account.invoice.line'].create({
@@ -54,13 +57,12 @@ class TestAccountTaxBalance(TransactionCase):
             'price_unit': 100.0,
             'invoice_id': invoice.id,
             'name': 'product that cost 100',
-            'account_id': invoice_line_account,
+            'account_id': invoice_line_account_id,
             'invoice_line_tax_ids': [(6, 0, [tax.id])],
-            'account_analytic_id': analytic_account.id,
         })
-
-        # :     check that Initially supplier bill state is "Draft"
-        self.assertTrue((invoice.state == 'draft'), "Initially vendor bill state is Draft")
+        invoice._onchange_invoice_line_ids()
+        invoice._convert_to_write(invoice._cache)
+        self.assertEqual(invoice.state, 'draft')
 
         # change the state of invoice to open by clicking Validate button
         invoice.signal_workflow('invoice_open')
@@ -68,5 +70,43 @@ class TestAccountTaxBalance(TransactionCase):
         self.assertEquals(tax.base_balance, 100)
         self.assertEquals(tax.balance, 10)
 
+        # testing wizard
+        current_range = self.range.search([
+            ('date_start', '=', '%s-%s-01' % (
+                self.current_year, self.current_month))
+        ])
+        wizard = self.env['wizard.open.tax.balances'].new({})
+        self.assertFalse(wizard.from_date)
+        self.assertFalse(wizard.to_date)
+        wizard = self.env['wizard.open.tax.balances'].new({
+            'date_range_id': current_range[0].id,
+        })
+        wizard.onchange_date_range_id()
+        wizard._convert_to_write(wizard._cache)
+        action = wizard.open_taxes()
+        self.assertEqual(
+            action['context']['from_date'], current_range[0].date_start)
+        self.assertEqual(
+            action['context']['to_date'], current_range[0].date_end)
+        self.assertEqual(
+            action['xml_id'], 'account_tax_balance.action_tax_balances_tree')
 
+        # testing buttons
+        tax_action = tax.view_tax_lines()
+        base_action = tax.view_base_lines()
+        self.assertTrue(
+            tax_action['domain'][0][2][0] in
+            [l.id for l in invoice.move_id.line_ids])
+        self.assertEqual(
+            tax_action['xml_id'], 'account.action_account_moves_all_tree')
+        self.assertTrue(
+            base_action['domain'][0][2][0] in
+            [l.id for l in invoice.move_id.line_ids])
+        self.assertEqual(
+            base_action['xml_id'], 'account.action_account_moves_all_tree')
 
+        # test specific method
+        state_list = tax.get_target_state_list(target_move='all')
+        self.assertEqual(state_list, ['posted', 'draft'])
+        state_list = tax.get_target_state_list(target_move='whatever')
+        self.assertEqual(state_list, [])

account_tax_balance/views/account_tax_view.xml

@@ -13,6 +13,8 @@
                     <field name="account_id"/>
                     <field name="balance" sum="Total"></field>
                     <field name="base_balance" sum="Base Total"></field>
+                    <button type="object" name="view_tax_lines" string="View tax lines" icon="gtk-find"></button>
+                    <button type="object" name="view_base_lines" string="View base lines" icon="gtk-copy"></button>
                 </tree>
             </field>
          </record>
@@ -36,7 +38,7 @@
             </field>
          </record>
         <record id="action_tax_balances_tree" model="ir.actions.act_window">
-            <field name="name">Tax Balances</field>
+            <field name="name">Taxes Balance</field>
             <field name="res_model">account.tax</field>
             <field name="view_type">form</field>
             <field name="view_mode">tree</field>

account_tax_balance/wizard/open_tax_balances_view.xml

@@ -5,7 +5,7 @@
             <field name="name">wizard_open_tax_balances</field>
             <field name="model">wizard.open.tax.balances</field>
             <field name="arch" type="xml">
-                <form string="Open Tax Balances">
+                <form string="Taxes Balance">
                     <group>
                         <field name="company_id"/>
                         <field name="date_range_id"/>
@@ -23,7 +23,7 @@
          </record>
 
         <record id="action_open_tax_balances" model="ir.actions.act_window">
-            <field name="name">Open Tax Balances</field>
+            <field name="name">Taxes Balance</field>
             <field name="res_model">wizard.open.tax.balances</field>
             <field name="view_type">form</field>
             <field name="view_mode">form</field>