def test_createVMServerBootUpErrorOutput(self): result = CommandsHandler.createVMServerRegistrationErrorOutput( "ServerName", "Message") expectedResult = (COMMAND_OUTPUT_TYPE.VM_SERVER_REGISTRATION_ERROR, "ServerName$Message") self.assertEquals(result, expectedResult, "createVMServerBootUpErrorOutput does not work")
def test_createVMServerRegistrationCommand(self): result = CommandsHandler.createVMServerRegistrationCommand( "VMServerIP", 1, "VMServerName") expectedResult = (COMMAND_TYPE.REGISTER_VM_SERVER, "VMServerIP$1$VMServerName") self.assertEquals(result, expectedResult, "createVMServerRegistrationCommand does not work")
def test_commandOutputDeserialization(self): outputs = ["ServerName$Error", "1$1$Error", "1$IP$1$Pass"] outputTypes = [ COMMAND_OUTPUT_TYPE.VM_SERVER_BOOTUP_ERROR, COMMAND_OUTPUT_TYPE.VM_BOOT_FAILURE, COMMAND_OUTPUT_TYPE.VM_CONNECTION_DATA ] expectedResults = [{ "ServerNameOrIPAddress": "ServerName", "ErrorMessage": "Error" }, { "VMID": 1, "UserID": 1, "ErrorMessage": "Error" }, { "UserID": 1, "VNCServerIPAddress": "IP", "VNCServerPort": 1, "VNCServerPassword": "******" }] i = 0 while (i < len(outputs)): result = CommandsHandler.deserializeCommandOutput( outputTypes[i], outputs[i]) self.assertEquals(result, expectedResults[i], "deserializeCommandOutput does not work") i += 1
def test_commandArgumentsDeserialization(self): arguments = [ "VMServerIP$1$VMServerName", "True$VMServerName$True", "VMServerName", "1$1", "True" ] commandTypes = [ COMMAND_TYPE.REGISTER_VM_SERVER, COMMAND_TYPE.UNREGISTER_OR_SHUTDOWN_VM_SERVER, COMMAND_TYPE.BOOTUP_VM_SERVER, COMMAND_TYPE.VM_BOOT_REQUEST, COMMAND_TYPE.HALT ] expectedResults = [{ "VMServerIP": "VMServerIP", "VMServerPort": 1, "VMServerName": "VMServerName" }, { "Unregister": True, "VMServerNameOrIP": "VMServerName", "Halt": True }, { "VMServerNameOrIP": "VMServerName" }, { "VMID": 1, "UserID": 1 }, { "HaltVMServers": True }] i = 0 while (i < len(arguments)): result = CommandsHandler.deserializeCommandArgs( commandTypes[i], arguments[i]) self.assertEquals(result, expectedResults[i], "deserializeCommandArgs does not work") i += 1
def test_createVMConnectionDataOutput(self): result = CommandsHandler.createVMConnectionDataOutput( 1, "IP", 1, "pass") expectedResult = (COMMAND_OUTPUT_TYPE.VM_CONNECTION_DATA, "1$IP$1$pass") self.assertEquals(result, expectedResult, "createVMConnectionDataOutput does not work")
def test_createVMServerUnregistrationOrShutdownCommand(self): result = CommandsHandler.createVMServerUnregistrationOrShutdownCommand( True, "VMServerIP", True) expectedResult = (COMMAND_TYPE.UNREGISTER_OR_SHUTDOWN_VM_SERVER, "True$VMServerIP$True") self.assertEquals( result, expectedResult, "createVMServerUnregistrationOrShutdownCommand does not work")
def test_commandOutputDeserialization(self): outputs = ["ServerName$Error", "1$1$Error", "1$IP$1$Pass"] outputTypes = [COMMAND_OUTPUT_TYPE.VM_SERVER_BOOTUP_ERROR, COMMAND_OUTPUT_TYPE.VM_BOOT_FAILURE, COMMAND_OUTPUT_TYPE.VM_CONNECTION_DATA] expectedResults = [{"ServerNameOrIPAddress": "ServerName", "ErrorMessage" : "Error"}, {"VMID" : 1, "UserID" : 1, "ErrorMessage" : "Error"}, {"UserID" : 1, "VNCServerIPAddress" : "IP", "VNCServerPort" : 1, "VNCServerPassword" : "Pass"}] i = 0 while (i < len(outputs)) : result = CommandsHandler.deserializeCommandOutput(outputTypes[i], outputs[i]) self.assertEquals(result, expectedResults[i], "deserializeCommandOutput does not work") i += 1
def test_commandArgumentsDeserialization(self): arguments = ["VMServerIP$1$VMServerName", "True$VMServerName$True", "VMServerName", "1$1", "True"] commandTypes = [COMMAND_TYPE.REGISTER_VM_SERVER, COMMAND_TYPE.UNREGISTER_OR_SHUTDOWN_VM_SERVER, COMMAND_TYPE.BOOTUP_VM_SERVER, COMMAND_TYPE.VM_BOOT_REQUEST, COMMAND_TYPE.HALT] expectedResults = [{"VMServerIP":"VMServerIP", "VMServerPort" : 1, "VMServerName" : "VMServerName"}, {"Unregister":True, "VMServerNameOrIP":"VMServerName", "Halt":True}, {"VMServerNameOrIP" : "VMServerName"}, {"VMID" : 1, "UserID" : 1}, {"HaltVMServers" : True}] i = 0 while (i < len(arguments)) : result = CommandsHandler.deserializeCommandArgs(commandTypes[i], arguments[i]) self.assertEquals(result, expectedResults[i], "deserializeCommandArgs does not work") i += 1
def test_createVMServerRegistrationCommand(self): result = CommandsHandler.createVMServerRegistrationCommand("VMServerIP", 1, "VMServerName") expectedResult = (COMMAND_TYPE.REGISTER_VM_SERVER, "VMServerIP$1$VMServerName") self.assertEquals(result, expectedResult, "createVMServerRegistrationCommand does not work")
def test_createVMConnectionDataOutput(self): result = CommandsHandler.createVMConnectionDataOutput(1, "IP", 1, "pass") expectedResult = (COMMAND_OUTPUT_TYPE.VM_CONNECTION_DATA, "1$IP$1$pass") self.assertEquals(result, expectedResult, "createVMConnectionDataOutput does not work")
def test_createVMBootFailureErrorOutput(self): result = CommandsHandler.createVMBootFailureErrorOutput(1, 1, "Message") expectedResult = (COMMAND_OUTPUT_TYPE.VM_BOOT_FAILURE, "1$1$Message") self.assertEquals(result, expectedResult, "createVMBootFailureErrorOutput does not work")
def test_createVMServerRegistrationErrorOutput(self): result = CommandsHandler.createVMServerRegistrationErrorOutput("ServerName", "Message") expectedResult = (COMMAND_OUTPUT_TYPE.VM_SERVER_REGISTRATION_ERROR, "ServerName$Message") self.assertEquals(result, expectedResult, "createVMServerRegistrationErrorOutput does not work")
def test_createHaltCommand(self): result = CommandsHandler.createHaltCommand(True) expectedResult = (COMMAND_TYPE.HALT, "True") self.assertEquals(result, expectedResult, "createHaltCommand does not work")
def test_createVMServerBootCommand(self): result = CommandsHandler.createVMServerBootCommand("Server1") expectedResult = (COMMAND_TYPE.BOOTUP_VM_SERVER, "Server1") self.assertEquals(result, expectedResult, "createVMServerBootCommand does not work")
def test_createVMBootCommand(self): result = CommandsHandler.createVMBootCommand(1, 1) expectedResult = (COMMAND_TYPE.VM_BOOT_REQUEST, "1$1") self.assertEquals(result, expectedResult, "createVMBootCommand does not work")
def test_createVMServerUnregistrationOrShutdownCommand(self): result = CommandsHandler.createVMServerUnregistrationOrShutdownCommand(True, "VMServerIP", True) expectedResult = (COMMAND_TYPE.UNREGISTER_OR_SHUTDOWN_VM_SERVER, "True$VMServerIP$True") self.assertEquals(result, expectedResult, "createVMServerUnregistrationOrShutdownCommand does not work")
def test_createVMBootFailureErrorOutput(self): result = CommandsHandler.createVMBootFailureErrorOutput( 1, 1, "Message") expectedResult = (COMMAND_OUTPUT_TYPE.VM_BOOT_FAILURE, "1$1$Message") self.assertEquals(result, expectedResult, "createVMBootFailureErrorOutput does not work")