hurui200320
bde74c82b5
This MR fixed the pipeline so it can run on forgejo runner. Currently we have two pipelines set up: + One pipeline will run pytest to test the code and collect coverage data, this pipeline will run on all branches and PRs. + One pipeline will build the wheel file and publish it to the company's pypl repo, this pipeline will only run on the master branch and develop branch. This RP also made the following changes to make the pipeline working: + Remove `setup.py` and `requirements.txt`, now the project is fully embraced `pyproject.toml`, which according to ChatGPT, is the most modern way of managing a python project. + Move all test files to `tests` folder, so it won't be packed into the final wheel file and affect the coverage rate. + Change version to `X.Y.Z-dev`, before building the wheel file, the pipeline will automatically add a UTC timestamp after it, so it becomes `X.Y.Z-devYYYYMMDDHHmmSS`. This is added due to the limitation of the pypl repo, unlike maven snapshot versions, you can't push the same version twice with our pypl repository. Reviewed-on: #10 Reviewed-by: Stanislav Hejny <stanislav.hejny@cleverthis.com> Co-authored-by: Rui Hu <rui.hu@cleverthis.com> Co-committed-by: Rui Hu <rui.hu@cleverthis.com>
130 lines
4.9 KiB
Python
130 lines
4.9 KiB
Python
from datetime import datetime, timezone
|
|
from unittest import TestCase
|
|
|
|
from amqp.adapter.data_message_factory import DataMessageFactory
|
|
|
|
|
|
class DataMessageFactoryTest(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 = DataMessageFactory.from_bytes(raw.encode("utf-8"))
|
|
self.assertIsNotNone(msg.id)
|
|
self.assertEqual(
|
|
msg.id.as_string(),
|
|
"64F3B5AF4D00000A4000".lower(),
|
|
"SnowflakeID incorrectly parsed.",
|
|
)
|
|
self.assertEqual(
|
|
msg.headers["Content-Type"][0],
|
|
"multipart/form-data; boundary=------------------------j1RjIqLrFH07XMsyMvpS1v",
|
|
)
|
|
|
|
def test_generate_snowflake_id(self):
|
|
_id = DataMessageFactory.get_instance(1).generate_snowflake_id()
|
|
self.assertEqual(
|
|
"00001000", _id.as_string()[-8:], "Instance ID not set correctly"
|
|
)
|
|
|
|
_id = DataMessageFactory(2).generate_snowflake_id()
|
|
self.assertEqual(
|
|
"00002000", _id.as_string()[-8:], "Instance ID not set correctly"
|
|
)
|
|
|
|
_f = DataMessageFactory(127)
|
|
_i0 = _f.generate_snowflake_id()
|
|
_i1 = _f.generate_snowflake_id()
|
|
_i2 = _f.generate_snowflake_id()
|
|
_i3 = _f.generate_snowflake_id()
|
|
self.assertEqual(
|
|
"0007F000".lower(),
|
|
_i0.as_string()[-8:],
|
|
"Instance ID and sequence not set correctly",
|
|
)
|
|
self.assertEqual(
|
|
"0007F001".lower(),
|
|
_i1.as_string()[-8:],
|
|
"Instance ID and sequence not set correctly",
|
|
)
|
|
self.assertEqual(
|
|
"0007F002".lower(),
|
|
_i2.as_string()[-8:],
|
|
"Instance ID and sequence not set correctly",
|
|
)
|
|
self.assertEqual(
|
|
"0007F003".lower(),
|
|
_i3.as_string()[-8:],
|
|
"Instance ID and sequence not set correctly",
|
|
)
|
|
|
|
def test_create_request_message(self):
|
|
_f = DataMessageFactory(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 = DataMessageFactory(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 = DataMessageFactory(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 = DataMessageFactory(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 = DataMessageFactory(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"
|
|
)
|