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