@Override public StreamEntity<T> buildClone() { return new BasicTestControlMessage(this.id); } }
@Test public void testControlMessages() throws Exception { MyExtractor extractor = new MyExtractor(new StreamEntity[]{new RecordEnvelope<>("a"), new BasicTestControlMessage("1"), new RecordEnvelope<>("b"), new BasicTestControlMessage("2")}); MyConverter converter = new MyConverter(); MyDataWriter writer = new MyDataWriter(); Task task = setupTask(extractor, writer, converter); task.run(); task.commit(); Assert.assertEquals(task.getTaskState().getWorkingState(), WorkUnitState.WorkingState.SUCCESSFUL); Assert.assertEquals(converter.records, Lists.newArrayList("a", "b")); Assert.assertEquals(converter.messages, Lists.newArrayList(new BasicTestControlMessage("1"), new BasicTestControlMessage("2"))); Assert.assertEquals(writer.records, Lists.newArrayList("a", "b")); Assert.assertEquals(writer.messages, Lists.newArrayList(new BasicTestControlMessage("1"), new BasicTestControlMessage("2"))); }
new BasicTestControlMessage("1"), new RecordEnvelope<>("b"), new BasicTestControlMessage("2")};
Assert.assertTrue(output2.poll() instanceof RecordEnvelope); flowable._subscriber.onNext(new BasicTestControlMessage<byte[]>("control")); Assert.assertTrue(output0.poll() instanceof BasicTestControlMessage); Assert.assertTrue(output1.poll() instanceof BasicTestControlMessage);
@Override public StreamEntity<T> buildClone() { return new BasicTestControlMessage(this.id); } }