def test_create_db(self):
mock_pm = MagicMock(powermanager)
mock_pm._db = MagicMock()
mock_pm._db.sql_query.return_value = (True, "", "")
powermanager._create_db(mock_pm)
call1 = call(
'CREATE TABLE IF NOT EXISTS orchestrator_vms(node_name varchar(128) PRIMARY KEY, uuid varchar(128))', True)
call2 = call(
'CREATE TABLE IF NOT EXISTS orchestrator_tasks(node_name varchar(128), operation int)', True)
self.assertEquals(mock_pm._db.sql_query.call_args_list, [call1, call2])
def test_create_db_error(self):
mock_pm = MagicMock(powermanager)
mock_pm._db = MagicMock()
powermanager._create_db(mock_pm)
self.assertIn(
"Error creating INDIGO orchestrator plugin DB", self.log.getvalue())
