add ut

dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/AlertServer.java

@@ -133,7 +133,7 @@ public class AlertServer {
                 logger.warn("No Alert Plugin . Can not send alert info. ");
             } else {
                 List<Alert> alerts = alertDao.listWaitExecutionAlert();
-                alertSender = new AlertSender(alerts, alertDao, alertPluginManager, pluginDao);
+                alertSender = new AlertSender(alerts, alertDao, alertPluginManager);
                 alertSender.run();
             }
         }

dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/processor/AlertRequestProcessor.java

@@ -59,7 +59,7 @@ public class AlertRequestProcessor implements NettyRequestProcessor {
                 command.getBody(), AlertSendRequestCommand.class);
         logger.info("received command : {}", alertSendRequestCommand);
 
-        AlertSender alertSender = new AlertSender(alertDao, alertPluginManager, pluginDao);
+        AlertSender alertSender = new AlertSender(alertDao, alertPluginManager);
         AlertSendResponseCommand alertSendResponseCommand = alertSender.syncHandler(alertSendRequestCommand.getGroupId(), alertSendRequestCommand.getTitle(), alertSendRequestCommand.getContent());
         channel.writeAndFlush(alertSendResponseCommand.convert2Command(command.getOpaque()));
 

dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/runner/AlertSender.java

@@ -48,25 +48,22 @@ public class AlertSender {
 
     private List<Alert> alertList;
     private AlertDao alertDao;
-    private PluginDao pluginDao;
     private AlertPluginManager alertPluginManager;
 
     public AlertSender(AlertPluginManager alertPluginManager) {
         this.alertPluginManager = alertPluginManager;
     }
 
-    public AlertSender(AlertDao alertDao, AlertPluginManager alertPluginManager, PluginDao pluginDao) {
+    public AlertSender(AlertDao alertDao, AlertPluginManager alertPluginManager) {
         super();
         this.alertDao = alertDao;
-        this.pluginDao = pluginDao;
         this.alertPluginManager = alertPluginManager;
     }
 
-    public AlertSender(List<Alert> alertList, AlertDao alertDao, AlertPluginManager alertPluginManager, PluginDao pluginDao) {
+    public AlertSender(List<Alert> alertList, AlertDao alertDao, AlertPluginManager alertPluginManager) {
         super();
         this.alertList = alertList;
         this.alertDao = alertDao;
-        this.pluginDao = pluginDao;
         this.alertPluginManager = alertPluginManager;
     }
 

dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/AlertServerTest.java

@@ -81,7 +81,8 @@ public class AlertServerTest {
         Assert.assertNotNull(alertServer);
 
         new Thread(() -> {
-            alertServer.start(); })
+            alertServer.start();
+        })
                 .start();
 
         Thread.sleep(5 * Constants.ALERT_SCAN_INTERVAL);

dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/plugin/EmailAlertPluginTest.java

@@ -136,7 +136,7 @@ public class EmailAlertPluginTest {
         alertPluginInstance.setPluginInstanceParams(getEmailAlertParams());
         alertDao.getAlertPluginInstanceMapper().insert(alertPluginInstance);
 
-        AlertSender alertSender = new AlertSender(alertList, alertDao, alertPluginManager, pluginDao);
+        AlertSender alertSender = new AlertSender(alertList, alertDao, alertPluginManager);
         alertSender.run();
 
         Alert alertResult = alertDao.getAlertMapper().selectById(alert1.getId());

dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/runner/AlertSenderTest.java

@@ -67,7 +67,7 @@ public class AlertSenderTest {
         int alertGroupId = 1;
         String title = "alert mail test title";
         String content = "alert mail test content";
-        alertSender = new AlertSender(alertDao,alertPluginManager,pluginDao);
+        alertSender = new AlertSender(alertDao, alertPluginManager);
 
         //1.alert instance does not exist
         PowerMockito.when(alertDao.listInstanceByAlertGroupId(alertGroupId)).thenReturn(null);
@@ -151,7 +151,7 @@ public class AlertSenderTest {
         alert.setContent(content);
         alertList.add(alert);
 
-        alertSender = new AlertSender(alertList,alertDao,alertPluginManager,pluginDao);
+        alertSender = new AlertSender(alertList, alertDao, alertPluginManager);
 
         int pluginDefineId = 1;
         String pluginInstanceParams = "alert-instance-mail-params";

dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/AlertPluginInstanceServiceTest.java

@@ -22,6 +22,7 @@ import org.apache.dolphinscheduler.api.service.impl.AlertPluginInstanceServiceIm
 import org.apache.dolphinscheduler.common.Constants;
 import org.apache.dolphinscheduler.common.enums.UserType;
 import org.apache.dolphinscheduler.dao.entity.AlertPluginInstance;
+import org.apache.dolphinscheduler.dao.entity.PluginDefine;
 import org.apache.dolphinscheduler.dao.entity.User;
 import org.apache.dolphinscheduler.dao.mapper.AlertGroupMapper;
 import org.apache.dolphinscheduler.dao.mapper.AlertPluginInstanceMapper;
@@ -29,6 +30,7 @@ import org.apache.dolphinscheduler.dao.mapper.PluginDefineMapper;
 
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 
@@ -60,6 +62,82 @@ public class AlertPluginInstanceServiceTest {
 
     private User user;
 
+    private String uiParams="[\n"
+            + "    {\n"
+            + "        \"field\":\"userParams\",\n"
+            + "        \"name\":\"user.params\",\n"
+            + "        \"props\":{\n"
+            + "            \"placeholder\":\"please enter your custom parameters, which will be passed to you when calling your script\",\n"
+            + "            \"size\":\"small\"\n"
+            + "        },\n"
+            + "        \"type\":\"input\",\n"
+            + "        \"title\":\"user.params\",\n"
+            + "        \"value\":\"userParams\",\n"
+            + "        \"validate\":[\n"
+            + "            {\n"
+            + "                \"required\":false,\n"
+            + "                \"message\":null,\n"
+            + "                \"type\":\"string\",\n"
+            + "                \"trigger\":\"blur\",\n"
+            + "                \"min\":null,\n"
+            + "                \"max\":null\n"
+            + "            }\n"
+            + "        ]\n"
+            + "    },\n"
+            + "    {\n"
+            + "        \"field\":\"path\",\n"
+            + "        \"name\":\"path\",\n"
+            + "        \"props\":{\n"
+            + "            \"placeholder\":\"please upload the file to the disk directory of the alert server, and ensure that the path is absolute and has the corresponding access rights\",\n"
+            + "            \"size\":\"small\"\n"
+            + "        },\n"
+            + "        \"type\":\"input\",\n"
+            + "        \"title\":\"path\",\n"
+            + "        \"value\":\"/kris/script/path\",\n"
+            + "        \"validate\":[\n"
+            + "            {\n"
+            + "                \"required\":true,\n"
+            + "                \"message\":null,\n"
+            + "                \"type\":\"string\",\n"
+            + "                \"trigger\":\"blur\",\n"
+            + "                \"min\":null,\n"
+            + "                \"max\":null\n"
+            + "            }\n"
+            + "        ]\n"
+            + "    },\n"
+            + "    {\n"
+            + "        \"field\":\"type\",\n"
+            + "        \"name\":\"type\",\n"
+            + "        \"props\":{\n"
+            + "            \"placeholder\":null,\n"
+            + "            \"size\":\"small\"\n"
+            + "        },\n"
+            + "        \"type\":\"radio\",\n"
+            + "        \"title\":\"type\",\n"
+            + "        \"value\":0,\n"
+            + "        \"validate\":[\n"
+            + "            {\n"
+            + "                \"required\":true,\n"
+            + "                \"message\":null,\n"
+            + "                \"type\":\"string\",\n"
+            + "                \"trigger\":\"blur\",\n"
+            + "                \"min\":null,\n"
+            + "                \"max\":null\n"
+            + "            }\n"
+            + "        ],\n"
+            + "        \"options\":[\n"
+            + "            {\n"
+            + "                \"label\":\"SHELL\",\n"
+            + "                \"value\":0,\n"
+            + "                \"disabled\":false\n"
+            + "            }\n"
+            + "        ]\n"
+            + "    }\n"
+            + "]\n"
+            + "\n";
+
+    private String paramsMap="{\"path\":\"/kris/script/path\",\"userParams\":\"userParams\",\"type\":\"0\"}";
+
     @Before
     public void before() {
         user = new User();
@@ -77,10 +155,10 @@ public class AlertPluginInstanceServiceTest {
     @Test
     public void testCreate() {
         Mockito.when(alertPluginInstanceMapper.queryByInstanceName("test")).thenReturn(alertPluginInstances);
-        Map<String, Object> result = alertPluginInstanceService.create(user, 1, "test", "test params");
+        Map<String, Object> result = alertPluginInstanceService.create(user, 1, "test", uiParams);
         Assert.assertEquals(Status.PLUGIN_INSTANCE_ALREADY_EXIT, result.get(Constants.STATUS));
         Mockito.when(alertPluginInstanceMapper.insert(Mockito.any())).thenReturn(1);
-        result = alertPluginInstanceService.create(user, 1, "test1", "test params");
+        result = alertPluginInstanceService.create(user, 1, "test1", uiParams);
         Assert.assertEquals(Status.SUCCESS, result.get(Constants.STATUS));
     }
 
@@ -96,4 +174,31 @@ public class AlertPluginInstanceServiceTest {
 
     }
 
+    @Test
+    public void testUpdate(){
+        Mockito.when(alertPluginInstanceMapper.updateById(Mockito.any())).thenReturn(0);
+        Map<String, Object> result= alertPluginInstanceService.update(user,1,"testUpdate",uiParams);
+        Assert.assertEquals(Status.SAVE_ERROR, result.get(Constants.STATUS));
+        Mockito.when(alertPluginInstanceMapper.updateById(Mockito.any())).thenReturn(1);
+        result= alertPluginInstanceService.update(user,1,"testUpdate",uiParams);
+        Assert.assertEquals(Status.SUCCESS, result.get(Constants.STATUS));
+    }
+
+    @Test
+    public  void testQueryAll(){
+        AlertPluginInstance alertPluginInstance=new AlertPluginInstance();
+        alertPluginInstance.setId(1);
+        alertPluginInstance.setPluginDefineId(1);
+        alertPluginInstance.setPluginInstanceParams(paramsMap);
+        alertPluginInstance.setInstanceName("test");
+        PluginDefine pluginDefine=new PluginDefine("script","script",uiParams);
+        pluginDefine.setId(1);
+        List<PluginDefine> pluginDefines= Collections.singletonList(pluginDefine);
+        List<AlertPluginInstance> pluginInstanceList= Collections.singletonList(alertPluginInstance);
+        Mockito.when(alertPluginInstanceMapper.queryAllAlertPluginInstanceList()).thenReturn(pluginInstanceList);
+        Mockito.when(pluginDefineMapper.queryAllPluginDefineList()).thenReturn(pluginDefines);
+        Map<String, Object> result = alertPluginInstanceService.queryAll();
+        Assert.assertEquals(Status.SUCCESS, result.get(Constants.STATUS));
+    }
+
 }