GOLEM/golem_activity/tests/test_golem_activity.py

48 lines
2.0 KiB
Python

# -*- coding: utf-8 -*-
# Copyright 2016 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/>.
from openerp.tests.common import TransactionCase
from openerp.models import ValidationError
class GolemActivityTestCase(TransactionCase):
def setUp(self):
super(GolemActivityTestCase, self).setUp()
self.season_model = self.env['golem.season'].sudo()
season_data = {'name': u'Current', 'date_start': '2010-01-01',
'date_end': '2010-12-31'}
self.season_current = self.season_model.create(season_data)
self.activity_model = self.env['golem.activity'].sudo()
def test_activity_creation(self):
""" Test creation of activity and periods """
categ = self.ref('golem_activity.golem_product_category_activities')
adata = {'name': 'a1', 'season_id': self.season_current.id,
'categ_id': categ}
a1 = self.activity_model.create(adata)
a1.onchange_season_dates()
self.assertEqual(a1.name, 'a1')
self.assertEqual(a1.date_start, self.season_current.date_start)
self.assertEqual(a1.date_end, self.season_current.date_end)
self.assertTrue(a1.is_current)
adata.update({'name': 'a2', 'date_start': '2010-01-01',
'date_end': '2009-12-01'})
with self.assertRaises(ValidationError):
self.activity_model.create(adata)