Stanislav Hejny
68 lines
4.1 KiB
Python
68 lines
4.1 KiB
Python
from datetime import datetime, timezone
|
|
from unittest import TestCase
|
|
|
|
from amqp.adapter.amq_message_factory import AMQMessageFactory
|
|
|
|
|
|
class AMQMessageFactoryTest(TestCase):
|
|
def test_from_bytes(self):
|
|
raw = "\002\027A64f3b5af4d00000a4000|m=POST|d=clever-pybook.localhost|p=/debug|h={Accept=[*/*],X-Forwarded-Host=[clever-pybook.localhost:8085],User-Agent=[curl/8.7.1],X-Forwarded-Proto=[http],X-Forwarded-For=[172.18.0.1],Host=[clever-pybook.localhost:8085],Accept-Encoding=[gzip],Content-Length=[1475],X-Forwarded-Port=[8085],X-Real-Ip=[172.18.0.1],X-Forwarded-Server=[c5945bef67d1],Content-Type=[multipart/form-data; boundary=------------------------j1RjIqLrFH07XMsyMvpS1v]}|t={traceparent=00-bed6eb7de0d8ccf4c80b145a4bfec357-8d1f3bafc471bf0f-01}|b=TEST_ME"
|
|
msg = AMQMessageFactory.from_bytes(raw.encode("utf-8"))
|
|
self.assertIsNotNone(msg.id)
|
|
self.assertEqual(msg.id.as_string(),'64F3B5AF4D00000A4000','SnowflakeID incorrectly parsed.')
|
|
self.assertEqual(msg.headers["Content-Type"][0], "multipart/form-data; boundary=------------------------j1RjIqLrFH07XMsyMvpS1v")
|
|
|
|
def test_generate_snowflake_id(self):
|
|
_id = AMQMessageFactory.get_instance(1).generate_snowflake_id()
|
|
self.assertEqual('00001000', _id.as_string()[-8:], 'Instance ID not set correctly')
|
|
|
|
_id = AMQMessageFactory(2).generate_snowflake_id()
|
|
self.assertEqual('00002000', _id.as_string()[-8:], 'Instance ID not set correctly')
|
|
|
|
_f = AMQMessageFactory(127)
|
|
_i0 = _f.generate_snowflake_id()
|
|
_i1 = _f.generate_snowflake_id()
|
|
_i2 = _f.generate_snowflake_id()
|
|
_i3 = _f.generate_snowflake_id()
|
|
self.assertEqual('0007F000', _i0.as_string()[-8:], 'Instance ID and sequence not set correctly')
|
|
self.assertEqual('0007F001', _i1.as_string()[-8:], 'Instance ID and sequence not set correctly')
|
|
self.assertEqual('0007F002', _i2.as_string()[-8:], 'Instance ID and sequence not set correctly')
|
|
self.assertEqual('0007F003', _i3.as_string()[-8:], 'Instance ID and sequence not set correctly')
|
|
|
|
def test_create_request_message(self):
|
|
_f = AMQMessageFactory(34)
|
|
_req = _f.create_request_message("PUT", "localhost", "/test/me", {"k1": ["v1"], "k2": ["v2"]}, "Quick brown fox")
|
|
self.assertEqual("localhost", _req.domain, 'Domain not set correctly')
|
|
self.assertEqual("/test/me", _req.path, 'Context path not set correctly')
|
|
self.assertEqual("v1", _req.headers['k1'][0], "Headers not set correctly")
|
|
self.assertEqual(b'Quick brown fox', _req.body(), "Body not correctly decoded")
|
|
|
|
def test_serialize(self):
|
|
_f = AMQMessageFactory(34)
|
|
_req = _f.create_request_message("PUT", "localhost", "/test/me", {"k1": ["v1"], "k2": ["v2"]}, "Quick brown fox")
|
|
_ser = _f.serialize(_req)
|
|
self.assertEqual(65, _ser[2])
|
|
_deser = _f.from_bytes(_ser)
|
|
self.assertEqual(_req.id.as_string(), _deser.id.as_string())
|
|
self.assertEqual("localhost", _deser.domain)
|
|
self.assertEqual("/test/me", _deser.path)
|
|
self.assertEqual(b"Quick brown fox", _deser.body())
|
|
self.assertEqual("v2", _deser.headers["k2"][0])
|
|
|
|
def test_amq_message_routing_key(self):
|
|
_f = AMQMessageFactory(35)
|
|
_req = _f.create_request_message("PUT", "localhost", "/test/me", {"k1": ["v1"], "k2": ["v2"]}, "Quick brown fox")
|
|
_key = _f.amq_message_routing_key(_req)
|
|
self.assertEqual("localhost.test.me", _key, "Message routing key wrongly extracted from the message")
|
|
|
|
def test_get_timestamp_from_id(self):
|
|
_f = AMQMessageFactory(35)
|
|
_timestamp = _f.get_timestamp_from_id(_f.generate_snowflake_id())
|
|
_x_timestamp = int(datetime.now(timezone.utc).timestamp() * 1000)
|
|
self.assertLessEqual(_x_timestamp - _timestamp, 1, "Timestamp in SnowflakeId differs too much from current time")
|
|
|
|
def test_to_string(self):
|
|
_f = AMQMessageFactory(35)
|
|
_req = _f.create_request_message("PUT", "localhost", "/test/me", {"k1": ["v1"], "k2": ["v2"]}, "Quick brown fox")
|
|
_as_str = _f.to_string(_req)
|
|
self.assertGreater(_as_str.index("domain='localhost'"), 0, "Expected content not found") |