diff options
Diffstat (limited to 'container-core/src/test/java/com/yahoo/processing/execution/test/StreamingTestCase.java')
-rw-r--r-- | container-core/src/test/java/com/yahoo/processing/execution/test/StreamingTestCase.java | 48 |
1 files changed, 24 insertions, 24 deletions
diff --git a/container-core/src/test/java/com/yahoo/processing/execution/test/StreamingTestCase.java b/container-core/src/test/java/com/yahoo/processing/execution/test/StreamingTestCase.java index d494e774f8e..557bef39276 100644 --- a/container-core/src/test/java/com/yahoo/processing/execution/test/StreamingTestCase.java +++ b/container-core/src/test/java/com/yahoo/processing/execution/test/StreamingTestCase.java @@ -10,13 +10,13 @@ import com.yahoo.processing.execution.Execution; import com.yahoo.processing.response.Data; import com.yahoo.processing.response.IncomingData; import com.yahoo.processing.test.ProcessorLibrary; -import org.junit.Test; +import org.junit.jupiter.api.Test; import java.util.concurrent.ExecutionException; import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeoutException; -import static org.junit.Assert.assertEquals; +import static org.junit.jupiter.api.Assertions.assertEquals; /** * Tests listening on every available new piece of data in a response @@ -28,48 +28,48 @@ public class StreamingTestCase { /** Tests adding a chain which is called every time new data is added to a data list */ @SuppressWarnings({"unchecked"}) @Test - public void testStreamingData() throws InterruptedException, ExecutionException, TimeoutException { + void testStreamingData() throws InterruptedException, ExecutionException, TimeoutException { // Set up StreamProcessor streamProcessor = new StreamProcessor(); Chain<Processor> streamProcessing = new Chain<Processor>(streamProcessor); - ProcessorLibrary.FutureDataSource futureDataSource=new ProcessorLibrary.FutureDataSource(); - Chain<Processor> main=new Chain<>(new ProcessorLibrary.DataCounter(), - new ProcessorLibrary.StreamProcessingInitiator(streamProcessing), - futureDataSource); + ProcessorLibrary.FutureDataSource futureDataSource = new ProcessorLibrary.FutureDataSource(); + Chain<Processor> main = new Chain<>(new ProcessorLibrary.DataCounter(), + new ProcessorLibrary.StreamProcessingInitiator(streamProcessing), + futureDataSource); // Execute - Request request=new Request(); - Response response= Execution.createRoot(main, 0, Execution.Environment.createEmpty()).process(request); + Request request = new Request(); + Response response = Execution.createRoot(main, 0, Execution.Environment.createEmpty()).process(request); IncomingData incomingData = futureDataSource.incomingData.get(0); // State prior to receiving any additional data - assertEquals(1,response.data().asList().size()); - assertEquals("Data count: 0",response.data().get(0).toString()); - assertEquals("Add data listener invoked also for DataCounter", 1, streamProcessor.invocationCount); - assertEquals("Initial data count", 1, response.data().asList().size()); + assertEquals(1, response.data().asList().size()); + assertEquals("Data count: 0", response.data().get(0).toString()); + assertEquals(1, streamProcessor.invocationCount, "Add data listener invoked also for DataCounter"); + assertEquals(1, response.data().asList().size(), "Initial data count"); // add first data - we have no listener so the data is held in the incoming buffer incomingData.add(new ProcessorLibrary.StringData(request, "d1")); - assertEquals("Data add listener not invoked as we are not listening on new data yet",1, streamProcessor.invocationCount); - assertEquals("New data is not consumed", 1, response.data().asList().size()); + assertEquals(1, streamProcessor.invocationCount, "Data add listener not invoked as we are not listening on new data yet"); + assertEquals(1, response.data().asList().size(), "New data is not consumed"); // start listening on incoming data - this is what a renderer will do incomingData.addNewDataListener(new MockNewDataListener(incomingData), MoreExecutors.directExecutor()); - assertEquals("We got a data add event for the data which was already added", 2, streamProcessor.invocationCount); - assertEquals("New data is consumed", 2, response.data().asList().size()); + assertEquals(2, streamProcessor.invocationCount, "We got a data add event for the data which was already added"); + assertEquals(2, response.data().asList().size(), "New data is consumed"); incomingData.add(new ProcessorLibrary.StringData(request, "d2")); - assertEquals("We are now getting data add events each time", 3, streamProcessor.invocationCount); - assertEquals("New data is consumed", 3, response.data().asList().size()); + assertEquals(3, streamProcessor.invocationCount, "We are now getting data add events each time"); + assertEquals(3, response.data().asList().size(), "New data is consumed"); incomingData.addLast(new ProcessorLibrary.StringData(request, "d3")); - assertEquals("We are getting data add events also the last time", 4, streamProcessor.invocationCount); - assertEquals("New data is consumed", 4, response.data().asList().size()); + assertEquals(4, streamProcessor.invocationCount, "We are getting data add events also the last time"); + assertEquals(4, response.data().asList().size(), "New data is consumed"); response.data().completeFuture().get(1000, TimeUnit.MILLISECONDS); // no-op here - assertEquals("d1",response.data().get(1).toString().toString()); - assertEquals("d2",response.data().get(2).toString().toString()); - assertEquals("d3",response.data().get(3).toString().toString()); + assertEquals("d1", response.data().get(1).toString().toString()); + assertEquals("d2", response.data().get(2).toString().toString()); + assertEquals("d3", response.data().get(3).toString().toString()); } private static class MockNewDataListener implements Runnable { |