def test_large_operation(self): """ Make sure operation length errors are triggering. """ valid_operation = "#" * SingleRequestMessage.operation_max_length message = SingleRequestMessage( request_id="001", operation=valid_operation, payload="yay") # This shouldn't raise an error. write_message(message) # Now make it too big (by one). message.operation += "#" self.assertRaises(OperationTooLongError, write_message, message)
def test_large_payload(self): """ Make sure payload length errors are triggering. """ self.skipTest("See if there's a way to do this without the RAM usage.") valid_payload = "#" * SingleRequestMessage.payload_max_length message = SingleRequestMessage( request_id="001", operation="echo", payload=valid_payload) # This shouldn't raise an error. write_message(message) # Now make it too big. message.payload += "#" self.assertRaises(PayloadTooLongError, write_message, message)
def test_write(self): """ Makes sure our notification serialization is good. """ message = NotificationMessage(name="test_name", payload="Hello World") m_bytes = write_message(message) self.assertEqual(m_bytes, 'n009test_name0000000bHello World')
def test_write_version(self): """ Test the writing of a version message. """ message = ProtocolVersionMessage() m_bytes = write_message(message) self.assertEqual(m_bytes, _PROTO_VERSION)
def test_write(self): """ Makes sure our error result serialization is good. """ message = ErrorResultMessage(request_id="0001", payload="Hello World") m_bytes = write_message(message) self.assertEqual(m_bytes, 'E00010000000bHello World')
def test_write(self): """ Makes sure our stream request serialization is good. """ message = StreamRequestPartMessage(request_id="0001", payload="Hello World") m_bytes = write_message(message) self.assertEqual(m_bytes, 'p00010000000bHello World')
def test_write(self): """ Makes sure our single request serialization is good. """ message = SingleRequestMessage( request_id="001", operation="echo", payload="Hello World") m_bytes = write_message(message) self.assertEqual(m_bytes, 'r001004echo0000000bHello World')
def test_write(self): """ Makes sure our error result serialization is good. """ message = RetryResultMessage(request_id="0001", wait=0, payload='"service restarting"') m_bytes = write_message(message) self.assertEqual(m_bytes, 'e00010000000000000014"service restarting"')