GOLEM/golem_activity_queue/tests/test_golem_member.py

# -*- coding: utf-8 -*-

#    Copyright 2018 Youssef El Ouahby <youssef@yaltik.com>
#    Copyright 2018 Fabien Bourgeois <fabien@yaltik.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/>.

""" GOLEM Member testing """

from odoo.tests.common import TransactionCase


class TestGolemMember(TransactionCase):
    """ GOLEM member testing """

    def setUp(self, *args, **kwargs):
        """ Bootstrap member """
        super(TestGolemMember, self).setUp(*args, **kwargs)
        #création du saison
        self.season = self.env['golem.season'].create({'name': u'Season 1'})
        self.data_member_1 = {'lastname': u'LAST1',
                              'firstname': u'First1',
                              'season_ids':[(4, self.season.id, False)]}
        self.data_member_2 = {'lastname': u'LAST2',
                              'firstname': u'First2',
                              'season_ids':[(4, self.season.id, False)]}
        self.member = self.env['golem.member']
        type_id = self.env.ref('golem_activity.golem_activity_type_activity')
        self.data_activity = {
            'name': u'Activity 1',
            'season_id': self.season.id,
            'type_id': type_id.id
        }
        self.activity = self.env['golem.activity']
        self.activity_registration = self.env['golem.activity.registration']
        self.activity_queue = self.env['golem.activity.queue']

    def test_member_creation(self):
        """ Test member queue default """
        member1 = self.member.create(self.data_member_1)
        self.assertFalse(member1.activity_queue_ids)

    def test_automated_registration(self):
        """ Test automated registration """
        member1 = self.member.create(self.data_member_1)
        member2 = self.member.create(self.data_member_2)
        activity = self.activity.create(self.data_activity)

        registration_data = {'activity_id' : activity.id, 'member_id' : member1.id}
        queue_data = {'activity_id' : activity.id, 'member_id' : member2.id}
        self.assertTrue(activity.queue_allowed)
        self.assertTrue(activity.auto_registration_from_queue)

        activity.places = 1
        registration = self.activity_registration.create(registration_data)
        self.activity_queue.create(queue_data)

        self.assertEqual(activity.activity_registration_ids[0].member_id, member1)
        self.assertEqual(activity.activity_queue_ids[0].member_id, member2)

        registration.unlink()
        self.assertEqual(activity.activity_registration_ids[0].member_id, member2)
        self.assertFalse(activity.activity_queue_ids)