X-Git-Url: https://juplo.de/gitweb/?p=demos%2Fkafka%2Fdeduplication;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fdemo%2Fkafka%2Fdeduplication%2FDeduplicationTransformerIT.java;h=339a3010111c9d9043f5b407cfe96a6b05478d56;hp=bf2fc2dd939e82b624ee0e2777e081c05f5565d1;hb=150907d53cc99a98f4c888eacff892380ffd0feb;hpb=1c19697420360e4fe2e6ab2386ef2d67133711b0

diff --git a/src/test/java/de/juplo/demo/kafka/deduplication/DeduplicationTransformerIT.java b/src/test/java/de/juplo/demo/kafka/deduplication/DeduplicationTransformerIT.java
index bf2fc2d..339a301 100644
--- a/src/test/java/de/juplo/demo/kafka/deduplication/DeduplicationTransformerIT.java
+++ b/src/test/java/de/juplo/demo/kafka/deduplication/DeduplicationTransformerIT.java
@@ -33,20 +33,19 @@ public class DeduplicationTransformerIT
     context.register(store, null);
     transformer.init(context);
     context.setTopic("foo");
+    context.setOffset(1);
 
     Iterator<String> transformed;
 
     context.setPartition(0);
-    context.setOffset(1);
-    transformed = transformer.transform("1", "1").iterator();
+    transformed = transformer.transform("1").iterator();
     assertThat(transformed.hasNext()).isTrue();
     assertThat(transformed.next()).isEqualTo("1");
     assertThat(transformed.hasNext()).isFalse();
     assertThat(store.get(0)).isEqualTo(1l);
 
     context.setPartition(1);
-    context.setOffset(1);
-    transformed = transformer.transform("2", "2").iterator();
+    transformed = transformer.transform("2").iterator();
     assertThat(transformed.hasNext()).isTrue();
     assertThat(transformed.next()).isEqualTo("2");
     assertThat(transformed.hasNext()).isFalse();
@@ -54,15 +53,13 @@ public class DeduplicationTransformerIT
     assertThat(store.get(1)).isEqualTo(2l);
 
     context.setPartition(0);
-    context.setOffset(2);
-    transformed = transformer.transform("1", "1").iterator();
+    transformed = transformer.transform("1").iterator();
     assertThat(transformed.hasNext()).isFalse();
     assertThat(store.get(0)).isEqualTo(1l);
     assertThat(store.get(1)).isEqualTo(2l);
 
     context.setPartition(0);
-    context.setOffset(3);
-    transformed = transformer.transform("1", "4").iterator();
+    transformed = transformer.transform("4").iterator();
     assertThat(transformed.hasNext()).isTrue();
     assertThat(transformed.next()).isEqualTo("4");
     assertThat(transformed.hasNext()).isFalse();
@@ -71,8 +68,7 @@ public class DeduplicationTransformerIT
 
     // The order is only guaranteed per partition!
     context.setPartition(2);
-    context.setOffset(1);
-    transformed = transformer.transform("3", "3").iterator();
+    transformed = transformer.transform("3").iterator();
     assertThat(transformed.hasNext()).isTrue();
     assertThat(transformed.next()).isEqualTo("3");
     assertThat(transformed.hasNext()).isFalse();
@@ -81,16 +77,14 @@ public class DeduplicationTransformerIT
     assertThat(store.get(2)).isEqualTo(3l);
 
     context.setPartition(1);
-    context.setOffset(2);
-    transformed = transformer.transform("2", "2").iterator();
+    transformed = transformer.transform("2").iterator();
     assertThat(transformed.hasNext()).isFalse();
     assertThat(store.get(0)).isEqualTo(4l);
     assertThat(store.get(1)).isEqualTo(2l);
     assertThat(store.get(2)).isEqualTo(3l);
 
     context.setPartition(2);
-    context.setOffset(2);
-    transformed = transformer.transform("3", "5").iterator();
+    transformed = transformer.transform("5").iterator();
     assertThat(transformed.hasNext()).isTrue();
     assertThat(transformed.next()).isEqualTo("5");
     assertThat(transformed.hasNext()).isFalse();
@@ -100,8 +94,7 @@ public class DeduplicationTransformerIT
 
     // The order is only guaranteed per partition!
     context.setPartition(1);
-    context.setOffset(3);
-    transformed = transformer.transform("2", "6").iterator();
+    transformed = transformer.transform("6").iterator();
     assertThat(transformed.hasNext()).isTrue();
     assertThat(transformed.next()).isEqualTo("6");
     assertThat(transformed.hasNext()).isFalse();