def test_avro_producer_produce(avro_producer):
key = 'a'
value = message
topic = 'z'
avro_producer.produce(key=key, value=value, topic=topic)
producer_produce_mock.assert_called_once_with(
topic, key, avro_producer.value_serializer.serialize(value, topic))
def test_avro_producer_produce(avro_producer):
key = 'a'
value = 'a'
topic = 'c'
avro_producer.produce(topic, key=key, value=value)
producer_produce_mock.assert_called_once_with(
topic, b'\x00\x00\x00\x00\x01\x02a', value)
def test_produce_consume_one_tombstone(producer, consumer):
producer_produce_mock.reset_mock()
producer_flush_mock.reset_mock()
key = bytes(str(uuid.uuid4()), encoding='utf8')
value = None
producer.produce(TEST_TOPIC, key, value, sync=True)
producer_produce_mock.assert_called_once_with(TEST_TOPIC, key, value)
producer_flush_mock.assert_called_once_with()