def test_instruction_invalid_arg_separator(self): """ Instruction with invalid arg separator. """ instruction_str = '3.args8.hostname,4.port;' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_arg_separator(self): """ Instruction with invalid arg separator. """ instruction_str = '3.args8.hostname,4.port;' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_element_separator(self): """ Instruction with invalid element separator. """ instruction_str = '3args,8.hostname,4.port;' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_termination(self): """ Instruction with invalid terminator. """ instruction_str = '4.args,8.hostname,4.port' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_arg_length_large(self): """ Instruction with invalid large arg length. """ instruction_str = '1000.args,8.hostname,4.port;' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_element_separator(self): """ Instruction with invalid element separator. """ instruction_str = '3args,8.hostname,4.port;' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_arg_length_large(self): """ Instruction with invalid large arg length. """ instruction_str = '1000.args,8.hostname,4.port;' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_invalid_termination(self): """ Instruction with invalid terminator. """ instruction_str = '4.args,8.hostname,4.port' with self.assertRaises(InvalidInstruction): Instruction.load(instruction_str)
def test_instruction_valid_encode(self): """ Test valid instruction encoding. """ instruction_str = '4.args,8.hostname,4.port,4.1984;' instruction_opcode = 'args' instruction_args = ('hostname', 'port', 1984) instruction = Instruction('args', 'hostname', 'port', 1984) self.assertEqual(instruction_str, instruction.encode()) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args)
def test_instruction_valid_encode(self): """ Test valid instruction encoding. """ instruction_str = '4.args,8.hostname,4.port,4.1984;' instruction_opcode = 'args' instruction_args = ('hostname', 'port', 1984) instruction = Instruction('args', 'hostname', 'port', 1984) self.assertEqual(instruction_str, instruction.encode()) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args)
def test_instruction_valid_encode_with_protocol_chars(self): """ Test valid instruction encoding with arg containing protocol characters. """ # arg includes ARG_SEP, ELEM_SEP, INST_TERM and a white space. arg_protocol_chars = 'p,.; t' instruction_str = '4.args,8.hostname,%s.%s;' %\ (len(arg_protocol_chars), arg_protocol_chars) instruction_opcode = 'args' instruction_args = ('hostname', arg_protocol_chars) instruction = Instruction('args', 'hostname', arg_protocol_chars) self.assertEqual(instruction_str, instruction.encode()) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args)
async def accept_and_send_error(self, error_text, error_code): """ Handle errors during connect with a reason and code In order for client side guacamole client to receive error code/text we need to accept connection and immediately send error instruction """ await self.accept(subprotocol='guacamole') await self.send(text_data=GuacamoleInstruction("error", error_text, error_code).encode())
def test_instruction_valid_encode_with_protocol_chars(self): """ Test valid instruction encoding with arg containing protocol characters. """ # arg includes ARG_SEP, ELEM_SEP, INST_TERM and a white space. arg_protocol_chars = 'p,.; t' instruction_str = '4.args,8.hostname,%s.%s;' %\ (len(arg_protocol_chars), arg_protocol_chars) instruction_opcode = 'args' instruction_args = ('hostname', arg_protocol_chars) instruction = Instruction('args', 'hostname', arg_protocol_chars) self.assertEqual(instruction_str, instruction.encode()) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args)
def test_instruction_valid_encode_unicode(self): """ Test valid instruction encoding with unicode characters. """ # instruction str for validation! instruction_str = '4.args,8.hostname,%s.%s;' %\ (self.u_arg_len, self.u_arg_utf8) instruction_opcode = 'args' instruction_args = ('hostname', self.u_arg) # passing a unicode arg! instruction = Instruction('args', 'hostname', self.u_arg) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args) # all args should be utf_8 after instruction.encode() self.assertEqual(instruction_str, instruction.encode())
def test_instruction_valid_encode_unicode(self): """ Test valid instruction encoding with unicode characters. """ # instruction str for validation! instruction_str = '4.args,8.hostname,%s.%s;' %\ (self.u_arg_len, self.u_arg_utf8) instruction_opcode = 'args' instruction_args = ('hostname', self.u_arg) # passing a unicode arg! instruction = Instruction('args', 'hostname', self.u_arg) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args) # all args should be utf_8 after instruction.encode() self.assertEqual(instruction_str, instruction.encode())
def test_instruction_valid_decode_unicode(self): """ Test valid instruction decoding with unicode characters. """ # messing up by passing unicode instruction (with valid arg length!) instruction_str_u = u'4.args,8.hostname,%s.%s;' %\ (self.u_arg_len, self.u_arg) # instruction str for validation! instruction_str = '4.args,8.hostname,%s.%s;' %\ (self.u_arg_len, self.u_arg_utf8) instruction_opcode = 'args' instruction_args = ('hostname', self.u_arg_utf8) # Instruction.load should handle unicode string! instruction = Instruction.load(instruction_str_u) self.assertEqual(instruction_str, instruction.encode()) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args)
def test_instruction_valid_decode_unicode(self): """ Test valid instruction decoding with unicode characters. """ # messing up by passing unicode instruction (with valid arg length!) instruction_str_u = u'4.args,8.hostname,%s.%s;' %\ (self.u_arg_len, self.u_arg) # instruction str for validation! instruction_str = '4.args,8.hostname,%s.%s;' %\ (self.u_arg_len, self.u_arg_utf8) instruction_opcode = 'args' instruction_args = ('hostname', self.u_arg_utf8) # Instruction.load should handle unicode string! instruction = Instruction.load(instruction_str_u) self.assertEqual(instruction_str, instruction.encode()) self.assertEqual(instruction_opcode, instruction.opcode) self.assertEqual(instruction_args, instruction.args)
def read_instruction(self): """ Read and decode instruction. """ self.logger.debug('Reading instruction.') return Instruction.load(self.receive())
def read_instruction(self): """ Read and decode instruction. """ self.logger.debug('Reading instruction.') return Instruction.load(self.receive())