[ADD] Add account_invoice_constraint_chronology

account_invoice_constraint_chronology/tests/test_account_constraint_chronology.py

@@ -0,0 +1,133 @@
+# -*- coding: utf-8 -*-
+#
+#
+#    Authors: Adrien Peiffer
+#    Copyright (c) 2014 Acsone SA/NV (http://www.acsone.eu)
+#    All Rights Reserved
+#
+#    WARNING: This program as such is intended to be used by professional
+#    programmers who take the whole responsibility of assessing all potential
+#    consequences resulting from its eventual inadequacies and bugs.
+#    End users who are looking for a ready-to-use solution with commercial
+#    guarantees and support are strongly advised to contact a Free Software
+#    Service Company.
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as
+#    published by the Free Software Foundation, either version 3 of the
+#    License, or (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+#
+
+import openerp.tests.common as common
+from openerp import workflow
+from openerp.osv import orm
+from datetime import datetime, timedelta
+
+DB = common.DB
+ADMIN_USER_ID = common.ADMIN_USER_ID
+
+
+def get_simple_product_id(self):
+    return self.registry('product.product').create(self.cr,
+                                                   self.uid,
+                                                   {'name': 'product_test_01',
+                                                    'lst_price': 2000.00,
+                                                    }, context={})
+
+
+def get_journal_check(self, value):
+    sale_journal_id = self.ref('account.sales_journal')
+    journal_id = self.registry('account.journal').copy(self.cr,
+                                                       self.uid,
+                                                       sale_journal_id,
+                                                       {},
+                                                       context={})
+    self.registry('account.journal').write(self.cr,
+                                           self.uid,
+                                           [journal_id],
+                                           {'check_chronology': value},
+                                           context={})
+    return journal_id
+
+
+def get_simple_account_invoice_line_values(self, product_id):
+    return {'name': 'test',
+            'account_id': self.ref('account.a_sale'),
+            'price_unit': 2000.00,
+            'quantity': 1,
+            'product_id': product_id,
+            }
+
+
+def create_simple_invoice(self, journal_id, date):
+    partner_id = self.ref('base.res_partner_2')
+    product_id = get_simple_product_id(self)
+    return self.env['account.invoice']\
+        .create({'partner_id': partner_id,
+                 'account_id':
+                 self.ref('account.a_recv'),
+                 'journal_id':
+                 journal_id,
+                 'date_invoice': date,
+                 'invoice_line': [(0, 0, {'name': 'test',
+                                          'account_id':
+                                          self.ref('account.a_sale'),
+                                          'price_unit': 2000.00,
+                                          'quantity': 1,
+                                          'product_id': product_id,
+                                          }
+                                   )
+                                  ],
+                 })
+
+
+class TestAccountConstraintChronology(common.TransactionCase):
+
+    def setUp(self):
+        super(TestAccountConstraintChronology, self).setUp()
+
+    def test_invoice_draft(self):
+        journal_id = get_journal_check(self, True)
+        today = datetime.now()
+        yesterday = today - timedelta(days=1)
+        date = yesterday.strftime('%Y-%m-%d')
+        create_simple_invoice(self, journal_id, date)
+        date = today.strftime('%Y-%m-%d')
+        invoice_2 = create_simple_invoice(self, journal_id, date)
+        self.assertRaises(orm.except_orm, workflow.trg_validate, self.uid,
+                          'account.invoice', invoice_2.id, 'invoice_open',
+                          self.cr)
+
+    def test_invoice_validate(self):
+        journal_id = get_journal_check(self, True)
+        today = datetime.now()
+        tomorrow = today + timedelta(days=1)
+        date = tomorrow.strftime('%Y-%m-%d')
+        invoice = create_simple_invoice(self, journal_id, date)
+        workflow.trg_validate(self.uid, 'account.invoice', invoice.id,
+                              'invoice_open', self.cr)
+        date = today.strftime('%Y-%m-%d')
+        invoice_2 = create_simple_invoice(self, journal_id, date)
+        self.assertRaises(orm.except_orm, workflow.trg_validate, self.uid,
+                          'account.invoice', invoice_2.id, 'invoice_open',
+                          self.cr)
+
+    def test_invoice_without_date(self):
+        journal_id = get_journal_check(self, True)
+        today = datetime.now()
+        yesterday = today - timedelta(days=1)
+        date = yesterday.strftime('%Y-%m-%d')
+        create_simple_invoice(self, journal_id, date)
+        invoice_2 = create_simple_invoice(self, journal_id, False)
+        self.assertRaises(orm.except_orm, workflow.trg_validate, self.uid,
+                          'account.invoice', invoice_2.id, 'invoice_open',
+                          self.cr)