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"')
