first commit

modules/account_fr/tests/tools.py

@@ -0,0 +1,70 @@
+# 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 proteus import Model, Wizard
+from trytond.modules.company.tests.tools import get_company
+
+__all__ = ['create_chart', 'get_accounts']
+
+
+def create_chart(company=None, config=None):
+    "Create chart of accounts"
+    AccountTemplate = Model.get('account.account.template', config=config)
+    ModelData = Model.get('ir.model.data', config=config)
+
+    if not company:
+        company = get_company(config=config)
+    chart_id = ModelData.get_id('account_fr', 'root', config.context)
+
+    account_template = AccountTemplate(chart_id)
+
+    create_chart = Wizard('account.create_chart', config=config)
+    create_chart.execute('account')
+    create_chart.form.account_template = account_template
+    create_chart.form.company = company
+    create_chart.execute('create_account')
+
+    accounts = get_accounts(company, config=config)
+
+    create_chart.form.account_receivable = accounts['receivable']
+    create_chart.form.account_payable = accounts['payable']
+    create_chart.execute('create_properties')
+    return create_chart
+
+
+def get_accounts(company=None, config=None):
+    "Return accounts per kind"
+    Account = Model.get('account.account', config=config)
+
+    if not company:
+        company = get_company(config=config)
+
+    accounts = {}
+    accounts['receivable'], = Account.find([
+            ('type.receivable', '=', True),
+            ('company', '=', company.id),
+            ('code', '=', '4111'),
+            ], limit=1)
+    accounts['payable'], = Account.find([
+            ('type.payable', '=', True),
+            ('company', '=', company.id),
+            ('code', '=', '4011'),
+            ], limit=1)
+    accounts['revenue'], = Account.find([
+            ('type.revenue', '=', True),
+            ('company', '=', company.id),
+            ('code', '=', '7011'),
+            ], limit=1)
+    accounts['expense'], = Account.find([
+            ('type.expense', '=', True),
+            ('company', '=', company.id),
+            ('code', '=', '6071'),
+            ], limit=1)
+    accounts['cash'], = Account.find([
+            ('company', '=', company.id),
+            ('code', '=', '5311'),
+            ])
+    accounts['tax'], = Account.find([
+            ('company', '=', company.id),
+            ('code', '=', '44558'),
+            ])
+    return accounts