[IMP] Add test

res_partner_mails_count/tests/__init__.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Business Applications
+#    Copyright (C) 2013-Today OpenERP SA (<http://www.openerp.com>)
+#
+#    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/>
+#
+##############################################################################
+
+from . import test_mail

res_partner_mails_count/tests/test_mail.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+
+from openerp.tests.common import TransactionCase
+
+
+class test_message_count(TransactionCase):
+    post_install = True
+    def test_count(self):
+        new_partner1 = self.env['res.partner'].sudo().create({'name': 'rpmc Test Partner one', 'email': 'tt@tt', 'notify_email': 'always'})
+        new_partner2 = self.env['res.partner'].sudo().create({'name': 'rpmc Test Partner two', 'email': 'rr@rr', 'notify_email': 'always'})
+        self.assertEqual(new_partner1.mails_to, 0, 'rpmc: new partner have mails_to != 0')
+        mail_compose = self.env['mail.compose.message']
+        compose = mail_compose.with_context(
+            {
+                'default_composition_mode': 'comment',
+                'default_model': 'res.partner',
+                'default_res_id': new_partner1.id,
+            }).create(
+            {
+                'subject': 'test subj',
+                'body': 'test body',
+                'partner_ids': [(4, new_partner2.id)],
+                'email_from': 'tt@tt',
+                'author_id': new_partner1.id
+            })
+        compose.send_mail()
+        self.assertEqual(new_partner1.mails_to, 0)
+        self.assertEqual(new_partner1.mails_from, 1, 'rpmc: one message but mails_from != 1')
+        self.assertEqual(new_partner2.mails_to, 1, 'rpmc: one message but mails_to != 1')
+        self.assertEqual(new_partner2.mails_from, 0)
+        compose.send_mail()
+        self.assertEqual(new_partner1.mails_to, 0)
+        self.assertEqual(new_partner1.mails_from, 2, 'rpmc: one message but mails_from != 2')
+        self.assertEqual(new_partner2.mails_to, 2, 'rpmc: one message but mails_to != 2')
+        self.assertEqual(new_partner2.mails_from, 0)