class RecoveryCreateClusterTest(unittest.TestCase): def setUp(self): Recovery.clusterList = {} self.recovery = Recovery() self.trueNodeList = ["compute1", "compute2"] self.falseNodeList = ["compute426", "compute321"] self.clusterName = "TestCluster-%s" % datetime.datetime.now().strftime("%Y/%m/%d-%H:%M") self.uuid = self.recovery.createCluster(self.clusterName).split(";")[1] def test_addNode_Success(self): self.assertEqual(self.recovery.addNode(self.uuid, self.trueNodeList).split(";")[0], "0") def test_addNode_WrongId(self): self.assertEqual(self.recovery.addNode("123456789", self.trueNodeList).split(";")[0], "1") def test_addNode_WrongNodeList(self): self.assertEqual(self.recovery.addNode("123456789", self.falseNodeList).split(";")[0], "1") def tearDown(self): import MySQLdb Recovery.clusterList = {} dbconn = MySQLdb.connect( host = config.get("mysql", "mysql_ip"), user = config.get("mysql", "mysql_username"), passwd = config.get("mysql", "mysql_password"), db = "hass", ) db = dbconn.cursor() db.execute("DELETE FROM ha_node;") db.execute("DELETE FROM ha_cluster;")
class RecoveryDeleteClusterTest(unittest.TestCase): def setUp(self): Recovery.clusterList = {} self.recovery = Recovery() self.clusterName = "TestCluster-%s" % datetime.datetime.now().strftime("%Y/%m/%d-%H:%M") self.uuid = self.recovery.createCluster(self.clusterName).split(";")[1] def test_deleteCluster_Sucess(self): self.assertEqual(self.recovery.deleteCluster(self.uuid).split(";")[0], "0") def test_deleteCluster_Exception(self): self.assertEqual(self.recovery.deleteCluster("123456789").split(";")[0], "1")
def setUp(self): Recovery.clusterList = {} self.recovery = Recovery() self.trueNodeList = ["compute1", "compute2"] self.falseNodeList = ["compute426", "compute321"] self.clusterName = "TestCluster-%s" % datetime.datetime.now().strftime("%Y/%m/%d-%H:%M") self.uuid = self.recovery.createCluster(self.clusterName).split(";")[1]
def setUp(self): Recovery.clusterList = {} self.recovery = Recovery() self.clusterName = "TestCluster-%s" % datetime.datetime.now().strftime("%Y/%m/%d-%H:%M") self.uuid = self.recovery.createCluster(self.clusterName).split(";")[1]