Apache
/
nifi
mirror of https://github.com/apache/nifi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NIFI-1748 removed un-used test data 

This closes #344.

Signed-off-by: Joe Skora <jskora@apache.org>
This commit is contained in:
Oleg Zhurakousky 2016-04-09 17:24:37 -04:00 committed by Joe Skora
parent e76a75c26b
commit bd1f0a82ae
1 changed files with 7 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
nifi-nar-bundles/nifi-kafka-bundle/nifi-kafka-processors/src/test/java/org/apache/nifi/processors/kafka/KafkaPublisherTest.java
View File

@ -43,13 +43,6 @@ import kafka.javaapi.consumer.ConsumerConnector;
public class KafkaPublisherTest { public class KafkaPublisherTest {
private static final String sampleData = "The true sign of intelligence is not knowledge but imagination.\n"
+ "It's not that I'm so smart, it's just that I stay with problems longer.\n"
+ "The only source of knowledge is experience.\n"
+ "Only two things are infinite, the universe and human stupidity, and I'm not sure about the former.\n";
private static final String sampleData2 = "foo|bar|baz";
private static EmbeddedKafka kafkaLocal; private static EmbeddedKafka kafkaLocal;
private static EmbeddedKafkaProducerHelper producerHelper; private static EmbeddedKafkaProducerHelper producerHelper;
@ -69,7 +62,7 @@ public class KafkaPublisherTest {
@Test @Test
public void validateSuccessfulSendAsWhole() throws Exception { public void validateSuccessfulSendAsWhole() throws Exception {
InputStream fis = new ByteArrayInputStream(sampleData.getBytes(StandardCharsets.UTF_8)); InputStream fis = new ByteArrayInputStream("Hello Kafka".getBytes(StandardCharsets.UTF_8));
String topicName = "validateSuccessfulSendAsWhole"; String topicName = "validateSuccessfulSendAsWhole";
Properties kafkaProperties = this.buildProducerProperties(); Properties kafkaProperties = this.buildProducerProperties();
@ -93,7 +86,8 @@ public class KafkaPublisherTest {
@Test @Test
public void validateSuccessfulSendAsDelimited() throws Exception { public void validateSuccessfulSendAsDelimited() throws Exception {
InputStream fis = new ByteArrayInputStream(sampleData.getBytes(StandardCharsets.UTF_8)); InputStream fis = new ByteArrayInputStream(
"Hello Kafka 1\nHello Kafka 2\nHello Kafka 3\nHello Kafka 4\n".getBytes(StandardCharsets.UTF_8));
String topicName = "validateSuccessfulSendAsDelimited"; String topicName = "validateSuccessfulSendAsDelimited";
Properties kafkaProperties = this.buildProducerProperties(); Properties kafkaProperties = this.buildProducerProperties();
@ -117,34 +111,10 @@ public class KafkaPublisherTest {
} }
} }
@Test
public void validateSuccessfulSendAsDelimited2() throws Exception {
InputStream fis = new ByteArrayInputStream(sampleData2.getBytes(StandardCharsets.UTF_8));
String topicName = "validateSuccessfulSendAsDelimited2";
Properties kafkaProperties = this.buildProducerProperties();
KafkaPublisher publisher = new KafkaPublisher(kafkaProperties);
SplittableMessageContext messageContext = new SplittableMessageContext(topicName, null, "|".getBytes(StandardCharsets.UTF_8));
publisher.publish(messageContext, fis, null, 2000);
publisher.close();
ConsumerIterator<byte[], byte[]> iter = this.buildConsumer(topicName);
assertNotNull(iter.next());
assertNotNull(iter.next());
assertNotNull(iter.next());
try {
iter.next();
fail();
} catch (ConsumerTimeoutException e) {
// that's OK since this is the Kafka mechanism to unblock
}
}
@Test @Test
public void validateSuccessfulReSendOfFailedSegments() throws Exception { public void validateSuccessfulReSendOfFailedSegments() throws Exception {
InputStream fis = new ByteArrayInputStream(sampleData.getBytes(StandardCharsets.UTF_8)); InputStream fis = new ByteArrayInputStream(
"Hello Kafka 1\nHello Kafka 2\nHello Kafka 3\nHello Kafka 4\n".getBytes(StandardCharsets.UTF_8));
String topicName = "validateSuccessfulReSendOfFailedSegments"; String topicName = "validateSuccessfulReSendOfFailedSegments";
Properties kafkaProperties = this.buildProducerProperties(); Properties kafkaProperties = this.buildProducerProperties();
@ -160,8 +130,8 @@ public class KafkaPublisherTest {
ConsumerIterator<byte[], byte[]> iter = this.buildConsumer(topicName); ConsumerIterator<byte[], byte[]> iter = this.buildConsumer(topicName);
String m1 = new String(iter.next().message()); String m1 = new String(iter.next().message());
String m2 = new String(iter.next().message()); String m2 = new String(iter.next().message());
assertEquals("It's not that I'm so smart, it's just that I stay with problems longer.", m1); assertEquals("Hello Kafka 2", m1);
assertEquals("Only two things are infinite, the universe and human stupidity, and I'm not sure about the former.", m2); assertEquals("Hello Kafka 4", m2);
try { try {
iter.next(); iter.next();
fail(); fail();