def _db_cleanup(self): self.db_context.session.query(models.OpenDaylightJournal).delete() self.db_context.session.query(models.OpenDaylightPeriodicTask).delete() row0 = models.OpenDaylightPeriodicTask(task='maintenance', state=constants.PENDING) row1 = models.OpenDaylightPeriodicTask(task='hostconfig', state=constants.PENDING) self.db_context.session.merge(row0) self.db_context.session.merge(row1) self.db_context.session.flush()
def setUp(self): super(PeriodicTaskThreadTestCase, self).setUp() row = models.OpenDaylightPeriodicTask(task='test-maintenance', state=odl_const.PENDING) self.db_session.add(row) self.db_session.flush() self.thread = periodic_task.PeriodicTask('test-maintenance', 0.01) self.addCleanup(self.thread.cleanup)
def setUp(self): super(PeriodicTaskThreadTestCase, self).setUp() row = models.OpenDaylightPeriodicTask(task=TEST_TASK_NAME, state=odl_const.PENDING) self.db_session.add(row) self.db_session.flush() self.thread = periodic_task.PeriodicTask(TEST_TASK_NAME, TEST_TASK_INTERVAL) self.addCleanup(self.thread.cleanup)
def _add_tasks(self, tasks): row = [] for count, task in enumerate(tasks): row.append( models.OpenDaylightPeriodicTask(state=odl_const.PENDING, task=task)) self.db_session.add(row[count]) self.db_session.flush() rows = self.db_session.query(models.OpenDaylightPeriodicTask).all() self.assertEqual(len(tasks), len(rows))
def _test_periodic_task_lock_unlock(self, db_func, existing_state, expected_state, expected_result, task='test_task'): row = models.OpenDaylightPeriodicTask(state=existing_state, task=task) self.db_session.add(row) self.db_session.flush() self.assertEqual(expected_result, db_func(self.db_session, task)) row = self.db_session.query( models.OpenDaylightPeriodicTask).filter_by(task=task).one() self.assertEqual(expected_state, row['state'])