def test_smart_kafka_request_init(self):
obj1 = kafka_request.SmartKitKafkaRequest(self.test_items1)
self.assertTrue(obj1.KAFKA_REPLY_TOPIC == "kafka_replyTopic")
self.assertTrue(obj1.KAFKA_KEY == "kafka_key")
self.assertTrue(obj1.topic_key == "12345")
with self.assertRaises(AttributeError):
obj2 = kafka_request.SmartKitKafkaRequest("")
self.assertTrue(obj1._callback_id == 22)
self.assertTrue(obj1._kafka_replyTopic == "any theme")
def test_smart_kafka_request_get_new_headers(self):
obj1 = kafka_request.SmartKitKafkaRequest(self.test_items1)
obj2 = kafka_request.SmartKitKafkaRequest(self.test_items2)
obj3 = kafka_request.SmartKitKafkaRequest(self.test_items3)
self.assertTrue(obj1._get_new_headers(self.test_source_mq_message) == [('any header 1', 1), (
'app_callback_id', b'22'),
('kafka_replyTopic', b'any theme')])
self.assertTrue(obj2._get_new_headers(self.test_source_mq_message) == [('any header 1', 1)])
self.assertTrue(obj3._get_new_headers(self.test_source_mq_message) == [('any header 1', 1),
('kafka_replyTopic', b'any theme')])
def test_smart_kafka_request_str(self):
obj1 = kafka_request.SmartKitKafkaRequest(self.test_items1)
obj2 = kafka_request.SmartKitKafkaRequest(self.test_items2)
self.assertTrue(obj1.__str__() == "KafkaRequest: kafka_key=54321")
self.assertTrue(obj2.__str__() == "KafkaRequest: kafka_key=None")
def test_smart_kafka_request_callback_id_header_name(self):
obj1 = kafka_request.SmartKitKafkaRequest(self.test_items1)
self.assertTrue(obj1._callback_id_header_name == "app_callback_id")