63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
|
|
from decimal import Decimal
|
|
import unittest
|
|
|
|
from trytond.exceptions import UserError
|
|
from trytond.modules.account_credit_note.invoice import Invoice
|
|
from trytond.tests.test_tryton import ModuleTestCase
|
|
|
|
|
|
class MockInvoice:
|
|
untaxed_amount = Decimal('100.00')
|
|
total_amount = Decimal('118.00')
|
|
|
|
|
|
class AccountCreditNoteTestCase(ModuleTestCase):
|
|
"Test Account Credit Note module"
|
|
module = 'account_credit_note'
|
|
|
|
|
|
class AccountCreditNoteLogicTestCase(unittest.TestCase):
|
|
|
|
def test_full_factor(self):
|
|
self.assertEqual(
|
|
Invoice._credit_partial_compute_factor(
|
|
MockInvoice(), 'full', None, None, 'total'),
|
|
Decimal('1'))
|
|
|
|
def test_percent_factor_supports_user_and_internal_values(self):
|
|
self.assertEqual(
|
|
Invoice._credit_partial_compute_factor(
|
|
MockInvoice(), 'percent', Decimal('20'), None, 'total'),
|
|
Decimal('0.2'))
|
|
self.assertEqual(
|
|
Invoice._credit_partial_compute_factor(
|
|
MockInvoice(), 'percent', Decimal('0.20'), None, 'total'),
|
|
Decimal('0.20'))
|
|
|
|
def test_amount_factor_uses_selected_basis(self):
|
|
self.assertEqual(
|
|
Invoice._credit_partial_compute_factor(
|
|
MockInvoice(), 'amount', None, Decimal('59'), 'total'),
|
|
Decimal('0.5'))
|
|
self.assertEqual(
|
|
Invoice._credit_partial_compute_factor(
|
|
MockInvoice(), 'amount', None, Decimal('25'), 'untaxed'),
|
|
Decimal('0.25'))
|
|
|
|
def test_amount_can_not_exceed_basis(self):
|
|
with self.assertRaises(UserError):
|
|
Invoice._credit_partial_compute_factor(
|
|
MockInvoice(), 'amount', None, Decimal('119'), 'total')
|
|
|
|
def test_fixed_amount_rejects_multiple_invoices(self):
|
|
with self.assertRaises(UserError):
|
|
Invoice.credit(
|
|
[MockInvoice(), MockInvoice()],
|
|
credit_mode='amount',
|
|
credit_amount=Decimal('10'),
|
|
credit_amount_basis='total')
|
|
|
|
|
|
del ModuleTestCase
|