Fix uts

bifromqio · Apr 15, 2024 · 9214c2c · 9214c2c
1 parent 7fd75e3
commit 9214c2c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/bifromq-apiserver/src/test/java/com/baidu/bifromq/apiserver/NonTLSServerInitializerTest.java b/bifromq-apiserver/src/test/java/com/baidu/bifromq/apiserver/NonTLSServerInitializerTest.java
@@ -25,6 +25,7 @@
 import io.netty.channel.ChannelHandler;
 import io.netty.channel.ChannelPipeline;
 import io.netty.channel.socket.SocketChannel;
+import io.netty.handler.codec.http.HttpObjectAggregator;
 import io.netty.handler.codec.http.HttpServerCodec;
 import java.util.List;
 import lombok.extern.slf4j.Slf4j;
@@ -51,10 +52,11 @@ public void initChannel() {
         serverInitializer.initChannel(mockChannel);
 
         ArgumentCaptor<ChannelHandler> handlersCaptor = ArgumentCaptor.forClass(ChannelHandler.class);
-        verify(mockPipeline, times(3)).addLast(handlersCaptor.capture());
+        verify(mockPipeline, times(4)).addLast(handlersCaptor.capture());
         List<ChannelHandler> handlers = handlersCaptor.getAllValues();
         assertTrue(handlers.get(0) instanceof HttpServerCodec);
-        assertTrue(handlers.get(1) instanceof HTTPRequestRouter);
-        assertSame(handlers.get(2), ExceptionHandler.INSTANCE);
+        assertTrue(handlers.get(1) instanceof HttpObjectAggregator);
+        assertTrue(handlers.get(2) instanceof HTTPRequestRouter);
+        assertSame(handlers.get(3), ExceptionHandler.INSTANCE);
     }
 }
diff --git a/bifromq-apiserver/src/test/java/com/baidu/bifromq/apiserver/TLSServerInitializerTest.java b/bifromq-apiserver/src/test/java/com/baidu/bifromq/apiserver/TLSServerInitializerTest.java
@@ -26,6 +26,7 @@
 import io.netty.channel.ChannelHandler;
 import io.netty.channel.ChannelPipeline;
 import io.netty.channel.socket.SocketChannel;
+import io.netty.handler.codec.http.HttpObjectAggregator;
 import io.netty.handler.codec.http.HttpServerCodec;
 import io.netty.handler.ssl.SslContext;
 import java.util.List;
@@ -58,11 +59,12 @@ public void initChannel() {
         serverInitializer.initChannel(mockChannel);
 
         ArgumentCaptor<ChannelHandler> handlersCaptor = ArgumentCaptor.forClass(ChannelHandler.class);
-        verify(mockPipeline, times(4)).addLast(handlersCaptor.capture());
+        verify(mockPipeline, times(5)).addLast(handlersCaptor.capture());
         verify(sslContext).newHandler(byteBufAllocator);
         List<ChannelHandler> handlers = handlersCaptor.getAllValues();
         assertTrue(handlers.get(1) instanceof HttpServerCodec);
-        assertTrue(handlers.get(2) instanceof HTTPRequestRouter);
-        assertSame(handlers.get(3), ExceptionHandler.INSTANCE);
+        assertTrue(handlers.get(2) instanceof HttpObjectAggregator);
+        assertTrue(handlers.get(3) instanceof HTTPRequestRouter);
+        assertSame(handlers.get(4), ExceptionHandler.INSTANCE);
     }
 }