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())