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