class ClusterServerDBTests(unittest.TestCase):
'''
This class runs the unit tests.
'''
def setUp(self):
'''
This method will be ran before EVERY unit test.
'''
# Create the database
dbConfigurator = DBConfigurator("")
dbConfigurator.runSQLScript("ClusterServerDBTest",
"./ClusterServerDBTest.sql")
# Add a user to it
dbConfigurator.addUser("cygnuscloud", "cygnuscloud",
"ClusterServerDBTest")
self.__connector = ClusterServerDatabaseConnector(
"cygnuscloud", "cygnuscloud", "ClusterServerDBTest")
self.__connector.connect()
def tearDown(self):
'''
This method will be ran after EVERY unit test.
'''
self.__connector.disconnect()
dbConfigurator = DBConfigurator("")
dbConfigurator.dropDatabase("ClusterServerDBTest")
def test_getBasicServerData(self):
'''
Retrieves basic information about a virtual machine server
'''
result = self.__connector.getVMServerBasicData(1)
expectedResult = {
'ServerName': 'Server1',
'ServerStatus': SERVER_STATE_T.BOOTING,
'ServerIP': '1.2.3.4',
'ServerPort': 8080
}
self.assertEquals(result, expectedResult,
'getVMServerBasicData does not work')
def test_getVMServerStatistics(self):
'''
Determines how many virtual machines are running on a virtual machine server
'''
result = self.__connector.getVMServerStatistics(100)
expectedResult = None
self.assertEquals(result, expectedResult,
'getVMServerStatistics does not work')
result = self.__connector.getVMServerStatistics(2)
expectedResult = {'ActiveHosts': 10}
self.assertEquals(result, expectedResult,
'getVMServerStatistics does not work')
def test_getVMServerIDs(self):
'''
Obtains the registerd virtual machine servers' IDs
'''
result = self.__connector.getVMServerIDs()
expectedResult = [1, 2, 3, 4]
self.assertEquals(result, expectedResult,
'getVMServerIDs does not work')
def test_registerVMServer(self):
'''
Tests the registration of a new virtual machine server
'''
self.__connector.registerVMServer('A new server', '100.101.102.103',
89221)
ids = self.__connector.getVMServerIDs()
expectedIds = [1, 2, 3, 4, ids[len(ids) - 1]]
self.assertEquals(ids, expectedIds, 'registerVMServer does not work')
def test_unregisterVMServer(self):
'''
Tests the deletion of a virtual machine server
'''
self.__connector.deleteVMServer('Server1')
self.__connector.deleteVMServer('1.2.3.5')
ids = self.__connector.getVMServerIDs()
expectedIds = [3, 4]
self.assertEquals(ids, expectedIds, 'unregisterVMServer does not work')
def test_getAvailableVMServers(self):
'''
Tries to retrieve the virtual machine servers that can host an image.
'''
result = self.__connector.getImageServers(1)
expectedResult = [2, 3]
self.assertEquals(result, expectedResult,
'getAvailableVMServers does not work')
def test_updateVMServerStatus(self):
'''
Updates a virtual machine server's status
'''
self.__connector.updateVMServerStatus(1, SERVER_STATE_T.READY)
d = self.__connector.getVMServerBasicData(1)
self.assertEquals(d['ServerStatus'], SERVER_STATE_T.READY,
"updateVMServerStatus does not work")
def test_getImages(self):
'''
Tries to retrieve the images that can be hosted on a virtual machine server
'''
result = self.__connector.getServerImages(1)
expectedResult = [1, 2, 3]
self.assertEquals(result, expectedResult, 'getImages does not work')
def test_assignImageToServer(self):
'''
Tries to assign an image to a virtual machine server
'''
self.__connector.assignImageToServer(4, 2)
result = self.__connector.getImageServers(2)
expectedResult = [2, 4]
self.assertEquals(result, expectedResult,
'assignImageToServer does not work')
def test_setServerBasicData(self):
'''
Tries to modify a virtual machine server's data
'''
self.__connector.setServerBasicData(1, 'Foo', SERVER_STATE_T.BOOTING,
'192.168.1.1', 9000)
result = self.__connector.getVMServerBasicData(1)
expectedResult = {
'ServerName': 'Foo',
'ServerStatus': SERVER_STATE_T.BOOTING,
'ServerIP': '192.168.1.1',
'ServerPort': 9000
}
self.assertEquals(result, expectedResult,
'setServerBasicData does not work')
def test_getServerID(self):
result = self.__connector.getVMServerID("1.2.3.4")
expectedResult = 1L
self.assertEquals(result, expectedResult, 'getServerID does not work')
def test_getActiveVMServersConnectionData(self):
result = self.__connector.getActiveVMServersConnectionData()
expectedResult = [{
"ServerIP": '1.2.3.5',
"ServerPort": 8080
}, {
"ServerIP": '1.2.3.6',
"ServerPort": 8080
}, {
"ServerIP": '1.2.3.7',
"ServerPort": 8080
}]
self.assertEquals(result, expectedResult,
'getActiveVMServersConnectionData does not work')
def test_setVMServerStatistics(self):
self.__connector.setVMServerStatistics(1, 1234)
result = self.__connector.getVMServerStatistics(1)
expectedResult = {'ActiveHosts': 1234}
self.assertEquals(result, expectedResult,
'setVMServerStatistics does not work')
def test_vmBootCommands(self):
result = self.__connector.getOldVMBootCommandID(1)
self.assertEquals(result, None, 'getOldVMBootCommand does not work')
self.__connector.registerVMBootCommand("Command1", 1)
sleep(2)
result = self.__connector.getOldVMBootCommandID(1)
self.assertEquals(result, ("Command1", 1),
'getOldVMBootCommand does not work')
self.__connector.registerVMBootCommand("Command1", 2)
sleep(1)
result = self.__connector.getOldVMBootCommandID(3)
self.assertEquals(result, None, 'getOldVMBootCommand does not work')
self.__connector.registerVMBootCommand("Command2", 2)
self.__connector.registerVMBootCommand("Command3", 3)
self.__connector.removeVMBootCommand("Command2")
sleep(5)
result = self.__connector.getOldVMBootCommandID(3)
self.assertEquals(result, ("Command1", 2),
'getOldVMBootCommand does not work')
result = self.__connector.getOldVMBootCommandID(3)
self.assertEquals(result, ("Command3", 3),
'getOldVMBootCommand does not work')
class ClusterServerDBTests(unittest.TestCase):
'''
This class runs the unit tests.
'''
def setUp(self):
'''
This method will be ran before EVERY unit test.
'''
# Create the database
dbConfigurator = DBConfigurator("")
dbConfigurator.runSQLScript("ClusterServerDBTest", "./ClusterServerDBTest.sql")
# Add a user to it
dbConfigurator.addUser("cygnuscloud", "cygnuscloud", "ClusterServerDBTest")
self.__connector = ClusterServerDatabaseConnector("cygnuscloud", "cygnuscloud", "ClusterServerDBTest")
self.__connector.connect()
def tearDown(self):
'''
This method will be ran after EVERY unit test.
'''
self.__connector.disconnect()
dbConfigurator = DBConfigurator("")
dbConfigurator.dropDatabase("ClusterServerDBTest")
def test_getBasicServerData(self):
'''
Retrieves basic information about a virtual machine server
'''
result = self.__connector.getVMServerBasicData(1)
expectedResult = {'ServerName':'Server1', 'ServerStatus':SERVER_STATE_T.BOOTING,
'ServerIP':'1.2.3.4', 'ServerPort':8080}
self.assertEquals(result, expectedResult, 'getVMServerBasicData does not work')
def test_getVMServerStatistics(self):
'''
Determines how many virtual machines are running on a virtual machine server
'''
result = self.__connector.getVMServerStatistics(100)
expectedResult = None
self.assertEquals(result, expectedResult, 'getVMServerStatistics does not work')
result = self.__connector.getVMServerStatistics(2)
expectedResult = {'ActiveHosts': 10}
self.assertEquals(result, expectedResult, 'getVMServerStatistics does not work')
def test_getVMServerIDs(self):
'''
Obtains the registerd virtual machine servers' IDs
'''
result = self.__connector.getVMServerIDs()
expectedResult = [1,2,3,4]
self.assertEquals(result, expectedResult, 'getVMServerIDs does not work')
def test_registerVMServer(self):
'''
Tests the registration of a new virtual machine server
'''
self.__connector.registerVMServer('A new server', '100.101.102.103', 89221)
ids = self.__connector.getVMServerIDs()
expectedIds = [1,2,3,4, ids[len(ids) - 1]]
self.assertEquals(ids, expectedIds, 'registerVMServer does not work')
def test_unregisterVMServer(self):
'''
Tests the deletion of a virtual machine server
'''
self.__connector.deleteVMServer('Server1')
self.__connector.deleteVMServer('1.2.3.5')
ids = self.__connector.getVMServerIDs()
expectedIds = [3,4]
self.assertEquals(ids, expectedIds, 'unregisterVMServer does not work')
def test_getAvailableVMServers(self):
'''
Tries to retrieve the virtual machine servers that can host an image.
'''
result = self.__connector.getImageServers(1)
expectedResult = [2,3]
self.assertEquals(result, expectedResult, 'getAvailableVMServers does not work')
def test_updateVMServerStatus(self):
'''
Updates a virtual machine server's status
'''
self.__connector.updateVMServerStatus(1, SERVER_STATE_T.READY)
d = self.__connector.getVMServerBasicData(1)
self.assertEquals(d['ServerStatus'], SERVER_STATE_T.READY, "updateVMServerStatus does not work")
def test_getImages(self):
'''
Tries to retrieve the images that can be hosted on a virtual machine server
'''
result = self.__connector.getServerImages(1)
expectedResult = [1,2,3]
self.assertEquals(result, expectedResult, 'getImages does not work')
def test_assignImageToServer(self):
'''
Tries to assign an image to a virtual machine server
'''
self.__connector.assignImageToServer(4, 2)
result = self.__connector.getImageServers(2)
expectedResult = [2,4]
self.assertEquals(result, expectedResult, 'assignImageToServer does not work')
def test_setServerBasicData(self):
'''
Tries to modify a virtual machine server's data
'''
self.__connector.setServerBasicData(1, 'Foo', SERVER_STATE_T.BOOTING, '192.168.1.1', 9000)
result = self.__connector.getVMServerBasicData(1)
expectedResult = {'ServerName':'Foo', 'ServerStatus':SERVER_STATE_T.BOOTING,
'ServerIP':'192.168.1.1', 'ServerPort':9000}
self.assertEquals(result, expectedResult, 'setServerBasicData does not work')
def test_getServerID(self):
result = self.__connector.getVMServerID("1.2.3.4")
expectedResult = 1L
self.assertEquals(result, expectedResult, 'getServerID does not work')
def test_getActiveVMServersConnectionData(self):
result = self.__connector.getActiveVMServersConnectionData()
expectedResult = [{"ServerIP" : '1.2.3.5', "ServerPort" : 8080},
{"ServerIP" : '1.2.3.6', "ServerPort" : 8080},
{"ServerIP" : '1.2.3.7', "ServerPort" : 8080}]
self.assertEquals(result, expectedResult, 'getActiveVMServersConnectionData does not work')
def test_setVMServerStatistics(self):
self.__connector.setVMServerStatistics(1, 1234)
result = self.__connector.getVMServerStatistics(1)
expectedResult = {'ActiveHosts': 1234}
self.assertEquals(result, expectedResult, 'setVMServerStatistics does not work')
def test_vmBootCommands(self):
result = self.__connector.getOldVMBootCommandID(1)
self.assertEquals(result, None, 'getOldVMBootCommand does not work')
self.__connector.registerVMBootCommand("Command1", 1)
sleep(2)
result = self.__connector.getOldVMBootCommandID(1)
self.assertEquals(result, ("Command1", 1), 'getOldVMBootCommand does not work')
self.__connector.registerVMBootCommand("Command1", 2)
sleep(1)
result = self.__connector.getOldVMBootCommandID(3)
self.assertEquals(result, None, 'getOldVMBootCommand does not work')
self.__connector.registerVMBootCommand("Command2",2)
self.__connector.registerVMBootCommand("Command3",3)
self.__connector.removeVMBootCommand("Command2")
sleep(5)
result = self.__connector.getOldVMBootCommandID(3)
self.assertEquals(result, ("Command1",2), 'getOldVMBootCommand does not work')
result = self.__connector.getOldVMBootCommandID(3)
self.assertEquals(result, ("Command3",3), 'getOldVMBootCommand does not work')
