1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
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.modules.company.tests import (
CompanyTestMixin, PartyCompanyCheckEraseMixin, create_company, set_company)
from trytond.pool import Pool
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
def create_product(name, list_price, categories=None):
pool = Pool()
Template = pool.get('product.template')
Product = pool.get('product.product')
Uom = pool.get('product.uom')
unit, = Uom.search([('name', '=', 'Unit')])
template = Template(
name=name,
type='service',
list_price=list_price,
default_uom=unit,
products=None,
)
if categories:
template.categories = categories
template.save()
product = Product(template=template)
product.save()
return product
def create_plan(lines):
pool = Pool()
Plan = pool.get('commission.plan')
commission_product = create_product("Commission", Decimal(10), [])
plan, = Plan.create([{
'name': "Commission Plan",
'commission_product': commission_product.id,
'lines': [('create', lines)]
}])
return plan
class CommissionTestCase(
PartyCompanyCheckEraseMixin, CompanyTestMixin, ModuleTestCase):
'Test Commission module'
module = 'commission'
extras = ['sale', 'stock', 'account_invoice_stock']
@with_transaction()
def test_plan_category(self):
"Test plan with category"
pool = Pool()
Category = pool.get('product.category')
category = Category(name="Category")
category.save()
child_category = Category(name="Child Category", parent=category)
child_category.save()
company = create_company()
with set_company(company):
product = create_product("Other", Decimal(10), [category])
plan = create_plan([{
'category': category.id,
'formula': 'amount * 0.8',
}, {
'formula': 'amount',
}])
self.assertEqual(plan.compute(Decimal(1), product), Decimal('0.8'))
template = product.template
template.categories = []
template.save()
self.assertEqual(plan.compute(Decimal(1), product), Decimal(1))
template.categories = [child_category]
template.save()
self.assertEqual(plan.compute(Decimal(1), product), Decimal('0.8'))
@with_transaction()
def test_plan_no_product(self):
"Test plan with no product"
pool = Pool()
Category = pool.get('product.category')
PlanLine = pool.get('commission.plan.line')
category = Category(name="Category")
category.save()
company = create_company()
with set_company(company):
product = create_product("Other", Decimal(10))
plan = create_plan([{
'category': category.id,
'formula': 'amount * 0.8',
}, {
'product': product.id,
'formula': 'amount * 0.7',
}, {
'formula': 'amount',
}])
self.assertEqual(plan.compute(Decimal(1), None), Decimal(1))
PlanLine.delete(plan.lines[1:])
self.assertEqual(plan.compute(Decimal(1), None), None)
del ModuleTestCase
|
