120 lines
4.0 KiB
Python
120 lines
4.0 KiB
Python
# 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
|