assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("1");
assertThat(transformed.hasNext()).isFalse();
assertThat(store.get(0)).isEqualTo(1l);
context.setPartition(1);
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("1");
assertThat(transformed.hasNext()).isFalse();
assertThat(store.get(0)).isEqualTo(1l);
context.setPartition(1);
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("2");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("2");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isFalse();
assertThat(store.get(0)).isEqualTo(1l);
assertThat(store.get(1)).isEqualTo(2l);
context.setPartition(0);
assertThat(transformed.hasNext()).isFalse();
assertThat(store.get(0)).isEqualTo(1l);
assertThat(store.get(1)).isEqualTo(2l);
context.setPartition(0);
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("4");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("4");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("3");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("3");
assertThat(transformed.hasNext()).isFalse();
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);
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);
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("5");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("5");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("6");
assertThat(transformed.hasNext()).isFalse();
assertThat(transformed.hasNext()).isTrue();
assertThat(transformed.next()).isEqualTo("6");
assertThat(transformed.hasNext()).isFalse();