def run(self): env = getdb() client = KafkaClient(self.kafka_host) consumer = self.consumer(client) for resp in consumer: if not resp.message.value: continue with env.begin(write=True) as txn: message = IWAMessage.loads(resp.message.value) if message: txn.put(bytes(message.key), bytes(message.value))
def test_dumps_then_loads(self): msg = IWAMessage('foo', 'bar') print(msg.dumps()) assert msg == IWAMessage.loads(msg.dumps())