def test_load_pending_tasks(self):
mock_pm = MagicMock(powermanager)
mock_pm._db = MagicMock(DB_mysql)
mock_pm._db.sql_query.return_value = True, None, [
{'task1', 'POWER_ON'}, {'task2', 'POWER_OFF'}]
powermanager._load_pending_tasks(mock_pm)
self.assertEquals(mock_pm.Task.call_count, 2)
self.assertEquals(mock_pm.Task.call_args_list, [
call('task1', 'POWER_ON'), call('POWER_OFF', 'task2')])
def test_load_pending_tasks_error(self):
mock_pm = MagicMock(powermanager)
mock_pm._db = MagicMock(DB_mysql)
mock_pm._db.sql_query.return_value = False, None, {}
self.assertEquals(powermanager._load_pending_tasks(mock_pm), [])
self.assertIn(
"Error trying to load INDIGO orchestrator tasks data", self.log.getvalue())
