From f339c7371a31ff7e401a316dfc1da71fb03fdbd5 Mon Sep 17 00:00:00 2001 From: abzymeinsjtu Date: Fri, 1 Mar 2024 18:28:54 +0800 Subject: [PATCH] [TEST] fix log related test --- .../api/service/LoggerServiceTest.java | 23 ++++++++++++++----- 1 file changed, 17 insertions(+), 6 deletions(-) diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/LoggerServiceTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/LoggerServiceTest.java index acfda80ccdcb..5b15c01e2189 100644 --- a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/LoggerServiceTest.java +++ b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/LoggerServiceTest.java @@ -49,6 +49,8 @@ import org.apache.dolphinscheduler.extract.common.transportor.TaskInstanceLogPageQueryRequest; import org.apache.dolphinscheduler.extract.common.transportor.TaskInstanceLogPageQueryResponse; +import java.io.IOException; +import java.net.ServerSocket; import java.text.MessageFormat; import java.util.HashMap; import java.util.Map; @@ -89,9 +91,18 @@ public class LoggerServiceTest { private NettyRemotingServer nettyRemotingServer; + private int nettyServerPort = 18080; + @BeforeEach public void setUp() { - nettyRemotingServer = new NettyRemotingServer(NettyServerConfig.builder().listenPort(8080).build()); + try (ServerSocket s = new ServerSocket(0)) { + nettyServerPort = s.getLocalPort(); + } catch (IOException e) { + return; + } + + + nettyRemotingServer = new NettyRemotingServer(NettyServerConfig.builder().listenPort(nettyServerPort).build()); nettyRemotingServer.start(); SpringServerMethodInvokerDiscovery springServerMethodInvokerDiscovery = new SpringServerMethodInvokerDiscovery(nettyRemotingServer); @@ -148,7 +159,7 @@ public void testQueryLog() { Assertions.assertEquals(Status.TASK_INSTANCE_HOST_IS_NULL.getCode(), result.getCode().intValue()); // PROJECT_NOT_EXIST - taskInstance.setHost("127.0.0.1:8080"); + taskInstance.setHost("127.0.0.1:" + nettyServerPort); taskInstance.setLogPath("/temp/log"); doThrow(new ServiceException(Status.PROJECT_NOT_EXIST)).when(projectService) .checkProjectAndAuthThrowException(loginUser, taskInstance.getProjectCode(), VIEW_LOG); @@ -198,7 +209,7 @@ public void testGetLogBytes() { } // PROJECT_NOT_EXIST - taskInstance.setHost("127.0.0.1:8080"); + taskInstance.setHost("127.0.0.1:" + nettyServerPort); taskInstance.setLogPath("/temp/log"); doThrow(new ServiceException(Status.PROJECT_NOT_EXIST)).when(projectService) .checkProjectAndAuthThrowException(loginUser, taskInstance.getProjectCode(), VIEW_LOG); @@ -215,7 +226,7 @@ public void testGetLogBytes() { doNothing().when(projectService).checkProjectAndAuthThrowException(loginUser, taskInstance.getProjectCode(), DOWNLOAD_LOG); byte[] logBytes = loggerService.getLogBytes(loginUser, 1); - Assertions.assertEquals(47, logBytes.length); + Assertions.assertEquals(48, logBytes.length); } @Test @@ -233,7 +244,7 @@ public void testQueryLogInSpecifiedProject() { // SUCCESS taskInstance.setTaskCode(1L); taskInstance.setId(1); - taskInstance.setHost("127.0.0.1:8080"); + taskInstance.setHost("127.0.0.1:" + nettyServerPort); taskInstance.setLogPath("/temp/log"); doNothing().when(projectService).checkProjectAndAuthThrowException(loginUser, projectCode, VIEW_LOG); when(taskInstanceDao.queryById(1)).thenReturn(taskInstance); @@ -259,7 +270,7 @@ public void testGetLogBytesInSpecifiedProject() { // SUCCESS taskInstance.setTaskCode(1L); taskInstance.setId(1); - taskInstance.setHost("127.0.0.1:8080"); + taskInstance.setHost("127.0.0.1:" + nettyServerPort); taskInstance.setLogPath("/temp/log"); doNothing().when(projectService).checkProjectAndAuthThrowException(loginUser, projectCode, DOWNLOAD_LOG); when(taskInstanceDao.queryById(1)).thenReturn(taskInstance);