From 4e76b9737e036c18e8a9e840dc443615ffcec348 Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期四, 20 十月 2022 12:20:18 +0800
Subject: [PATCH] 大气农耕

---
 ximon-admin/src/test/java/com/sandu/ximon/admin/base/BaseMockMvcRequestBuilders.java |   28 +++++++++++++++++++++-------
 1 files changed, 21 insertions(+), 7 deletions(-)

diff --git a/ximon-admin/src/test/java/com/sandu/ximon/admin/base/BaseMockMvcRequestBuilders.java b/ximon-admin/src/test/java/com/sandu/ximon/admin/base/BaseMockMvcRequestBuilders.java
index 9266cd2..b08b99f 100644
--- a/ximon-admin/src/test/java/com/sandu/ximon/admin/base/BaseMockMvcRequestBuilders.java
+++ b/ximon-admin/src/test/java/com/sandu/ximon/admin/base/BaseMockMvcRequestBuilders.java
@@ -4,11 +4,17 @@
 import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
 import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
 
+import java.util.HashMap;
+import java.util.Map;
+
 public class BaseMockMvcRequestBuilders {
     //鍏ㄥ眬鐨勮姹傚ご鏁版嵁
-   // public final static String TOKEN_KEY = "DeviceId";
-    //public final static String TOKEN = "123456";
-    public final static String TOKEN = "Bearer eyJhbGciOiJIUzUxMiJ9.eyJzdWIiOiJhZG1pbiIsImF1dGgiOiJhZG1pbiIsImNyZWQiOjAsImV4cCI6MTY0MzIxNTgwN30.b9hHNpLDe3XThtCr8alFl4DZlfWM1Rx2o_AvFnutUQEqnyuh0JDlYigcev8zk_tKuSPDpXtWVsdKKQqlKLRb3g";
+    public final static Map<String, Object> headers = new HashMap<>();
+
+    //閰嶇疆璇锋眰澶寸殑鏁版嵁锛堝寘鍚玹oken涔嬬被锛�
+    static {
+        headers.put("Authorization", "Bearer eyJhbGciOiJIUzUxMiJ9.eyJzdWIiOiJhZG1pbiIsImF1dGgiOiJhZG1pbiIsImNyZWQiOjAsImV4cCI6MTY0MzIxNTgwN30.b9hHNpLDe3XThtCr8alFl4DZlfWM1Rx2o_AvFnutUQEqnyuh0JDlYigcev8zk_tKuSPDpXtWVsdKKQqlKLRb3g");
+    }
 
     /**
      * post璇锋眰
@@ -17,9 +23,13 @@
      * @return
      */
     public static MockHttpServletRequestBuilder post(String url) {
-        return MockMvcRequestBuilders.post(url).contentType(MediaType.APPLICATION_JSON)//璇锋眰鐨勬暟鎹牸寮�
-                .header(TOKEN)
+        MockHttpServletRequestBuilder temp = MockMvcRequestBuilders.post(url).contentType(MediaType.APPLICATION_JSON)//璇锋眰鐨勬暟鎹牸寮�
                 .characterEncoding("UTF-8");//璁剧疆璇锋眰鐨勭紪鐮�
+        //璁剧疆璇锋眰澶存暟鎹�
+        for (String key : headers.keySet()) {
+            temp.header(key, headers.get(key));
+        }
+        return temp;
     }
 
     /**
@@ -29,8 +39,12 @@
      * @return
      */
     public static MockHttpServletRequestBuilder get(String url) {
-        return MockMvcRequestBuilders.get(url).contentType(MediaType.APPLICATION_JSON)//璇锋眰鐨勬暟鎹牸寮�
-                .header(TOKEN)
+        MockHttpServletRequestBuilder temp = MockMvcRequestBuilders.get(url).contentType(MediaType.APPLICATION_JSON)//璇锋眰鐨勬暟鎹牸寮�
                 .characterEncoding("UTF-8");//璁剧疆璇锋眰鐨勭紪鐮�
+        //璁剧疆璇锋眰澶存暟鎹�
+        for (String key : headers.keySet()) {
+            temp.header(key, headers.get(key));
+        }
+        return temp;
     }
 }

--
Gitblit v1.9.3