first commit

modules/analytic_account/tests/__init__.py

@@ -0,0 +1,2 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.

modules/analytic_account/tests/scenario_analytic_account.rst

@@ -0,0 +1,220 @@
+=========================
+Analytic Account Scenario
+=========================
+
+Imports::
+
+    >>> from decimal import Decimal
+
+    >>> from proteus import Model, Wizard
+    >>> from trytond.modules.account.tests.tools import (
+    ...     create_chart, create_fiscalyear, get_accounts)
+    >>> from trytond.modules.company.tests.tools import create_company
+    >>> from trytond.tests.tools import activate_modules, assertEqual
+
+Activate modules::
+
+    >>> config = activate_modules('analytic_account', create_company, create_chart)
+
+    >>> Reconciliation = Model.get('account.move.reconciliation')
+    >>> Journal = Model.get('account.journal')
+
+Create fiscal year::
+
+    >>> fiscalyear = create_fiscalyear()
+    >>> fiscalyear.click('create_period')
+    >>> period = fiscalyear.periods[0]
+
+Get accounts::
+
+    >>> accounts = get_accounts()
+    >>> receivable = accounts['receivable']
+    >>> revenue = accounts['revenue']
+    >>> expense = accounts['expense']
+    >>> journal_revenue, = Journal.find([
+    ...         ('code', '=', 'REV'),
+    ...         ])
+
+Create analytic accounts::
+
+    >>> AnalyticAccount = Model.get('analytic_account.account')
+    >>> root = AnalyticAccount(type='root', name='Root')
+    >>> root.save()
+    >>> analytic_account = AnalyticAccount(root=root, parent=root,
+    ...     name='Analytic')
+    >>> analytic_account.save()
+    >>> analytic_account2 = AnalyticAccount(root=root, parent=root,
+    ...     name='Analytic 2')
+    >>> analytic_account2.save()
+
+Create analytic rules::
+
+    >>> AnalyticRule = Model.get('analytic_account.rule')
+    >>> rule1 = AnalyticRule(account=expense)
+    >>> entry, = rule1.analytic_accounts
+    >>> entry.account = analytic_account
+    >>> rule1.save()
+    >>> rule2 = AnalyticRule(account=revenue)
+    >>> entry, = rule2.analytic_accounts
+    >>> entry.account = analytic_account2
+    >>> rule2.save()
+
+Create parties::
+
+    >>> Party = Model.get('party.party')
+    >>> customer = Party(name='Customer')
+    >>> customer.save()
+
+Create Move analytic accounts::
+
+    >>> Move = Model.get('account.move')
+    >>> move = Move()
+    >>> move.period = period
+    >>> move.journal = journal_revenue
+    >>> move.date = period.start_date
+    >>> line = move.lines.new()
+    >>> line.account = revenue
+    >>> line.credit = Decimal(42)
+    >>> analytic_line = line.analytic_lines.new()
+    >>> analytic_line.credit = line.credit
+    >>> analytic_line.account = analytic_account
+    >>> line = move.lines.new()
+    >>> line.account = receivable
+    >>> line.debit = Decimal(42)
+    >>> line.party = customer
+    >>> move.click('post')
+    >>> analytic_account.reload()
+    >>> analytic_account.credit
+    Decimal('42.00')
+    >>> analytic_account.debit
+    Decimal('0.00')
+
+Cancel reversal move::
+
+    >>> cancel_move = Wizard('account.move.cancel', [move])
+    >>> cancel_move.form.description = 'Cancel'
+    >>> cancel_move.execute('cancel')
+    >>> move.reload()
+    >>> line, = [l for l in move.lines if l.account == receivable]
+    >>> bool(line.reconciliation)
+    True
+    >>> cancel_move, = [l.move for l in line.reconciliation.lines
+    ...     if l.move != move]
+    >>> assertEqual(cancel_move.origin, move)
+    >>> analytic_account.reload()
+    >>> analytic_account.credit
+    Decimal('42.00')
+    >>> analytic_account.debit
+    Decimal('42.00')
+
+    >>> reconciliations = {
+    ...     l.reconciliation for l in cancel_move.lines if l.reconciliation}
+    >>> Reconciliation.delete(list(reconciliations))
+    >>> cancel_move = Wizard('account.move.cancel', [cancel_move])
+    >>> cancel_move.form.reversal = False
+    >>> cancel_move.execute('cancel')
+    >>> analytic_account.reload()
+    >>> analytic_account.credit
+    Decimal('42.00')
+    >>> analytic_account.debit
+    Decimal('0.00')
+
+Cancel move::
+
+    >>> cancel_move = Wizard('account.move.cancel', [move])
+    >>> cancel_move.form.description = 'Cancel'
+    >>> cancel_move.form.reversal = False
+    >>> cancel_move.execute('cancel')
+    >>> move.reload()
+    >>> line, = [l for l in move.lines if l.account == receivable]
+    >>> bool(line.reconciliation)
+    True
+    >>> cancel_move, = [l.move for l in line.reconciliation.lines
+    ...     if l.move != move]
+    >>> assertEqual(cancel_move.origin, move)
+    >>> analytic_account.reload()
+    >>> analytic_account.credit
+    Decimal('0.00')
+    >>> analytic_account.debit
+    Decimal('0.00')
+
+Create Move without analytic accounts::
+
+    >>> move = Move()
+    >>> move.period = period
+    >>> move.journal = journal_revenue
+    >>> move.date = period.start_date
+    >>> line = move.lines.new()
+    >>> line.account = revenue
+    >>> line.credit = Decimal(73)
+    >>> line = move.lines.new()
+    >>> line.account = receivable
+    >>> line.debit = Decimal(73)
+    >>> line.party = customer
+
+Check analytic lines are created on posting::
+
+    >>> move.click('post')
+    >>> line, = [l for l in move.lines if l.analytic_lines]
+    >>> analytic_line, = line.analytic_lines
+    >>> assertEqual(analytic_line.account, analytic_account2)
+    >>> analytic_line.credit
+    Decimal('73')
+    >>> assertEqual(analytic_line.date, analytic_line.move_line.date)
+
+Prepare to balance non-deferral accounts::
+
+    >>> Period = Model.get('account.period')
+    >>> AccountType = Model.get('account.account.type')
+    >>> Account = Model.get('account.account')
+    >>> journal_closing = Journal()
+    >>> journal_closing.code = 'CLO'
+    >>> journal_closing.type = 'situation'
+    >>> journal_closing.name = "Closing"
+    >>> journal_closing.save()
+    >>> period_closing = Period()
+    >>> period_closing.name = "Closing"
+    >>> period_closing.type = 'adjustment'
+    >>> period_closing.fiscalyear = fiscalyear
+    >>> period_closing.start_date = fiscalyear.end_date
+    >>> period_closing.end_date = fiscalyear.end_date
+    >>> period_closing.save()
+    >>> account_pl, = Account.find([('code', '=', '3.2.1')])
+
+Balance non-deferral accounts::
+
+    >>> balance_non_deferral = Wizard('account.fiscalyear.balance_non_deferral')
+    >>> balance_non_deferral.form.fiscalyear = fiscalyear
+    >>> balance_non_deferral.form.journal = journal_closing
+    >>> balance_non_deferral.form.period = period_closing
+    >>> balance_non_deferral.form.credit_account = account_pl
+    >>> balance_non_deferral.form.debit_account = account_pl
+    >>> balance_non_deferral.execute('balance')
+    >>> move, = balance_non_deferral.actions[0]
+    >>> move.click('post')
+    >>> [l for l in move.lines if l.analytic_lines]
+    []
+
+Prevent changing root of account with entries::
+
+    >>> root2 = AnalyticAccount(type='root', name="Root2")
+    >>> root2.save()
+    >>> analytic_account.root = root2
+    >>> analytic_account.save()
+    Traceback (most recent call last):
+        ...
+    AccessError: ...
+
+    >>> analytic_account.reload()
+    >>> analytic_account.code = "1"
+    >>> analytic_account.save()
+    >>> analytic_account.code
+    '1'
+
+Prevent changing type of analytic account with lines::
+
+    >>> analytic_account.type = 'view'
+    >>> analytic_account.save()
+    Traceback (most recent call last):
+        ...
+    AccountValidationError: ...

modules/analytic_account/tests/test_module.py

@@ -0,0 +1,286 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+from decimal import Decimal
+
+from trytond.modules.account.tests import create_chart, get_fiscalyear
+from trytond.modules.company.tests import (
+    CompanyTestMixin, create_company, set_company)
+from trytond.modules.currency.tests import create_currency
+from trytond.pool import Pool
+from trytond.tests.test_tryton import ModuleTestCase, with_transaction
+from trytond.transaction import Transaction
+
+
+class AnalyticAccountTestCase(CompanyTestMixin, ModuleTestCase):
+    'Test AnalyticAccount module'
+    module = 'analytic_account'
+
+    @with_transaction()
+    def test_account_debit_credit(self):
+        'Test account debit/credit'
+        pool = Pool()
+        Party = pool.get('party.party')
+        AnalyticAccount = pool.get('analytic_account.account')
+        Journal = pool.get('account.journal')
+        Account = pool.get('account.account')
+        Move = pool.get('account.move')
+        transaction = Transaction()
+
+        party = Party(name='Party')
+        party.save()
+        company = create_company()
+        with set_company(company):
+            root, = AnalyticAccount.create([{
+                        'type': 'root',
+                        'name': 'Root',
+                        }])
+            analytic_account, = AnalyticAccount.create([{
+                        'type': 'normal',
+                        'name': 'Analytic Account',
+                        'parent': root.id,
+                        'root': root.id,
+                        }])
+            create_chart(company)
+            fiscalyear = get_fiscalyear(company)
+            fiscalyear.save()
+            fiscalyear.create_period([fiscalyear])
+            period = fiscalyear.periods[0]
+            journal_revenue, = Journal.search([
+                    ('code', '=', 'REV'),
+                    ])
+            journal_expense, = Journal.search([
+                    ('code', '=', 'EXP'),
+                    ])
+            revenue, = Account.search([
+                    ('type.revenue', '=', True),
+                    ('closed', '=', False),
+                    ], limit=1)
+            receivable, = Account.search([
+                    ('type.receivable', '=', True),
+                    ('closed', '=', False),
+                    ], limit=1)
+            expense, = Account.search([
+                    ('type.expense', '=', True),
+                    ('closed', '=', False),
+                    ], limit=1)
+            payable, = Account.search([
+                    ('type.payable', '=', True),
+                    ('closed', '=', False),
+                    ], limit=1)
+
+            first_account_line = {
+                'account': revenue.id,
+                'credit': Decimal(100),
+                'analytic_lines': [
+                    ('create', [{
+                                'account': analytic_account.id,
+                                'credit': Decimal(100),
+                                'debit': Decimal(0),
+                                'date': period.start_date,
+                                }])
+                    ]}
+            second_account_line = {
+                'account': expense.id,
+                'debit': Decimal(30),
+                'analytic_lines': [
+                    ('create', [{
+                                'account': analytic_account.id,
+                                'debit': Decimal(30),
+                                'credit': Decimal(0),
+                                'date': period.start_date,
+                                }])
+                    ]}
+            # Create some moves
+            vlist = [{
+                    'period': period.id,
+                    'journal': journal_revenue.id,
+                    'date': period.start_date,
+                    'lines': [
+                        ('create', [first_account_line, {
+                                    'account': receivable.id,
+                                    'debit': Decimal(100),
+                                    'party': party.id,
+                                    }]),
+                        ],
+                    }, {
+                    'period': period.id,
+                    'journal': journal_expense.id,
+                    'date': period.start_date,
+                    'lines': [
+                        ('create', [second_account_line, {
+                                    'account': payable.id,
+                                    'credit': Decimal(30),
+                                    'party': party.id,
+                                    }]),
+                        ],
+                    },
+                ]
+            Move.create(vlist)
+
+            self.assertEqual((analytic_account.debit, analytic_account.credit),
+                (Decimal(30), Decimal(100)))
+            self.assertEqual(analytic_account.balance, Decimal(70))
+
+            with transaction.set_context(start_date=period.end_date):
+                analytic_account = AnalyticAccount(analytic_account.id)
+                self.assertEqual((analytic_account.debit,
+                        analytic_account.credit),
+                    (Decimal(0), Decimal(0)))
+                self.assertEqual(analytic_account.balance, Decimal(0))
+
+            with transaction.set_context(end_date=period.end_date):
+                analytic_account = AnalyticAccount(analytic_account.id)
+                self.assertEqual((analytic_account.debit,
+                        analytic_account.credit),
+                    (Decimal(30), Decimal(100)))
+                self.assertEqual(analytic_account.balance, Decimal(70))
+
+    def _test_analytic_line_state(self):
+        pool = Pool()
+        Party = pool.get('party.party')
+        AnalyticAccount = pool.get('analytic_account.account')
+        AnalyticLine = pool.get('analytic_account.line')
+        Journal = pool.get('account.journal')
+        Account = pool.get('account.account')
+        Move = pool.get('account.move')
+        MoveLine = pool.get('account.move.line')
+
+        party = Party(name='Party')
+        party.save()
+        company = create_company()
+        with set_company(company):
+            root, = AnalyticAccount.create([{
+                        'type': 'root',
+                        'name': 'Root',
+                        }])
+            analytic_account1, analytic_account2 = AnalyticAccount.create([{
+                        'type': 'normal',
+                        'name': 'Analytic Account 1',
+                        'parent': root.id,
+                        'root': root.id,
+                        }, {
+                        'type': 'normal',
+                        'name': 'Analytic Account 2',
+                        'parent': root.id,
+                        'root': root.id,
+                        }])
+            create_chart(company)
+            fiscalyear = get_fiscalyear(company)
+            fiscalyear.save()
+            fiscalyear.create_period([fiscalyear])
+            period = fiscalyear.periods[0]
+            journal_expense, = Journal.search([
+                    ('code', '=', 'EXP'),
+                    ])
+            expense, = Account.search([
+                    ('type.expense', '=', True),
+                    ('closed', '=', False),
+                    ], limit=1)
+            payable, = Account.search([
+                    ('type.payable', '=', True),
+                    ('closed', '=', False),
+                    ], limit=1)
+
+            move = Move()
+            move.period = period
+            move.journal = journal_expense
+            move.date = period.start_date
+            move.lines = [
+                MoveLine(account=expense, debit=Decimal(100)),
+                MoveLine(account=payable, credit=Decimal(100), party=party),
+                ]
+            move.save()
+            Move.post([move])
+
+            expense_line, = [l for l in move.lines if l.account == expense]
+            payable_line, = [l for l in move.lines if l.account == payable]
+
+            self.assertEqual(expense_line.analytic_state, 'draft')
+            self.assertEqual(payable_line.analytic_state, 'valid')
+
+            expense_line.analytic_lines = [
+                AnalyticLine(
+                    account=analytic_account1,
+                    debit=Decimal(50),
+                    date=period.start_date),
+                AnalyticLine(
+                    account=analytic_account2,
+                    debit=Decimal(50),
+                    date=period.start_date),
+                ]
+            expense_line.save()
+
+            self.assertEqual(expense_line.analytic_state, 'valid')
+
+    @with_transaction()
+    def test_move_line_state(self):
+        "Test move line state"
+        self._test_analytic_line_state()
+
+    @with_transaction()
+    def test_move_line_state_roots_several_companies(self):
+        "Test move line state with roots from several companies"
+        pool = Pool()
+        Account = pool.get('analytic_account.account')
+
+        extra_company = create_company()
+        with set_company(extra_company):
+            root, = Account.create([{
+                        'type': 'root',
+                        'name': 'Root',
+                        }])
+            analytic_account, = Account.create([{
+                        'type': 'normal',
+                        'name': 'Analytic Account',
+                        'parent': root.id,
+                        'root': root.id,
+                        }])
+        self._test_analytic_line_state()
+
+    @with_transaction()
+    def test_account_distribute(self):
+        "Test account distribute"
+        pool = Pool()
+        Account = pool.get('analytic_account.account')
+        Distribution = pool.get('analytic_account.account.distribution')
+
+        currency = create_currency('usd')
+        account1 = Account(type='normal', currency=currency)
+        account2 = Account(type='normal', currency=currency)
+        account = Account(type='distribution', currency=currency)
+        account.distributions = [
+            Distribution(account=account1, ratio=Decimal('0.7')),
+            Distribution(account=account2, ratio=Decimal('0.3')),
+            ]
+
+        self.assertListEqual(
+            account.distribute(Decimal('100.03')),
+            [(account1, Decimal('70.02')), (account2, Decimal('30.01'))])
+
+    @with_transaction()
+    def test_account_distribute_remainder(self):
+        "Test account distribute remainder"
+        pool = Pool()
+        Account = pool.get('analytic_account.account')
+        Distribution = pool.get('analytic_account.account.distribution')
+
+        currency = create_currency('usd')
+        account1 = Account(type='normal', currency=currency)
+        account2 = Account(type='normal', currency=currency)
+        account3 = Account(type='normal', currency=currency)
+        account = Account(type='distribution', currency=currency)
+        account.distributions = [
+            Distribution(account=account1, ratio=Decimal('0.5')),
+            Distribution(account=account2, ratio=Decimal('0.375')),
+            Distribution(account=account3, ratio=Decimal('0.125')),
+            ]
+
+        self.assertListEqual(
+            account.distribute(Decimal('65.35')), [
+                (account1, Decimal('32.67')),
+                (account2, Decimal('24.51')),
+                (account3, Decimal('8.17'))])
+
+
+del ModuleTestCase

modules/analytic_account/tests/test_scenario.py

@@ -0,0 +1,8 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+from trytond.tests.test_tryton import load_doc_tests
+
+
+def load_tests(*args, **kwargs):
+    return load_doc_tests(__name__, __file__, *args, **kwargs)