|
1 | 1 | #!/usr/bin/env python
|
2 | 2 |
|
| 3 | +from io import StringIO |
3 | 4 | import confluent_kafka
|
4 | 5 | import confluent_kafka.avro
|
5 | 6 | import logging
|
@@ -114,3 +115,46 @@ def test_logging_constructor():
|
114 | 115 | p.poll(timeout=0.5)
|
115 | 116 |
|
116 | 117 | print('%s: %s: %d log messages seen' % (how, f.name, f.cnt))
|
| 118 | + |
| 119 | + |
| 120 | +def test_producer_logger_logging_in_given_format(): |
| 121 | + """Test that asserts that logging is working by matching part of the log message""" |
| 122 | + |
| 123 | + stringBuffer = StringIO() |
| 124 | + logger = logging.getLogger('Producer') |
| 125 | + logger.setLevel(logging.DEBUG) |
| 126 | + handler = logging.StreamHandler(stringBuffer) |
| 127 | + handler.setFormatter(logging.Formatter('%(name)s Logger | %(message)s')) |
| 128 | + logger.addHandler(handler) |
| 129 | + |
| 130 | + p = confluent_kafka.Producer( |
| 131 | + {"bootstrap.servers": "test", "logger": logger, "debug": "msg"}) |
| 132 | + val = 1 |
| 133 | + while val > 0: |
| 134 | + val = p.flush() |
| 135 | + logMessage = stringBuffer.getvalue().strip() |
| 136 | + stringBuffer.close() |
| 137 | + print(logMessage) |
| 138 | + |
| 139 | + assert "Producer Logger | INIT" in logMessage |
| 140 | + |
| 141 | + |
| 142 | +def test_consumer_logger_logging_in_given_format(): |
| 143 | + """Test that asserts that logging is working by matching part of the log message""" |
| 144 | + |
| 145 | + stringBuffer = StringIO() |
| 146 | + logger = logging.getLogger('Consumer') |
| 147 | + logger.setLevel(logging.DEBUG) |
| 148 | + handler = logging.StreamHandler(stringBuffer) |
| 149 | + handler.setFormatter(logging.Formatter('%(name)s Logger | %(message)s')) |
| 150 | + logger.addHandler(handler) |
| 151 | + |
| 152 | + c = confluent_kafka.Consumer( |
| 153 | + {"bootstrap.servers": "test", "group.id": "test", "logger": logger, "debug": "msg"}) |
| 154 | + c.poll(0) |
| 155 | + |
| 156 | + logMessage = stringBuffer.getvalue().strip() |
| 157 | + stringBuffer.close() |
| 158 | + c.close() |
| 159 | + |
| 160 | + assert "Consumer Logger | INIT" in logMessage |
0 commit comments