KafkaTopicCheckToolUnitTest.java

package com.neverwinterdp.kafka.tool;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.neverwinterdp.kafka.tool.server.KafkaCluster;
import com.neverwinterdp.util.FileUtil;

public class KafkaTopicCheckToolUnitTest {
  static {
    System.setProperty("log4j.configuration", "file:src/test/resources/test-log4j.properties");
  }

  private KafkaCluster cluster;

  @Before
  public void setUp() throws Exception {
    FileUtil.removeIfExist("./build/cluster", false);
    cluster = new KafkaCluster("./build/cluster", 1, 2);
    cluster.start();
    Thread.sleep(5000);
  }
  
  @After
  public void tearDown() throws Exception {
    cluster.shutdown();
  }

  @Test
  public void testTool() throws Exception {
    String[] args = {
      "--zk-connect", cluster.getZKConnect(),
      "--num-partition", "3",
      "--replication", "1",
      "--send-period", "10",
      "--send-message-size", "500",
      "--send-max-per-partition", "100",
      "--send-max-duration", "60000",
      "--junit-report", "build/test.xml"
    };
    KafkaTopicCheckTool.main(args);
  }
}