Fix test

rchodava · Apr 11, 2016 · 915c069 · 915c069
1 parent a5ebed3
commit 915c069
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/core/src/test/java/org/chodavarapu/datamill/http/impl/ClientToServerChannelHandlerTest.java b/core/src/test/java/org/chodavarapu/datamill/http/impl/ClientToServerChannelHandlerTest.java
@@ -31,9 +31,6 @@ public class ClientToServerChannelHandlerTest {
     @Mock
     private ChannelHandlerContext context;
 
-    @Captor
-    private ArgumentCaptor<LastHttpContent> lastContentCaptor;
-
     @Mock
     private Route route;
 
@@ -49,6 +46,9 @@ public class ClientToServerChannelHandlerTest {
     @Captor
     private ArgumentCaptor<HttpObject> responseFragmentsCaptor;
 
+    @Captor
+    private ArgumentCaptor<HttpObject> responseStartCaptor;
+
     private void waitForExecutorToFinishAllTasks(ExecutorService executor) throws Exception {
         for (int i = 0; i < 5; i++) {
             // We submit an empty task and wait for it so that other tasks submitted ahead of this get executed first
@@ -154,12 +154,12 @@ public void multipeResponseChunksSent() throws Exception {
 
         waitForExecutorToFinishAllTasks(service);
 
-        verify(context, times(2)).write(responseFragmentsCaptor.capture());
-        verify(context).writeAndFlush(lastContentCaptor.capture());
+        verify(context).write(responseStartCaptor.capture());
+        verify(context, times(2)).writeAndFlush(responseFragmentsCaptor.capture());
 
-        assertEquals(HttpResponseStatus.OK, ((HttpResponse) responseFragmentsCaptor.getAllValues().get(0)).status());
-        byte[] bytes = new byte[((HttpContent) responseFragmentsCaptor.getAllValues().get(1)).content().readableBytes()];
-        ((HttpContent) responseFragmentsCaptor.getAllValues().get(1)).content().readBytes(bytes);
+        assertEquals(HttpResponseStatus.OK, ((HttpResponse) responseStartCaptor.getValue()).status());
+        byte[] bytes = new byte[((HttpContent) responseFragmentsCaptor.getAllValues().get(0)).content().readableBytes()];
+        ((HttpContent) responseFragmentsCaptor.getAllValues().get(0)).content().readBytes(bytes);
         assertArrayEquals("Test Content".getBytes(), bytes);
     }
 }