ADD account_tax_balance: testing wizard with date range

ADD account_tax_balance: testing buttons

IMP account_tax_balance: covering specific methods with tests

account_tax_balance/tests/test_account_tax_balance.py

@@ -4,10 +4,31 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 from openerp.tests.common import TransactionCase
+from datetime import datetime
+from dateutil.rrule import MONTHLY
 
 
 class TestAccountTaxBalance(TransactionCase):
 
+    def setUp(self):
+        super(TestAccountTaxBalance, self).setUp()
+        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):
         tax_account_id = self.env['account.account'].search(
             [('name', '=', 'Tax Paid')], limit=1).id
@@ -48,3 +69,44 @@ 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, [])