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'])
