Files
tryton/tests/test_currency.py
T

305 lines
9.4 KiB
Python

#!/usr/bin/env 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.
import sys, os
DIR = os.path.abspath(os.path.normpath(os.path.join(__file__,
'..', '..', '..', '..', '..', 'trytond')))
if os.path.isdir(DIR):
sys.path.insert(0, os.path.dirname(DIR))
import unittest
from decimal import Decimal
import trytond.tests.test_tryton
from trytond.tests.test_tryton import POOL, DB_NAME, USER, CONTEXT, test_view,\
test_depends
from trytond.transaction import Transaction
class CurrencyTestCase(unittest.TestCase):
'''
Test Currency module.
'''
def setUp(self):
trytond.tests.test_tryton.install_module('currency')
self.rate = POOL.get('currency.currency.rate')
self.currency = POOL.get('currency.currency')
self.date = POOL.get('ir.date')
def get_currency(self, code):
return self.currency.search([
('code', '=', code),
], limit=1)[0]
def test0005views(self):
'''
Test views.
'''
test_view('currency')
def test0006depends(self):
'''
Test depends.
'''
test_depends()
def test0010currencies(self):
'''
Create currencies
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
cu1_id = self.currency.create({
'name': 'cu1',
'symbol': 'cu1',
'code': 'cu1'
})
self.assert_(cu1_id)
cu2_id = self.currency.create({
'name': 'cu2',
'symbol': 'cu2',
'code': 'cu2'
})
self.assert_(cu2_id)
transaction.cursor.commit()
def test0020mon_grouping(self):
'''
Check grouping
'''
with Transaction().start(DB_NAME, USER, context=CONTEXT):
cu1_id = self.get_currency('cu1')
self.assertRaises(Exception, self.currency.write, cu1_id,
{'mon_grouping': ''})
self.assertRaises(Exception, self.currency.write, cu1_id,
{'mon_grouping': '[a]'})
self.assertRaises(Exception, self.currency.write, cu1_id,
{'mon_grouping': '[1,"a"]'})
self.assertRaises(Exception, self.currency.write, cu1_id,
{'mon_grouping': '[1,"1"]'})
def test0030rate(self):
'''
Create rates.
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
cu1_id = self.get_currency('cu1')
cu2_id = self.get_currency('cu2')
rate1 = Decimal("1.3")
rate1_id = self.rate.create({
'rate': rate1,
'currency': cu1_id,
})
self.assert_(rate1_id)
rate2_id = self.rate.create({
'rate': Decimal("1"),
'currency': cu2_id,
})
self.assert_(rate2_id)
self.assertEqual(rate1, self.currency.read(cu1_id,
['rate'])['rate'])
transaction.cursor.commit()
def test0040rate_unicity(self):
'''
Rate unicity
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
today = self.date.today()
cu_id = self.currency.create({
'name': 'cu',
'symbol': 'cu',
'code': 'cu'
})
self.rate.create({
'rate': Decimal("1.3"),
'currency': cu_id,
'date': today,
})
self.assertRaises(Exception, self.rate.create, {
'rate': Decimal("1.3"),
'currency': cu_id,
'date': today,
})
transaction.cursor.rollback()
def test0050compute_simple(self):
'''
Simple conversion
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
cu1_id = self.get_currency('cu1')
cu2_id = self.get_currency('cu2')
amount = Decimal("10")
expected = Decimal("13")
converted_amount = self.currency.compute(
cu2_id, amount, cu1_id, True)
self.assertEqual(converted_amount, expected)
transaction.cursor.commit()
def test0060compute_nonfinite(self):
'''
Conversion with rounding on non-finite decimal representation
'''
with Transaction().start(DB_NAME, USER, context=CONTEXT):
cu1_id = self.get_currency('cu1')
cu2_id = self.get_currency('cu2')
amount = Decimal("10")
expected = Decimal("7.69")
converted_amount = self.currency.compute(
cu1_id, amount, cu2_id, True)
self.assert_(converted_amount == expected)
def test0070compute_nonfinite_worounding(self):
'''
Same without rounding
'''
with Transaction().start(DB_NAME, USER, context=CONTEXT):
cu1_id = self.get_currency('cu1')
cu2_id = self.get_currency('cu2')
amount = Decimal("10")
expected = Decimal('7.692307692307692307692307692')
converted_amount = self.currency.compute(
cu1_id, amount, cu2_id, False)
self.assert_(converted_amount == expected)
def test0080compute_same(self):
'''
Conversion to the same currency
'''
with Transaction().start(DB_NAME, USER, context=CONTEXT):
cu1_id = self.get_currency('cu1')
amount = Decimal("10")
converted_amount = self.currency.compute(
cu1_id, amount, cu1_id, True)
self.assert_(converted_amount == amount)
def test0090compute_zeroamount(self):
'''
Conversion with zero amount
'''
with Transaction().start(DB_NAME, USER, context=CONTEXT):
cu1_id = self.get_currency('cu1')
cu2_id = self.get_currency('cu2')
expected = Decimal("0")
converted_amount = self.currency.compute(
cu1_id, Decimal("0"), cu2_id, True)
self.assert_(converted_amount == expected)
def test0100compute_zerorate(self):
'''
Conversion with zero rate
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
cu1_id = self.get_currency('cu1')
cu2_id = self.get_currency('cu2')
rate_ids = self.rate.search([
('currency', '=', cu1_id),
], 0, 1, None)
self.rate.write(rate_ids, {
'rate': Decimal("0"),
})
amount = Decimal("10")
self.assertRaises(Exception, self.currency.compute,
cu1_id, amount, cu2_id, True)
self.assertRaises(Exception, self.currency.compute,
cu2_id, amount, cu1_id, True)
transaction.cursor.rollback()
def test0110compute_missingrate(self):
'''
Conversion with missing rate
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
cu1_id = self.get_currency('cu1')
cu3_id = self.currency.create({
'name': 'cu3',
'symbol': 'cu3',
'code': 'cu3'
})
amount = Decimal("10")
self.assertRaises(Exception, self.currency.compute,
cu3_id, amount, cu1_id, True)
self.assertRaises(Exception, self.currency.compute,
cu1_id, amount, cu3_id, True)
transaction.cursor.rollback()
def test0120compute_bothmissingrate(self):
'''
Conversion with both missing rate
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
cu3_id = self.currency.create({
'name': 'cu3',
'symbol': 'cu3',
'code': 'cu3'
})
cu4_id = self.currency.create({
'name': 'cu4',
'symbol': 'cu4',
'code': 'cu4'
})
amount = Decimal("10")
self.assertRaises(Exception, self.currency.compute,
cu3_id, amount, cu4_id, True)
transaction.cursor.rollback()
def test0130delete_cascade(self):
'''
Test deletion of currency deletes rates
'''
with Transaction().start(DB_NAME, USER,
context=CONTEXT) as transaction:
codes = ['cu%s' % (i + 1) for i in range(2)]
currency_ids = [self.get_currency(i) for i in codes]
self.currency.delete(currency_ids)
rate_ids = self.rate.search([(
'currency', 'in', currency_ids,
)], 0, None, None)
self.assertFalse(rate_ids)
transaction.cursor.rollback()
def suite():
suite = trytond.tests.test_tryton.suite()
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(CurrencyTestCase))
return suite
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=2).run(suite())