def setUp(self):
""" Set up the environnment, ran before every tests. """
self._clean()
self.old_path = tests.DB_PATH
tests.DB_PATH = DB_PATH
super(Crontests, self).setUp()
fedocal_cron.fedocal.APP.config['DB_URL'] = DB_PATH
fedocal_cron.fedocal.APP.config['TESTING'] = True
fedocal_cron.fedocal.APP.debug = True
fedocal_cron.fedocal.APP.logger.handlers = []
fedocal_cron.fedocal.APP.logger.setLevel(logging.CRITICAL)
fedocal_cron.fedocal.SESSION = self.session
# Fills some data in the database in memory
# A calendar
calendar = model.Calendar(
calendar_name='test_calendar',
calendar_contact='*****@*****.**',
calendar_description='This is a test calendar',
calendar_editor_group='fi-apprentice',
calendar_admin_group='infrastructure-main2')
calendar.save(self.session)
self.session.commit()
self.assertNotEqual(calendar, None)
def test_init_calendar(self):
""" Test the Calendar init function. """
obj = model.Calendar(calendar_name='test_calendar',
calendar_contact='*****@*****.**',
calendar_description='This is a test calendar',
calendar_editor_group='fi-apprentice',
calendar_admin_group='infrastructure-main2')
obj.save(self.session)
self.session.commit()
self.assertNotEqual(obj, None)
obj = model.Calendar(
calendar_name='test_calendar2',
calendar_contact='*****@*****.**',
calendar_description='This is another test calendar',
calendar_editor_group='packager')
obj.save(self.session)
self.session.commit()
self.assertNotEqual(obj, None)
obj = model.Calendar(
calendar_name='test_calendar3',
calendar_contact='*****@*****.**',
calendar_description='This is the third test calendar')
obj.save(self.session)
self.session.commit()
self.assertNotEqual(obj, None)
obj = model.Calendar(
calendar_name='test_calendar4',
calendar_contact='*****@*****.**',
calendar_description='This is yet another test calendar',
calendar_editor_group='fi-apprentice')
obj.save(self.session)
self.session.commit()
self.assertNotEqual(obj, None)
obj = model.Calendar(
calendar_name='test_calendar_disabled',
calendar_contact='*****@*****.**',
calendar_description='This is a disabled test calendar',
calendar_status='Disabled')
obj.save(self.session)
self.session.commit()
self.assertNotEqual(obj, None)
def __setup_calendar(self):
""" Set up basic calendar information. """
obj = model.Calendar(calendar_name='test_calendar',
calendar_contact='*****@*****.**',
calendar_description='This is a test calendar',
calendar_editor_group='fi-apprentice',
calendar_admin_group='infrastructure-main2')
obj.save(self.session)
self.session.commit()
self.assertNotEqual(obj, None)
def convert_calendars(fed03_sess, fed04_sess):
''' Convert the Calendars from fedocal-0.3.x to fedocal-0.4.0.
'''
cnt = 0
for calendar in fed03_sess.query(F03Calendars).all():
calendarobj = model.Calendar(
calendar_name=calendar.calendar_name,
calendar_contact=calendar.calendar_contact,
calendar_description=calendar.calendar_description,
calendar_editor_group=calendar.calendar_editor_group,
calendar_admin_group=calendar.calendar_admin_group,
calendar_status=calendar.calendar_status)
fed04_sess.add(calendarobj)
cnt += 1
fed04_sess.commit()
print '%s calendars added' % cnt
