class Test(unittest.TestCase):
def setUp(self):
dbConfigurator = DBConfigurator("")
dbConfigurator.runSQLScript("SystemStatusDBTest",
"./SystemStatusDBTest.sql")
dbConfigurator.addUser("website", "cygnuscloud", "SystemStatusDBTest",
False)
dbConfigurator.addUser("statusDBUpdater", "cygnuscloud",
"SystemStatusDBTest", True)
self.__reader = SystemStatusDatabaseReader("website", "cygnuscloud",
"SystemStatusDBTest")
self.__reader.connect()
self.__writer = SystemStatusDatabaseWriter("statusDBUpdater",
"cygnuscloud",
"SystemStatusDBTest")
self.__writer.connect()
def tearDown(self):
self.__reader.disconnect()
self.__writer.disconnect()
dbConfigurator = DBConfigurator("")
dbConfigurator.dropDatabase("SystemStatusDBTest")
def testUpdateVMServerData(self):
segment1Data = [('Server1', 'Ready', 'IP1', 1)]
segment2Data = [('Server2', 'Booting', 'IP2', 1)]
self.__writer.processVMServerSegment(1, 2, segment1Data)
serversData = self.__reader.getVMServersData()
self.assertEquals(serversData, [],
"processVMServerSegment does not work")
self.__writer.processVMServerSegment(2, 2, segment2Data)
serversData = self.__reader.getVMServersData()
d2 = {
"VMServerName": "Server2",
"VMServerStatus": "Booting",
"VMServerIP": "IP2",
"VMServerListenningPort": 1
}
d1 = {
"VMServerName": "Server1",
"VMServerStatus": "Ready",
"VMServerIP": "IP1",
"VMServerListenningPort": 1
}
self.assertEquals(serversData, [d1, d2],
"processVMServerSegment does not work")
segment3Data = [("Server3", "Ready", "IP3", 1)]
self.__writer.processVMServerSegment(1, 1, segment3Data)
serversData = self.__reader.getVMServersData()
d3 = {
"VMServerName": "Server3",
"VMServerStatus": "Ready",
"VMServerIP": "IP3",
"VMServerListenningPort": 1
}
self.assertEquals(serversData, [d3],
"processVMServerSegment does not work")
segment4Data = [("Server3", "Ready", "IP3", 1),
("Server4", "Ready", "IP4", 1)]
self.__writer.processVMServerSegment(1, 1, segment4Data)
serversData = self.__reader.getVMServersData()
self.assertEquals(serversData, [d3],
"processVMServerSegment does not work")
def testUpdateVMDistributionData(self):
segment1Data = [('Server1', 1), ('Server1', 2), ('Server1', 3)]
segment2Data = [('Server2', 1), ('Server2', 4), ('Server2', 5)]
self.__writer.processVMDistributionSegment(1, 2, segment1Data)
serversData = self.__reader.getVMServersData()
self.assertEquals(serversData, [],
"processVMDistributionSegment does not work")
self.__writer.processVMDistributionSegment(2, 2, segment2Data)
serversData = self.__reader.getVMDistributionData()
d1 = {"VMServerName": "Server1", "VMID": 1}
d2 = {"VMServerName": "Server1", "VMID": 2}
d3 = {"VMServerName": "Server1", "VMID": 3}
d4 = {"VMServerName": "Server2", "VMID": 1}
d5 = {"VMServerName": "Server2", "VMID": 4}
d6 = {"VMServerName": "Server2", "VMID": 5}
self.assertEquals(serversData, [d1, d2, d3, d4, d5, d6],
"processVMDistributionSegment does not work")
segment3Data = [('Server4', 10)]
self.__writer.processVMDistributionSegment(1, 1, segment3Data)
serversData = self.__reader.getVMDistributionData()
d1 = {"VMServerName": "Server4", "VMID": 10}
self.assertEquals(serversData, [d1],
"processVMDistributionSegment does not work")
def testUpdateActiveVMData(self):
segment1Data = [('Server1', 'Ready', 'IP1', 1)]
self.__writer.processVMServerSegment(1, 1, segment1Data)
segment1Data = [(1, 1, 'Debian1', 15800, 'Password')]
segment2Data = [(2, 1, 'Debian1', 15802, 'Password')]
self.__writer.processActiveVMSegment(1, 2, 'IP1', segment1Data)
self.__writer.processActiveVMSegment(2, 2, 'IP1', segment2Data)
result = self.__reader.getActiveVMsData(True)
expectedResult = [{
"VMServerName": "Server1",
"UserID": 2,
"VMID": 1,
"VMName": "Debian1",
"VNCPort": 15802,
"VNCPassword": "******"
}, {
"VMServerName": "Server1",
"UserID": 1,
"VMID": 1,
"VMName": "Debian1",
"VNCPort": 15800,
"VNCPassword": "******"
}]
self.assertEquals(result, expectedResult,
"processVMServerSegment does not work")
result = self.__reader.getActiveVMsData(True)
expectedResult = [{
"VMServerName": "Server1",
"UserID": 1,
"VMID": 1,
"VMName": "Debian1",
"VNCPort": 15800,
"VNCPassword": "******"
}]
self.assertEquals(result, expectedResult,
"getActiveVMsData does not work")
class Test(unittest.TestCase):
def setUp(self):
dbConfigurator = DBConfigurator("")
dbConfigurator.runSQLScript("SystemStatusDBTest", "./SystemStatusDBTest.sql")
dbConfigurator.addUser("website", "cygnuscloud", "SystemStatusDBTest", False)
dbConfigurator.addUser("statusDBUpdater", "cygnuscloud", "SystemStatusDBTest", True)
self.__reader = SystemStatusDatabaseReader("website", "cygnuscloud", "SystemStatusDBTest")
self.__reader.connect()
self.__writer = SystemStatusDatabaseWriter("statusDBUpdater", "cygnuscloud", "SystemStatusDBTest")
self.__writer.connect()
def tearDown(self):
self.__reader.disconnect()
self.__writer.disconnect()
dbConfigurator = DBConfigurator("")
dbConfigurator.dropDatabase("SystemStatusDBTest")
def testUpdateVMServerData(self):
segment1Data = [('Server1', 'Ready', 'IP1', 1)]
segment2Data = [('Server2', 'Booting', 'IP2', 1)]
self.__writer.processVMServerSegment(1, 2, segment1Data)
serversData = self.__reader.getVMServersData()
self.assertEquals(serversData, [], "processVMServerSegment does not work")
self.__writer.processVMServerSegment(2, 2, segment2Data)
serversData = self.__reader.getVMServersData()
d2 = {"VMServerName":"Server2", "VMServerStatus":"Booting", "VMServerIP":"IP2", "VMServerListenningPort":1}
d1 = {"VMServerName":"Server1", "VMServerStatus":"Ready", "VMServerIP":"IP1", "VMServerListenningPort":1}
self.assertEquals(serversData, [d1,d2], "processVMServerSegment does not work")
segment3Data = [("Server3", "Ready", "IP3", 1)]
self.__writer.processVMServerSegment(1, 1, segment3Data)
serversData = self.__reader.getVMServersData()
d3 = {"VMServerName":"Server3", "VMServerStatus":"Ready", "VMServerIP":"IP3", "VMServerListenningPort":1}
self.assertEquals(serversData, [d3], "processVMServerSegment does not work")
segment4Data =[("Server3", "Ready", "IP3", 1), ("Server4", "Ready", "IP4", 1)]
self.__writer.processVMServerSegment(1, 1, segment4Data)
serversData = self.__reader.getVMServersData()
self.assertEquals(serversData, [d3], "processVMServerSegment does not work")
def testUpdateVMDistributionData(self):
segment1Data = [('Server1', 1), ('Server1', 2), ('Server1', 3)]
segment2Data = [('Server2', 1), ('Server2', 4), ('Server2', 5)]
self.__writer.processVMDistributionSegment(1, 2, segment1Data)
serversData = self.__reader.getVMServersData()
self.assertEquals(serversData, [], "processVMDistributionSegment does not work")
self.__writer.processVMDistributionSegment(2, 2, segment2Data)
serversData = self.__reader.getVMDistributionData()
d1 = {"VMServerName":"Server1", "VMID":1}
d2 = {"VMServerName":"Server1", "VMID":2}
d3 = {"VMServerName":"Server1", "VMID":3}
d4 = {"VMServerName":"Server2", "VMID":1}
d5 = {"VMServerName":"Server2", "VMID":4}
d6 = {"VMServerName":"Server2", "VMID":5}
self.assertEquals(serversData, [d1,d2,d3,d4,d5,d6], "processVMDistributionSegment does not work")
segment3Data = [('Server4', 10)]
self.__writer.processVMDistributionSegment(1, 1, segment3Data)
serversData = self.__reader.getVMDistributionData()
d1 = {"VMServerName":"Server4", "VMID":10}
self.assertEquals(serversData, [d1], "processVMDistributionSegment does not work")
def testUpdateActiveVMData(self):
segment1Data = [('Server1', 'Ready', 'IP1', 1)]
self.__writer.processVMServerSegment(1, 1, segment1Data)
segment1Data = [(1, 1, 'Debian1', 15800, 'Password')]
segment2Data = [(2, 1, 'Debian1', 15802, 'Password')]
self.__writer.processActiveVMSegment(1, 2, 'IP1', segment1Data)
self.__writer.processActiveVMSegment(2, 2, 'IP1', segment2Data)
result = self.__reader.getActiveVMsData(True)
expectedResult = [
{"VMServerName" : "Server1", "UserID" : 2, "VMID" : 1, "VMName" : "Debian1", "VNCPort" : 15802, "VNCPassword" : "Password" },
{"VMServerName" : "Server1", "UserID" : 1, "VMID" : 1, "VMName" : "Debian1", "VNCPort" : 15800, "VNCPassword" : "Password" }
]
self.assertEquals(result, expectedResult, "processVMServerSegment does not work")
result = self.__reader.getActiveVMsData(True)
expectedResult = [
{"VMServerName" : "Server1", "UserID" : 1, "VMID" : 1, "VMName" : "Debian1", "VNCPort" : 15800, "VNCPassword" : "Password" }
]
self.assertEquals(result, expectedResult, "getActiveVMsData does not work")