增加单测

Author: yzhfd

trpc-proto/trpc-proto-http/src/test/java/com/tencent/trpc/proto/http/server/AbstractHttpExecutorTest.java

@@ -66,7 +66,6 @@ public void buildRpcInvocation_shouldSuccess() throws Exception {
 
     @Test
     public void execute_shouldHandleTimeoutException() throws Exception {
-        // 准备测试数据
         HttpServletRequest request = mock(HttpServletRequest.class);
         HttpServletResponse response = mock(HttpServletResponse.class);
         AbstractHttpExecutor abstractHttpExecutor = mock(AbstractHttpExecutor.class);
@@ -76,39 +75,32 @@ public void execute_shouldHandleTimeoutException() throws Exception {
         ProviderConfig providerConfig = mock(ProviderConfig.class);
         WorkerPool workerPool = mock(WorkerPool.class);
 
-        // 模拟配置和基本信息
         when(methodInfoAndInvoker.getMethodInfo()).thenReturn(methodInfo);
         doReturn(invoker).when(methodInfoAndInvoker, "getInvoker");
         when(invoker.getConfig()).thenReturn(providerConfig);
         when(providerConfig.getRequestTimeout()).thenReturn(100); // 设置100ms超时
         when(providerConfig.getWorkerPoolObj()).thenReturn(workerPool);
 
-        // 模拟请求属性
         when(request.getAttribute(HttpConstants.REQUEST_ATTRIBUTE_TRPC_SERVICE)).thenReturn("trpc.demo.server");
         when(request.getAttribute(HttpConstants.REQUEST_ATTRIBUTE_TRPC_METHOD)).thenReturn("hello");
         when(request.getRemoteAddr()).thenReturn("127.0.0.1");
         when(request.getRemotePort()).thenReturn(8080);
 
-        // 模拟一个永远不会完成的CompletionStage，导致超时
         CompletableFuture<com.tencent.trpc.core.rpc.Response> neverCompleteFuture = new CompletableFuture<>();
         when(invoker.invoke(any())).thenReturn(neverCompleteFuture);
 
-        // 模拟私有方法调用
         doReturn(null).when(abstractHttpExecutor, "parseRpcParams", any(), any());
         when(abstractHttpExecutor, "execute", request, response, methodInfoAndInvoker).thenCallRealMethod();
         doCallRealMethod().when(abstractHttpExecutor, "doErrorReply", any(), any(), any());
         doCallRealMethod().when(abstractHttpExecutor, "httpErrorReply", any(), any(), any());
 
-        // 执行测试
         Whitebox.invokeMethod(abstractHttpExecutor, "execute", request, response, methodInfoAndInvoker);
 
-        // 验证超时后调用了错误响应（由于异步执行，超时也通过handleError处理，返回503）
         verify(response).setStatus(HttpStatus.SC_SERVICE_UNAVAILABLE);
     }
 
     @Test
     public void execute_shouldHandleInvokeException() throws Exception {
-        // 准备测试数据
         HttpServletRequest request = mock(HttpServletRequest.class);
         HttpServletResponse response = mock(HttpServletResponse.class);
         AbstractHttpExecutor abstractHttpExecutor = mock(AbstractHttpExecutor.class);
@@ -118,49 +110,41 @@ public void execute_shouldHandleInvokeException() throws Exception {
         ProviderConfig providerConfig = mock(ProviderConfig.class);
         WorkerPool workerPool = mock(WorkerPool.class);
 
-        // 模拟配置和基本信息
         when(methodInfoAndInvoker.getMethodInfo()).thenReturn(methodInfo);
         doReturn(invoker).when(methodInfoAndInvoker, "getInvoker");
         when(invoker.getConfig()).thenReturn(providerConfig);
         when(providerConfig.getRequestTimeout()).thenReturn(5000);
         when(providerConfig.getWorkerPoolObj()).thenReturn(workerPool);
 
-        // 模拟请求属性
         when(request.getAttribute(HttpConstants.REQUEST_ATTRIBUTE_TRPC_SERVICE)).thenReturn("trpc.demo.server");
         when(request.getAttribute(HttpConstants.REQUEST_ATTRIBUTE_TRPC_METHOD)).thenReturn("hello");
         when(request.getRemoteAddr()).thenReturn("127.0.0.1");
         when(request.getRemotePort()).thenReturn(8080);
         when(request.getMethod()).thenReturn("POST");
         when(request.getRequestURI()).thenReturn("/api/test");
 
-        // 模拟WorkerPool同步执行任务
         doAnswer(invocation -> {
             Task task = invocation.getArgumentAt(0, Task.class);
             task.run();
             return null;
         }).when(workerPool).execute(any(Task.class));
 
-        // 模拟invoke抛出异常
         CompletableFuture<com.tencent.trpc.core.rpc.Response> failedFuture = new CompletableFuture<>();
         failedFuture.completeExceptionally(new RuntimeException("Service invoke failed"));
         when(invoker.invoke(any())).thenReturn(failedFuture);
 
-        // 模拟私有方法调用
         doReturn(null).when(abstractHttpExecutor, "parseRpcParams", any(), any());
         when(abstractHttpExecutor, "execute", request, response, methodInfoAndInvoker).thenCallRealMethod();
         doCallRealMethod().when(abstractHttpExecutor, "doErrorReply", any(), any(), any());
         doCallRealMethod().when(abstractHttpExecutor, "httpErrorReply", any(), any(), any());
 
-        // 执行测试
         Whitebox.invokeMethod(abstractHttpExecutor, "execute", request, response, methodInfoAndInvoker);
 
-        // 验证调用了错误响应（invoke异常在handleError中处理，返回503）
         verify(response).setStatus(HttpStatus.SC_SERVICE_UNAVAILABLE);
     }
 
     @Test
     public void execute_shouldHandleResponseException() throws Exception {
-        // 准备测试数据
         HttpServletRequest request = mock(HttpServletRequest.class);
         HttpServletResponse response = mock(HttpServletResponse.class);
         AbstractHttpExecutor abstractHttpExecutor = mock(AbstractHttpExecutor.class);
@@ -170,45 +154,38 @@ public void execute_shouldHandleResponseException() throws Exception {
         ProviderConfig providerConfig = mock(ProviderConfig.class);
         WorkerPool workerPool = mock(WorkerPool.class);
 
-        // 模拟配置和基本信息
         when(methodInfoAndInvoker.getMethodInfo()).thenReturn(methodInfo);
         doReturn(invoker).when(methodInfoAndInvoker, "getInvoker");
         when(invoker.getConfig()).thenReturn(providerConfig);
         when(providerConfig.getRequestTimeout()).thenReturn(5000);
         when(providerConfig.getWorkerPoolObj()).thenReturn(workerPool);
 
-        // 模拟请求属性
         when(request.getAttribute(HttpConstants.REQUEST_ATTRIBUTE_TRPC_SERVICE)).thenReturn("trpc.demo.server");
         when(request.getAttribute(HttpConstants.REQUEST_ATTRIBUTE_TRPC_METHOD)).thenReturn("hello");
         when(request.getRemoteAddr()).thenReturn("127.0.0.1");
         when(request.getRemotePort()).thenReturn(8080);
         when(request.getMethod()).thenReturn("POST");
         when(request.getRequestURI()).thenReturn("/api/test");
 
-        // 模拟WorkerPool同步执行任务
         doAnswer(invocation -> {
             Task task = invocation.getArgumentAt(0, Task.class);
             task.run();
             return null;
         }).when(workerPool).execute(any(Task.class));
 
-        // 模拟Response包含异常
         com.tencent.trpc.core.rpc.Response rpcResponse = mock(com.tencent.trpc.core.rpc.Response.class);
         when(rpcResponse.getException()).thenReturn(
                 TRpcException.newFrameException(ErrorCode.TRPC_SERVER_VALIDATE_ERR, "Validation failed"));
         CompletableFuture<com.tencent.trpc.core.rpc.Response> responseFuture = CompletableFuture.completedFuture(rpcResponse);
         when(invoker.invoke(any())).thenReturn(responseFuture);
 
-        // 模拟私有方法调用
         doReturn(null).when(abstractHttpExecutor, "parseRpcParams", any(), any());
         when(abstractHttpExecutor, "execute", request, response, methodInfoAndInvoker).thenCallRealMethod();
         doCallRealMethod().when(abstractHttpExecutor, "doErrorReply", any(), any(), any());
         doCallRealMethod().when(abstractHttpExecutor, "httpErrorReply", any(), any(), any());
 
-        // 执行测试
         Whitebox.invokeMethod(abstractHttpExecutor, "execute", request, response, methodInfoAndInvoker);
 
-        // 验证调用了错误响应（Response异常在handleError中处理，返回503）
         verify(response).setStatus(HttpStatus.SC_SERVICE_UNAVAILABLE);
     }
 }