72 lines
2.8 KiB
Python
72 lines
2.8 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.modules.account.tests import create_chart
|
|
from trytond.modules.company.tests import (
|
|
CompanyTestMixin, PartyCompanyCheckEraseMixin, create_company, set_company)
|
|
from trytond.modules.party.tests import PartyCheckReplaceMixin
|
|
from trytond.pool import Pool
|
|
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
|
|
from trytond.transaction import Transaction
|
|
|
|
|
|
class SaleTestCase(
|
|
PartyCompanyCheckEraseMixin, PartyCheckReplaceMixin, CompanyTestMixin,
|
|
ModuleTestCase):
|
|
'Test Sale module'
|
|
module = 'sale'
|
|
|
|
@with_transaction()
|
|
def test_sale_price(self):
|
|
"Test sale price"
|
|
pool = Pool()
|
|
Account = pool.get('account.account')
|
|
Template = pool.get('product.template')
|
|
Product = pool.get('product.product')
|
|
Uom = pool.get('product.uom')
|
|
|
|
company = create_company()
|
|
with set_company(company):
|
|
create_chart(company)
|
|
|
|
receivable, = Account.search([
|
|
('closed', '!=', True),
|
|
('type.receivable', '=', True),
|
|
('party_required', '=', True),
|
|
('company', '=', company.id),
|
|
], limit=1)
|
|
payable, = Account.search([
|
|
('closed', '!=', True),
|
|
('type.payable', '=', True),
|
|
('party_required', '=', True),
|
|
('company', '=', company.id),
|
|
], limit=1)
|
|
|
|
kg, = Uom.search([('name', '=', 'Kilogram')])
|
|
g, = Uom.search([('name', '=', 'Gram')])
|
|
pound, = Uom.search([('name', '=', 'Pound')])
|
|
|
|
template, = Template.create([{
|
|
'name': 'Product',
|
|
'default_uom': g.id,
|
|
'sale_uom': kg.id,
|
|
'list_price': Decimal(5),
|
|
'products': [('create', [{}])],
|
|
}])
|
|
product, = template.products
|
|
|
|
prices = Product.get_sale_price([product], quantity=100)
|
|
self.assertEqual(prices, {product.id: Decimal(5000)})
|
|
prices = Product.get_sale_price([product], quantity=1500)
|
|
self.assertEqual(prices, {product.id: Decimal(5000)})
|
|
|
|
with Transaction().set_context(uom=pound.id):
|
|
prices = Product.get_sale_price([product], quantity=0.5)
|
|
self.assertEqual(prices, {product.id: Decimal('2267.9618')})
|
|
prices = Product.get_sale_price([product], quantity=1.5)
|
|
self.assertEqual(prices, {product.id: Decimal('2267.9618')})
|
|
|
|
|
|
del ModuleTestCase
|