This is an automated email from the ASF dual-hosted git repository.

lgcareer pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/incubator-dolphinscheduler.git


The following commit(s) were added to refs/heads/dev by this push:
     new 28766df  send mail error, #2466 bug fix (#2470)
28766df is described below

commit 28766df53fac118dd0740ad33f2fc6b52e014670
Author: qiaozhanwei <[email protected]>
AuthorDate: Tue Apr 21 14:40:19 2020 +0800

    send mail error, #2466 bug fix (#2470)
    
    * dispatch task fail will set task status failed
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,no worker condition , master will while ture wait for worker startup
    2,worker response task status sync wait for result
    
    * 1,task status statistics and process status statistics bug fix (#2357)
    2,worker group bug fix
    
    * 1,task status statistics and process status statistics bug fix (#2357)
    2,worker group bug fix
    
    * 1,task status statistics and process status statistics bug fix (#2357)
    2,worker group bug fix
    
    * 1,task status statistics and process status statistics bug fix (#2357)
    2,worker group bug fix
    
    * send mail error, #2466 bug fix
    
    * send mail error, #2466 bug fix
    
    * send mail error, #2466 bug fix
    
    * send mail error, #2466 bug fix
    
    Co-authored-by: qiaozhanwei <[email protected]>
---
 .../alert/utils/EnterpriseWeChatUtils.java         |   3 +-
 .../alert/utils/EnterpriseWeChatUtilsTest.java     | 269 ++++++++++++++++-----
 2 files changed, 203 insertions(+), 69 deletions(-)

diff --git 
a/dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtils.java
 
b/dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtils.java
index 900c120..170c0dd 100644
--- 
a/dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtils.java
+++ 
b/dolphinscheduler-alert/src/main/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtils.java
@@ -48,7 +48,7 @@ public class EnterpriseWeChatUtils {
     private static final String ENTERPRISE_WE_CHAT_SECRET = 
PropertyUtils.getString(Constants.ENTERPRISE_WECHAT_SECRET);
 
     private static final String ENTERPRISE_WE_CHAT_TOKEN_URL = 
PropertyUtils.getString(Constants.ENTERPRISE_WECHAT_TOKEN_URL);
-    private static final String ENTERPRISE_WE_CHAT_TOKEN_URL_REPLACE = 
ENTERPRISE_WE_CHAT_TOKEN_URL
+    private static final String ENTERPRISE_WE_CHAT_TOKEN_URL_REPLACE = 
ENTERPRISE_WE_CHAT_TOKEN_URL == null ? null : ENTERPRISE_WE_CHAT_TOKEN_URL
             .replaceAll("\\$corpId", ENTERPRISE_WE_CHAT_CORP_ID)
             .replaceAll("\\$secret", ENTERPRISE_WE_CHAT_SECRET);
 
@@ -266,5 +266,4 @@ public class EnterpriseWeChatUtils {
         return result;
 
     }
-
 }
diff --git 
a/dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtilsTest.java
 
b/dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtilsTest.java
index 15b92a6..852d245 100644
--- 
a/dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtilsTest.java
+++ 
b/dolphinscheduler-alert/src/test/java/org/apache/dolphinscheduler/alert/utils/EnterpriseWeChatUtilsTest.java
@@ -17,13 +17,21 @@
 package org.apache.dolphinscheduler.alert.utils;
 
 import com.alibaba.fastjson.JSON;
+import org.apache.dolphinscheduler.common.enums.AlertType;
+import org.apache.dolphinscheduler.common.enums.ShowType;
+import org.apache.dolphinscheduler.dao.entity.Alert;
 import org.junit.Assert;
 import org.junit.Ignore;
 import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
 
 import java.io.IOException;
-import java.util.Arrays;
-import java.util.Collection;
+import java.util.*;
 
 /**
  * Please manually modify the configuration file before testing.
@@ -36,84 +44,211 @@ import java.util.Collection;
  *   enterprise.wechat.agent.id
  *   enterprise.wechat.users
  */
-@Ignore
+@PrepareForTest(PropertyUtils.class)
+@RunWith(PowerMockRunner.class)
 public class EnterpriseWeChatUtilsTest {
 
-    private String agentId = 
PropertyUtils.getString(Constants.ENTERPRISE_WECHAT_AGENT_ID); // app id
-    private Collection<String> listUserId = 
Arrays.asList(PropertyUtils.getString(Constants.ENTERPRISE_WECHAT_USERS).split(","));
+    private static final String toParty = "wwc99134b6fc1edb6";
+    private static final String enterpriseWechatSecret = 
"Uuv2KFrkdf7SeKOsTDCpsTkpawXBMNRhFy6VKX5FV";
+    private static final String enterpriseWechatAgentId = "1000004";
+    private static final String enterpriseWechatUsers="LiGang,journey";
+    private static final String msg = "hello world";
+
+
+    @Test
+    public void testIsEnable(){
+        PowerMockito.mockStatic(PropertyUtils.class);
+        
Mockito.when(PropertyUtils.getBoolean(Constants.ENTERPRISE_WECHAT_ENABLE)).thenReturn(true);
+        Boolean weChartEnable = EnterpriseWeChatUtils.isEnable();
+        Assert.assertTrue(weChartEnable);
+    }
+
+
+    @Test
+    public void testMakeTeamSendMsg1(){
+        String sendMsg = EnterpriseWeChatUtils.makeTeamSendMsg(toParty, 
enterpriseWechatSecret, msg);
+        Assert.assertTrue(sendMsg.contains(toParty));
+        Assert.assertTrue(sendMsg.contains(enterpriseWechatSecret));
+        Assert.assertTrue(sendMsg.contains(msg));
+
+    }
+
 
-    // Please change
-    private String partyId = "2";
-    private Collection<String> listPartyId = Arrays.asList("2","4");
     @Test
-    public void testSendSingleTeamWeChat() {
-        try {
-            String token = EnterpriseWeChatUtils.getToken();
-            String msg = EnterpriseWeChatUtils.makeTeamSendMsg(partyId, 
agentId, "hello world");
-            String resp = EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", 
msg, token);
-
-            String errmsg = JSON.parseObject(resp).getString("errmsg");
-            Assert.assertEquals("ok",errmsg);
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
+    public void testMakeTeamSendMsg2(){
+        List<String> parties = new ArrayList<>();
+        parties.add(toParty);
+        parties.add("test1");
+
+        String sendMsg = EnterpriseWeChatUtils.makeTeamSendMsg(parties, 
enterpriseWechatSecret, msg);
+        Assert.assertTrue(sendMsg.contains(toParty));
+        Assert.assertTrue(sendMsg.contains(enterpriseWechatSecret));
+        Assert.assertTrue(sendMsg.contains(msg));
     }
 
     @Test
-    public void testSendMultiTeamWeChat() {
-
-        try {
-            String token = EnterpriseWeChatUtils.getToken();
-            String msg = EnterpriseWeChatUtils.makeTeamSendMsg(listPartyId, 
agentId, "hello world");
-            String resp = EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", 
msg, token);
-
-            String errmsg = JSON.parseObject(resp).getString("errmsg");
-            Assert.assertEquals("ok",errmsg);
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
+    public void tesMakeUserSendMsg1(){
+        String sendMsg = 
EnterpriseWeChatUtils.makeUserSendMsg(enterpriseWechatUsers, 
enterpriseWechatAgentId, msg);
+        Assert.assertTrue(sendMsg.contains(enterpriseWechatUsers));
+        Assert.assertTrue(sendMsg.contains(enterpriseWechatAgentId));
+        Assert.assertTrue(sendMsg.contains(msg));
     }
 
     @Test
-    public void testSendSingleUserWeChat() {
-        try {
-            String token = EnterpriseWeChatUtils.getToken();
-            String msg = 
EnterpriseWeChatUtils.makeUserSendMsg(listUserId.stream().findFirst().get(), 
agentId, "your meeting room has been booked and will be synced to the 'mailbox' 
later \n" +
-                    ">**matter details** \n" +
-                    ">matter:<font color='info'>meeting</font> <br>" +
-                    ">organizer:@miglioguan \n" +
-                    
">participant:@miglioguan、@kunliu、@jamdeezhou、@kanexiong、@kisonwang \n" +
-                    "> \n" +
-                    ">meeting room:<font color='info'>Guangzhou TIT 1st Floor 
301</font> \n" +
-                    ">date:<font color='warning'>May 18, 2018</font> \n" +
-                    ">time:<font color='comment'>9:00-11:00 am</font> \n" +
-                    "> \n" +
-                    ">please attend the meeting on time\n" +
-                    "> \n" +
-                    ">to modify the meeting information, please click: [Modify 
Meeting Information](https://work.weixin.qq.com)\"");
-
-            String resp = EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", 
msg, token);
-
-            String errmsg = JSON.parseObject(resp).getString("errmsg");
-            Assert.assertEquals("ok",errmsg);
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
+    public void tesMakeUserSendMsg2(){
+        List<String> users = new ArrayList<>();
+        users.add("user1");
+        users.add("user2");
+
+        String sendMsg = EnterpriseWeChatUtils.makeUserSendMsg(users, 
enterpriseWechatAgentId, msg);
+        Assert.assertTrue(sendMsg.contains(users.get(0)));
+        Assert.assertTrue(sendMsg.contains(users.get(1)));
+        Assert.assertTrue(sendMsg.contains(enterpriseWechatAgentId));
+        Assert.assertTrue(sendMsg.contains(msg));
     }
 
     @Test
-    public void testSendMultiUserWeChat() {
-        try {
-            String token = EnterpriseWeChatUtils.getToken();
-
-            String msg = EnterpriseWeChatUtils.makeUserSendMsg(listUserId, 
agentId, "hello world");
-            String resp = EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", 
msg, token);
-
-            String errmsg = JSON.parseObject(resp).getString("errmsg");
-            Assert.assertEquals("ok",errmsg);
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
+    public void testMarkdownByAlertForText(){
+        Alert alertForText = createAlertForText();
+        String result = EnterpriseWeChatUtils.markdownByAlert(alertForText);
+        Assert.assertNotNull(result);
     }
 
+    @Test
+    public void testMarkdownByAlertForTable(){
+        Alert alertForText = createAlertForTable();
+        String result = EnterpriseWeChatUtils.markdownByAlert(alertForText);
+        Assert.assertNotNull(result);
+    }
+
+    private Alert createAlertForText(){
+        String content ="[\"id:69\"," +
+                "\"name:UserBehavior-0--1193959466\"," +
+                "\"Job name: Start workflow\"," +
+                "\"State: SUCCESS\"," +
+                "\"Recovery:NO\"," +
+                "\"Run time: 1\"," +
+                "\"Start time: 2018-08-06 10:31:34.0\"," +
+                "\"End time: 2018-08-06 10:31:49.0\"," +
+                "\"Host: 192.168.xx.xx\"," +
+                "\"Notify group :4\"]";
+
+        Alert alert = new Alert();
+        alert.setTitle("Mysql Exception");
+        alert.setShowType(ShowType.TEXT);
+        alert.setContent(content);
+        alert.setAlertType(AlertType.EMAIL);
+        alert.setAlertGroupId(4);
+
+        return alert;
+    }
+
+    private String list2String(){
+
+        LinkedHashMap<String, Object> map1 = new LinkedHashMap<>();
+        map1.put("mysql service name","mysql200");
+        map1.put("mysql address","192.168.xx.xx");
+        map1.put("port","3306");
+        map1.put("no index of number","80");
+        map1.put("database client connections","190");
+
+        LinkedHashMap<String, Object> map2 = new LinkedHashMap<>();
+        map2.put("mysql service name","mysql210");
+        map2.put("mysql address","192.168.xx.xx");
+        map2.put("port", "3306");
+        map2.put("no index of number", "10");
+        map2.put("database client connections", "90");
+
+        List<LinkedHashMap<String, Object>> maps = new ArrayList<>();
+        maps.add(0, map1);
+        maps.add(1, map2);
+        String mapjson = JSONUtils.toJsonString(maps);
+        return mapjson;
+    }
+
+    private Alert createAlertForTable(){
+        Alert alert = new Alert();
+        alert.setTitle("Mysql Exception");
+        alert.setShowType(ShowType.TABLE);
+        String content= list2String();
+        alert.setContent(content);
+        alert.setAlertType(AlertType.EMAIL);
+        alert.setAlertGroupId(1);
+        return alert;
+    }
+
+
+
+
+//    @Test
+//    public void testSendSingleTeamWeChat() {
+//        try {
+//            String token = EnterpriseWeChatUtils.getToken();
+//            String msg = EnterpriseWeChatUtils.makeTeamSendMsg(partyId, 
agentId, "hello world");
+//            String resp = 
EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", msg, token);
+//
+//            String errmsg = JSON.parseObject(resp).getString("errmsg");
+//            Assert.assertEquals("ok",errmsg);
+//        } catch (IOException e) {
+//            e.printStackTrace();
+//        }
+//    }
+//
+//    @Test
+//    public void testSendMultiTeamWeChat() {
+//
+//        try {
+//            String token = EnterpriseWeChatUtils.getToken();
+//            String msg = EnterpriseWeChatUtils.makeTeamSendMsg(listPartyId, 
agentId, "hello world");
+//            String resp = 
EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", msg, token);
+//
+//            String errmsg = JSON.parseObject(resp).getString("errmsg");
+//            Assert.assertEquals("ok",errmsg);
+//        } catch (IOException e) {
+//            e.printStackTrace();
+//        }
+//    }
+//
+//    @Test
+//    public void testSendSingleUserWeChat() {
+//        try {
+//            String token = EnterpriseWeChatUtils.getToken();
+//            String msg = 
EnterpriseWeChatUtils.makeUserSendMsg(listUserId.stream().findFirst().get(), 
agentId, "your meeting room has been booked and will be synced to the 'mailbox' 
later \n" +
+//                    ">**matter details** \n" +
+//                    ">matter:<font color='info'>meeting</font> <br>" +
+//                    ">organizer:@miglioguan \n" +
+//                    
">participant:@miglioguan、@kunliu、@jamdeezhou、@kanexiong、@kisonwang \n" +
+//                    "> \n" +
+//                    ">meeting room:<font color='info'>Guangzhou TIT 1st 
Floor 301</font> \n" +
+//                    ">date:<font color='warning'>May 18, 2018</font> \n" +
+//                    ">time:<font color='comment'>9:00-11:00 am</font> \n" +
+//                    "> \n" +
+//                    ">please attend the meeting on time\n" +
+//                    "> \n" +
+//                    ">to modify the meeting information, please click: 
[Modify Meeting Information](https://work.weixin.qq.com)\"");
+//
+//            String resp = 
EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", msg, token);
+//
+//            String errmsg = JSON.parseObject(resp).getString("errmsg");
+//            Assert.assertEquals("ok",errmsg);
+//        } catch (IOException e) {
+//            e.printStackTrace();
+//        }
+//    }
+//
+//    @Test
+//    public void testSendMultiUserWeChat() {
+//        try {
+//            String token = EnterpriseWeChatUtils.getToken();
+//
+//            String msg = EnterpriseWeChatUtils.makeUserSendMsg(listUserId, 
agentId, "hello world");
+//            String resp = 
EnterpriseWeChatUtils.sendEnterpriseWeChat("utf-8", msg, token);
+//
+//            String errmsg = JSON.parseObject(resp).getString("errmsg");
+//            Assert.assertEquals("ok",errmsg);
+//        } catch (IOException e) {
+//            e.printStackTrace();
+//        }
+//    }
+
 }

Reply via email to