Examples with MockPartitioner org.apache.kafka.test.MockPartitioner used on opensource projects

Example 1 with MockPartitioner

use of org.apache.kafka.test.MockPartitioner in project kafka by apache.

the class KafkaProducerTest method testPartitionerClose.

@Test
public void testPartitionerClose() {
    try {
        Properties props = new Properties();
        props.setProperty(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, "localhost:9999");
        MockPartitioner.resetCounters();
        props.setProperty(ProducerConfig.PARTITIONER_CLASS_CONFIG, MockPartitioner.class.getName());
        KafkaProducer<String, String> producer = new KafkaProducer<>(props, new StringSerializer(), new StringSerializer());
        assertEquals(1, MockPartitioner.INIT_COUNT.get());
        assertEquals(0, MockPartitioner.CLOSE_COUNT.get());
        producer.close();
        assertEquals(1, MockPartitioner.INIT_COUNT.get());
        assertEquals(1, MockPartitioner.CLOSE_COUNT.get());
    } finally {
        // cleanup since we are using mutable static variables in MockPartitioner
        MockPartitioner.resetCounters();
    }
}

Example 2 with MockPartitioner

use of org.apache.kafka.test.MockPartitioner in project apache-kafka-on-k8s by banzaicloud.

the class KafkaProducerTest method testPartitionerClose.

@Test
public void testPartitionerClose() throws Exception {
    try {
        Properties props = new Properties();
        props.setProperty(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, "localhost:9999");
        MockPartitioner.resetCounters();
        props.setProperty(ProducerConfig.PARTITIONER_CLASS_CONFIG, MockPartitioner.class.getName());
        KafkaProducer<String, String> producer = new KafkaProducer<String, String>(props, new StringSerializer(), new StringSerializer());
        assertEquals(1, MockPartitioner.INIT_COUNT.get());
        assertEquals(0, MockPartitioner.CLOSE_COUNT.get());
        producer.close();
        assertEquals(1, MockPartitioner.INIT_COUNT.get());
        assertEquals(1, MockPartitioner.CLOSE_COUNT.get());
    } finally {
        // cleanup since we are using mutable static variables in MockPartitioner
        MockPartitioner.resetCounters();
    }
}