[MERGE] Sync with trunk

bzr revid: tde@openerp.com-20140106135700-rt0g44w5xmpbtxiu

2 files changed, 115 insertions, 0 deletions

diff --git a/addons/gamification/tests/__init__.py b/addons/gamification/tests/__init__.py
new file mode 100644
index 00000000000..9cafd242b28
--- /dev/null
+++ b/addons/gamification/tests/__init__.py
@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Business Applications
+#    Copyright (c) 2013 OpenERP S.A. <http://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_challenge
+
+checks = [
+    test_challenge,
+]
diff --git a/addons/gamification/tests/test_challenge.py b/addons/gamification/tests/test_challenge.py
new file mode 100644
index 00000000000..6b93da727ac
--- /dev/null
+++ b/addons/gamification/tests/test_challenge.py
@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Business Applications
+#    Copyright (c) 2013 OpenERP S.A. <http://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 openerp.tests import common
+
+
+class test_challenge(common.TransactionCase):
+
+    def setUp(self):
+        super(test_challenge, self).setUp()
+        cr, uid = self.cr, self.uid
+        self.data_obj = self.registry('ir.model.data')
+        self.user_obj = self.registry('res.users')
+
+        self.challenge_obj = self.registry('gamification.challenge')
+        self.line_obj = self.registry('gamification.challenge.line')
+        self.goal_obj = self.registry('gamification.goal')
+        self.badge_obj = self.registry('gamification.badge')
+        self.badge_user_obj = self.registry('gamification.badge.user')
+
+        self.demo_user_id = self.data_obj.get_object_reference(cr, uid, 'base', 'user_demo')[1]
+        self.group_user_id = self.data_obj.get_object_reference(cr, uid, 'base', 'group_user')[1]
+        self.challenge_base_id = self.data_obj.get_object_reference(cr, uid, 'gamification', 'challenge_base_discover')[1]
+        self.definition_timezone_id = self.data_obj.get_object_reference(cr, uid, 'gamification', 'definition_base_timezone')[1]
+        self.badge_id = self.data_obj.get_object_reference(cr, uid, 'gamification', 'badge_good_job')[1]
+
+    def test_00_join_challenge(self):
+        cr, uid, context = self.cr, self.uid, {}
+
+        user_ids = self.user_obj.search(cr, uid, [('groups_id', '=', self.group_user_id)])
+        challenge = self.challenge_obj.browse(cr, uid, self.challenge_base_id, context=context)
+
+        self.assertGreaterEqual(len(challenge.user_ids), len(user_ids), "Not enough users in base challenge")
+
+        self.user_obj.create(cr, uid, {
+            'name': 'R2D2',
+            'login': 'r2d2@openerp.com',
+            'email': 'r2d2@openerp.com',
+            'groups_id': [(6, 0, [self.group_user_id])]
+        }, {'no_reset_password': True})
+
+        challenge = self.challenge_obj.browse(cr, uid, self.challenge_base_id, context=context)
+        self.assertGreaterEqual(len(challenge.user_ids), len(user_ids)+1, "These are not droids you are looking for")
+
+    def test_10_reach_challenge(self):
+        cr, uid, context = self.cr, self.uid, {}
+        
+        self.challenge_obj.write(cr, uid, [self.challenge_base_id], {'state': 'inprogress'}, context=context)
+        challenge = self.challenge_obj.browse(cr, uid, self.challenge_base_id, context=context)
+        challenge_user_ids = [user.id for user in challenge.user_ids]
+
+        self.assertEqual(challenge.state, 'inprogress', "Challenge failed the change of state")
+
+        line_ids = self.line_obj.search(cr, uid, [('challenge_id', '=', self.challenge_base_id)], context=context)
+        goal_ids = self.goal_obj.search(cr, uid, [('challenge_id', '=', self.challenge_base_id), ('state', '!=', 'draft')], context=context)
+        self.assertEqual(len(goal_ids), len(line_ids)*len(challenge_user_ids), "Incorrect number of goals generated, should be 1 goal per user, per challenge line")
+
+        # demo user will set a timezone
+        self.user_obj.write(cr, uid, self.demo_user_id, {'tz': "Europe/Brussels"}, context=context)
+        goal_ids = self.goal_obj.search(cr, uid, [('user_id', '=', self.demo_user_id), ('definition_id', '=', self.definition_timezone_id)], context=context)
+        
+        self.goal_obj.update(cr, uid, goal_ids, context=context)
+        reached_goal_ids = self.goal_obj.search(cr, uid, [('id', 'in', goal_ids), ('state', '=', 'reached')], context=context)
+        self.assertEqual(set(goal_ids), set(reached_goal_ids), "Not every goal was reached after changing timezone")
+
+        # reward for two firsts as admin may have timezone
+        self.challenge_obj.write(cr, uid, self.challenge_base_id, {'reward_first_id': self.badge_id, 'reward_second_id': self.badge_id}, context=context)
+        self.challenge_obj.write(cr, uid, self.challenge_base_id,  {'state': 'done'}, context=context)
+
+        badge_ids = self.badge_user_obj.search(cr, uid, [('badge_id', '=', self.badge_id), ('user_id', '=', self.demo_user_id)])
+        self.assertGreater(len(badge_ids), 0, "Demo user has not received the badge")
\ No newline at end of file