def test_consumer_handle_event():
config = Config(kafka_bootstrap='localhost:9002',
kafka_topic='checks',
kafka_ca_path='./',
kafka_key_path='./',
kafka_cert_path='./',
postgres_uri='postgres://*****:*****@localhost:port/db')
check = Check(response_time=100, status_code=201, url='http://google.com')
event = Mock()
event.value = check.serialize()
kafka_consumer = [event]
db = Mock()
cursor = Mock()
db.cursor = Mock(return_value=cursor)
consumer = Consumer(config, kafka_consumer, db) # type: ignore
consumer.run()
cursor.execute.assert_called()
def send_check_event(self, event: Check):
self.producer.send(self.topic, event.serialize())
def test_check_serialize():
check = Check(response_time=100, status_code=201, url='http://google.com')
assert check.serialize(
) == b'{"response_time": 100, "status_code": 201, "url": "http://google.com"}'