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()
예제 #2
0
def teardown_function(function):
    producer_produce_mock.reset_mock()