first commit

modules/account_product/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/account_product/tests/test_module.py

@@ -0,0 +1,119 @@
+# 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.exceptions import UserError
+from trytond.modules.account.tests import create_chart
+from trytond.modules.company.tests import (
+    CompanyTestMixin, create_company, set_company)
+from trytond.pool import Pool
+from trytond.tests.test_tryton import ModuleTestCase, with_transaction
+
+
+class AccountProductTestCase(CompanyTestMixin, ModuleTestCase):
+    'Test AccountProduct module'
+    module = 'account_product'
+    extras = ['analytic_account']
+
+    @with_transaction()
+    def test_account_chart(self):
+        "Test creation of chart of accounts"
+        company = create_company()
+        with set_company(company):
+            create_chart(company, tax=True)
+
+    @with_transaction()
+    def test_account_chart_many_companies(self):
+        "Test creation of chart of accounts for many companies"
+        company1 = create_company()
+        with set_company(company1):
+            create_chart(company1, tax=True)
+
+        company2 = create_company()
+        with set_company(company2):
+            create_chart(company2, tax=True)
+
+    @with_transaction()
+    def test_account_used(self):
+        'Test account used'
+        pool = Pool()
+        ProductTemplate = pool.get('product.template')
+        ProductCategory = pool.get('product.category')
+        Uom = pool.get('product.uom')
+        Account = pool.get('account.account')
+
+        company = create_company()
+        with set_company(company):
+            create_chart(company)
+
+            unit, = Uom.search([
+                    ('name', '=', 'Unit'),
+                    ])
+            account_expense, = Account.search([
+                    ('closed', '!=', True),
+                    ('type.expense', '=', True),
+                    ], limit=1)
+
+            # raise when empty
+            template = ProductTemplate(
+                name='Product',
+                list_price=Decimal(10),
+                default_uom=unit.id,
+                products=[],
+                )
+            template.save()
+
+            with self.assertRaisesRegex(
+                    UserError, 'Account Category.*Product'):
+                template.account_expense_used
+
+            # with account on category
+            category = ProductCategory(
+                name='Category', accounting=True,
+                account_expense=None)
+            category.save()
+            template.account_category = category
+            template.save()
+
+            with self.assertRaisesRegex(
+                    UserError, 'Account Expense.*Product'):
+                template.account_expense_used
+
+            category.account_expense = account_expense
+            category.save()
+
+            self.assertEqual(template.account_expense_used, account_expense)
+
+            # with account on grant category
+            parent_category = ProductCategory(name='Parent Category',
+                account_expense=account_expense, accounting=True)
+            parent_category.save()
+            category.account_expense = None
+            category.account_parent = True
+            category.parent = parent_category
+            category.save()
+
+            self.assertEqual(template.account_expense_used, account_expense)
+            self.assertEqual(category.account_expense_used, account_expense)
+
+            # raise only at direct usage
+            categories = ProductCategory.create([{
+                        'name': 'Category 1',
+                        'accounting': True,
+                        'account_expense': account_expense.id,
+                        }, {
+                        'name': 'Category 2',
+                        'accounting': True,
+                        'account_expense': None,
+                        }])
+
+            self.assertEqual(categories[0].account_expense_used.id,
+                account_expense.id)
+
+            with self.assertRaisesRegex(
+                    UserError, 'Account Expense.*Category 2'):
+                categories[1].account_expense_used
+
+
+del ModuleTestCase