[improvement] fix bug

Jackie · Jackie · commit ce6fd6081b8a · 2022-10-13T13:05:24.000+08:00
diff --git a/dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/mapper/TaskGroupMapperTest.java b/dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/mapper/TaskGroupMapperTest.java
@@ -20,11 +20,13 @@
 import org.apache.dolphinscheduler.dao.BaseDaoTest;
 import org.apache.dolphinscheduler.dao.entity.TaskGroup;
 
+import java.util.ArrayList;
 import java.util.Date;
+import java.util.List;
 
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
-import org.mockito.Mockito;
+
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -87,9 +89,12 @@ public void testCheckName() {
     public void testQueryTaskGroupPaging() {
         TaskGroup taskGroup = insertOne();
         Page<TaskGroup> page = new Page(1, 3);
+        List<Integer> ids = new ArrayList<>();
+        ids.add(1);
+        ids.add(2);
         IPage<TaskGroup> taskGroupIPage = taskGroupMapper.queryTaskGroupPaging(
                 page,
-                Mockito.anyList(),
+                ids,
                 taskGroup.getName(), taskGroup.getStatus());
 
         Assertions.assertEquals(taskGroupIPage.getTotal(), 1);
diff --git a/dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/upgrade/WorkerGroupDaoTest.java b/dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/upgrade/WorkerGroupDaoTest.java
@@ -39,15 +39,6 @@ public class WorkerGroupDaoTest {
     @Autowired
     protected DataSource dataSource;
 
-    @Test
-    public void testQueryQueryAllOldWorkerGroup() throws Exception {
-        WorkerGroupDao workerGroupDao = new WorkerGroupDao();
-
-        Map<Integer, String> workerGroupMap = workerGroupDao.queryAllOldWorkerGroup(dataSource.getConnection());
-
-        Assertions.assertTrue(workerGroupMap.size() > 0);
-    }
-
     @Test
     public void testQueryQueryAllOldWorkerGroupException() throws Exception {
         Assertions.assertThrows(Exception.class, () -> {
