org.apache.kafka.clients.producer

Class MockProducer<K,V>

java.lang.Object

org.apache.kafka.clients.producer.MockProducer<K,V>

All Implemented Interfaces:

java.io.Closeable, java.lang.AutoCloseable, Producer<K,V>

public class MockProducer<K,V>
extends java.lang.Object
implements Producer<K,V>

A mock of the producer interface you can use for testing code that uses Kafka.

By default this mock will synchronously complete each send call successfully. However it can be configured to allow the user to control the completion of the call and supply an optional error for the producer to throw.

Constructor Summary

Constructors

Constructor and Description
MockProducer() Create a new mock producer with invented metadata.
MockProducer(boolean autoComplete, Partitioner partitioner, Serializer<K> keySerializer, Serializer<V> valueSerializer) Create a new mock producer with invented metadata the given autoComplete setting, partitioner and key\value serializers.
MockProducer(boolean autoComplete, Serializer<K> keySerializer, Serializer<V> valueSerializer) Create a new mock producer with invented metadata the given autoComplete setting and key\value serializers.
MockProducer(Cluster cluster, boolean autoComplete, Partitioner partitioner, Serializer<K> keySerializer, Serializer<V> valueSerializer) Create a mock producer

Method Summary

Modifier and Type	Method and Description
void	abortTransaction() See KafkaProducer.abortTransaction()
void	beginTransaction() See KafkaProducer.beginTransaction()
void	clear() Clear the stored history of sent records, consumer group offsets, and transactional state
void	close() See KafkaProducer.close()
void	close(long timeout, java.util.concurrent.TimeUnit timeUnit) See KafkaProducer.close(long, TimeUnit)
boolean	closed()
long	commitCount()
void	commitTransaction() See KafkaProducer.commitTransaction()
boolean	completeNext() Complete the earliest uncompleted call successfully.
java.util.List<java.util.Map<java.lang.String,java.util.Map<TopicPartition,OffsetAndMetadata>>>	consumerGroupOffsetsHistory() Get the list of committed consumer group offsets since the last call to clear()
boolean	errorNext(java.lang.RuntimeException e) Complete the earliest uncompleted call with the given error.
void	fenceProducer()
void	fenceProducerOnClose()
void	flush() See KafkaProducer.flush()
boolean	flushed()
java.util.List<ProducerRecord<K,V>>	history() Get the list of sent records since the last call to clear()
void	initTransactions() See KafkaProducer.initTransactions()
java.util.Map<MetricName,Metric>	metrics() See KafkaProducer.metrics()
java.util.List<PartitionInfo>	partitionsFor(java.lang.String topic) See KafkaProducer.partitionsFor(String)
java.util.concurrent.Future<RecordMetadata>	send(ProducerRecord<K,V> record) Adds the record to the list of sent records.
java.util.concurrent.Future<RecordMetadata>	send(ProducerRecord<K,V> record, Callback callback) Adds the record to the list of sent records.
void	sendOffsetsToTransaction(java.util.Map<TopicPartition,OffsetAndMetadata> offsets, java.lang.String consumerGroupId) See KafkaProducer.sendOffsetsToTransaction(Map, String)
boolean	sentOffsets()
void	setMockMetrics(MetricName name, Metric metric) Set a mock metric for testing purpose
boolean	transactionAborted()
boolean	transactionCommitted()
boolean	transactionInFlight()
boolean	transactionInitialized()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

MockProducer

public MockProducer(Cluster cluster,
                    boolean autoComplete,
                    Partitioner partitioner,
                    Serializer<K> keySerializer,
                    Serializer<V> valueSerializer)

Create a mock producer

Parameters:

cluster - The cluster holding metadata for this producer

autoComplete - If true automatically complete all requests successfully and execute the callback. Otherwise the user must call completeNext() or errorNext(RuntimeException) after send() to complete the call and unblock the Future<RecordMetadata> that is returned.

partitioner - The partition strategy

keySerializer - The serializer for key that implements Serializer.

valueSerializer - The serializer for value that implements Serializer.

MockProducer

public MockProducer(boolean autoComplete,
                    Serializer<K> keySerializer,
                    Serializer<V> valueSerializer)

Create a new mock producer with invented metadata the given autoComplete setting and key\value serializers. Equivalent to MockProducer(Cluster, boolean, Partitioner, Serializer, Serializer) new MockProducer(Cluster.empty(), autoComplete, new DefaultPartitioner(), keySerializer, valueSerializer)}

MockProducer

public MockProducer(boolean autoComplete,
                    Partitioner partitioner,
                    Serializer<K> keySerializer,
                    Serializer<V> valueSerializer)

Create a new mock producer with invented metadata the given autoComplete setting, partitioner and key\value serializers. Equivalent to MockProducer(Cluster, boolean, Partitioner, Serializer, Serializer) new MockProducer(Cluster.empty(), autoComplete, partitioner, keySerializer, valueSerializer)}

MockProducer

public MockProducer()

Create a new mock producer with invented metadata. Equivalent to MockProducer(Cluster, boolean, Partitioner, Serializer, Serializer) new MockProducer(Cluster.empty(), false, null, null, null)}

Method Detail

initTransactions

public void initTransactions()

Description copied from interface: Producer

See KafkaProducer.initTransactions()

Specified by:

initTransactions in interface Producer<K,V>

beginTransaction

public void beginTransaction()
                      throws ProducerFencedException

Description copied from interface: Producer

See KafkaProducer.beginTransaction()

Specified by:

beginTransaction in interface Producer<K,V>

Throws:

ProducerFencedException

sendOffsetsToTransaction

public void sendOffsetsToTransaction(java.util.Map<TopicPartition,OffsetAndMetadata> offsets,
                                     java.lang.String consumerGroupId)
                              throws ProducerFencedException

Description copied from interface: Producer

See KafkaProducer.sendOffsetsToTransaction(Map, String)

Specified by:

sendOffsetsToTransaction in interface Producer<K,V>

Throws:

ProducerFencedException

commitTransaction

public void commitTransaction()
                       throws ProducerFencedException

Description copied from interface: Producer

See KafkaProducer.commitTransaction()

Specified by:

commitTransaction in interface Producer<K,V>

Throws:

ProducerFencedException

abortTransaction

public void abortTransaction()
                      throws ProducerFencedException

Description copied from interface: Producer

See KafkaProducer.abortTransaction()

Specified by:

abortTransaction in interface Producer<K,V>

Throws:

ProducerFencedException

send

public java.util.concurrent.Future<RecordMetadata> send(ProducerRecord<K,V> record)

Adds the record to the list of sent records. The RecordMetadata returned will be immediately satisfied.

Specified by:

send in interface Producer<K,V>

See Also:

history()

send

public java.util.concurrent.Future<RecordMetadata> send(ProducerRecord<K,V> record,
                                                        Callback callback)

Adds the record to the list of sent records.

Specified by:

send in interface Producer<K,V>

See Also:

history()

flush

public void flush()

Description copied from interface: Producer

See KafkaProducer.flush()

Specified by:

flush in interface Producer<K,V>

partitionsFor

public java.util.List<PartitionInfo> partitionsFor(java.lang.String topic)

Description copied from interface: Producer

See KafkaProducer.partitionsFor(String)

Specified by:

partitionsFor in interface Producer<K,V>

metrics

public java.util.Map<MetricName,Metric> metrics()

Description copied from interface: Producer

See KafkaProducer.metrics()

Specified by:

metrics in interface Producer<K,V>

setMockMetrics

public void setMockMetrics(MetricName name,
                           Metric metric)

Set a mock metric for testing purpose

close

public void close()

Description copied from interface: Producer

See KafkaProducer.close()

Specified by:

close in interface java.io.Closeable

Specified by:

close in interface java.lang.AutoCloseable

Specified by:

close in interface Producer<K,V>

close

public void close(long timeout,
                  java.util.concurrent.TimeUnit timeUnit)

Description copied from interface: Producer

See KafkaProducer.close(long, TimeUnit)

Specified by:

close in interface Producer<K,V>

closed

public boolean closed()

fenceProducer

public void fenceProducer()

fenceProducerOnClose

public void fenceProducerOnClose()

transactionInitialized

public boolean transactionInitialized()

transactionInFlight

public boolean transactionInFlight()

transactionCommitted

public boolean transactionCommitted()

transactionAborted

public boolean transactionAborted()

flushed

public boolean flushed()

sentOffsets

public boolean sentOffsets()

commitCount

public long commitCount()

history

public java.util.List<ProducerRecord<K,V>> history()

Get the list of sent records since the last call to clear()

consumerGroupOffsetsHistory

public java.util.List<java.util.Map<java.lang.String,java.util.Map<TopicPartition,OffsetAndMetadata>>> consumerGroupOffsetsHistory()

Get the list of committed consumer group offsets since the last call to clear()

clear

public void clear()

Clear the stored history of sent records, consumer group offsets, and transactional state

completeNext

public boolean completeNext()

Complete the earliest uncompleted call successfully.

Returns:

true if there was an uncompleted call to complete

errorNext

public boolean errorNext(java.lang.RuntimeException e)

Complete the earliest uncompleted call with the given error.

Returns:

true if there was an uncompleted call to complete