first commit

modules/account_credit_limit/tests/__init__.py

@@ -0,0 +1,2 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.

modules/account_credit_limit/tests/test_module.py

@@ -0,0 +1,86 @@
+# 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, UserWarning
+from trytond.modules.account.tests import create_chart, get_fiscalyear
+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 AccountCreditLimitTestCase(CompanyTestMixin, ModuleTestCase):
+    'Test AccountCreditLimit module'
+    module = 'account_credit_limit'
+    extras = ['account_dunning']
+
+    @with_transaction()
+    def test_check_credit_limit(self):
+        'Test check_credit_limit'
+        pool = Pool()
+        Account = pool.get('account.account')
+        Move = pool.get('account.move')
+        Journal = pool.get('account.journal')
+        Party = pool.get('party.party')
+
+        company = create_company()
+        with set_company(company):
+            create_chart(company)
+            fiscalyear = get_fiscalyear(company)
+            fiscalyear.save()
+            fiscalyear.create_period([fiscalyear])
+            period = fiscalyear.periods[0]
+
+            receivable, = Account.search([
+                    ('closed', '!=', True),
+                    ('type.receivable', '=', True),
+                    ('party_required', '=', True),
+                    ], limit=1)
+            revenue, = Account.search([
+                    ('closed', '!=', True),
+                    ('type.revenue', '=', True),
+                    ], limit=1)
+            journal, = Journal.search([], limit=1)
+            party, = Party.create([{
+                        'name': 'Party',
+                        }])
+            Move.create([{
+                        'journal': journal.id,
+                        'period': period.id,
+                        'date': period.start_date,
+                        'lines': [
+                            ('create', [{
+                                        'debit': Decimal(100),
+                                        'account': receivable.id,
+                                        'party': party.id,
+                                        }, {
+                                        'credit': Decimal(100),
+                                        'account': revenue.id,
+                                        }]),
+                            ],
+                        }])
+            self.assertEqual(party.credit_amount, Decimal(100))
+            self.assertEqual(party.credit_limit_amount, None)
+            party.check_credit_limit(Decimal(0), company)
+            party.check_credit_limit(Decimal(0), company, 'test')
+            party.check_credit_limit(Decimal(100), company)
+            party.check_credit_limit(Decimal(100), company, 'test')
+            party.credit_limit_amount = Decimal(0)
+            party.save()
+            self.assertRaises(UserError, party.check_credit_limit,
+                Decimal(1), company)
+            self.assertRaises(UserWarning, party.check_credit_limit,
+                Decimal(1), company, party)
+            party.credit_limit_amount = Decimal(200)
+            party.save()
+            party.check_credit_limit(Decimal(1), company)
+            party.check_credit_limit(Decimal(1), company, 'test')
+            self.assertRaises(UserError, party.check_credit_limit,
+                Decimal(150), company)
+            self.assertRaises(UserWarning, party.check_credit_limit,
+                Decimal(150), company, party)
+
+
+del ModuleTestCase