#23 refactor to use shared code between services
This commit is contained in:
@@ -2,6 +2,7 @@ from unittest import TestCase
|
||||
|
||||
from amqp.adapter.data_message_factory import DataMessageFactory
|
||||
from amqp.adapter.data_response_factory import DataResponseFactory
|
||||
from amqp.model.model import DataMessageMagicByte
|
||||
from amqp.model.snowflake_id import SnowflakeId
|
||||
|
||||
|
||||
@@ -12,7 +13,7 @@ class TestDataResponseFactory(TestCase):
|
||||
_g = DataMessageFactory(111)
|
||||
_resp = _f.create_async_response_message(_g.generate_snowflake_id())
|
||||
self.assertEqual(200, _resp.response_code)
|
||||
self.assertEqual(b"OK", _resp.response)
|
||||
self.assertEqual(b"OK", _resp.base64body())
|
||||
|
||||
def test_serialize(self):
|
||||
_f = DataResponseFactory()
|
||||
@@ -22,7 +23,7 @@ class TestDataResponseFactory(TestCase):
|
||||
_ser = _f.serialize(_resp)
|
||||
self.assertEqual(
|
||||
2,
|
||||
str(_ser[2:]).index(_f.MAGIC_BYTE_HTTP_REQUEST + _snowflakeId.as_string()),
|
||||
str(_ser[2:]).index(DataMessageMagicByte.HTTP_REQUEST.value + _snowflakeId.as_string()),
|
||||
"Magic byte and SnowflakeId not found",
|
||||
)
|
||||
self.assertGreater(str(_ser[2:]).index("|e=|"), 2)
|
||||
@@ -37,4 +38,4 @@ class TestDataResponseFactory(TestCase):
|
||||
_deser = _f.from_bytes(_ser)
|
||||
self.assertEqual(_snowflakeId.as_string(), _deser.id)
|
||||
self.assertEqual(200, _deser.response_code)
|
||||
self.assertEqual(b"OK", _deser.response)
|
||||
self.assertEqual(b"OK", _deser.base64body())
|
||||
|
||||
Reference in New Issue
Block a user