From 7166140eefb0e2b7f1618b966c8c2ed3e38db94a Mon Sep 17 00:00:00 2001
From: lucian <author@corpmail.com>
Date: Tue, 5 Jan 2021 14:45:34 +0800
Subject: [PATCH] =?UTF-8?q?REPORT-46495=20=E6=96=B0=E7=89=88=E6=9C=AC?=
 =?UTF-8?q?=E6=8F=92=E4=BB=B6=E7=AE=A1=E7=90=86=EF=BC=8C=E5=9C=A8=E8=AE=BE?=
 =?UTF-8?q?=E8=AE=A1=E5=99=A8=E8=AF=AD=E8=A8=80=E4=B8=BA=E4=B8=AD=E6=96=87?=
 =?UTF-8?q?=E6=97=B6=E4=BC=9A=E6=98=BE=E7=A4=BA=E8=AF=AD=E8=A8=80=E7=AD=9B?=
 =?UTF-8?q?=E9=80=89=E7=9A=84=E6=A0=87=E7=AD=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../main/java/com/fr/design/upm/UpmUtils.java |  2 +
 .../java/com/fr/design/upm/UpmUtilsTest.java  | 57 +++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 designer-base/src/test/java/com/fr/design/upm/UpmUtilsTest.java

diff --git a/designer-base/src/main/java/com/fr/design/upm/UpmUtils.java b/designer-base/src/main/java/com/fr/design/upm/UpmUtils.java
index d27b6721a..e97a5826a 100644
--- a/designer-base/src/main/java/com/fr/design/upm/UpmUtils.java
+++ b/designer-base/src/main/java/com/fr/design/upm/UpmUtils.java
@@ -3,6 +3,7 @@ package com.fr.design.upm;
 import com.fr.common.annotations.Negative;
 import com.fr.config.ServerPreferenceConfig;
 import com.fr.general.CloudCenter;
+import com.fr.general.GeneralContext;
 import com.fr.stable.StringUtils;
 
 import java.util.ArrayList;
@@ -31,6 +32,7 @@ public class UpmUtils {
         Map<String, String> map4Tpl = new HashMap<>();
         map4Tpl.put("version", ServerPreferenceConfig.getInstance().getOptimizedUPMVersion());
         map4Tpl.put("new_version", fetchLatestVersion());
+        map4Tpl.put("language", GeneralContext.getLocale().toString());
         return map4Tpl;
     }
 
diff --git a/designer-base/src/test/java/com/fr/design/upm/UpmUtilsTest.java b/designer-base/src/test/java/com/fr/design/upm/UpmUtilsTest.java
new file mode 100644
index 000000000..31119d049
--- /dev/null
+++ b/designer-base/src/test/java/com/fr/design/upm/UpmUtilsTest.java
@@ -0,0 +1,57 @@
+package com.fr.design.upm;
+
+import com.fr.config.ServerPreferenceConfig;
+import com.fr.general.CloudCenter;
+import com.fr.general.GeneralContext;
+import org.easymock.EasyMock;
+import org.junit.Assert;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
+import org.powermock.modules.junit4.PowerMockRunner;
+
+import java.util.Locale;
+import java.util.Map;
+
+/**
+ * @author Lucian.Chen
+ * @version 10.0
+ * Created by Lucian.Chen on 2021/1/5
+ */
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({ServerPreferenceConfig.class, CloudCenter.class})
+@SuppressStaticInitializationFor("com.fr.general.CloudCenter")
+public class UpmUtilsTest {
+
+    @Test
+    public void testRenderMap() {
+
+        ServerPreferenceConfig serverPreferenceConfig = EasyMock.mock(ServerPreferenceConfig.class);
+        EasyMock.expect(serverPreferenceConfig.getOptimizedUPMVersion()).andReturn("1.0").anyTimes();
+        PowerMock.mockStatic(ServerPreferenceConfig.class);
+        EasyMock.expect(ServerPreferenceConfig.getInstance()).andReturn(serverPreferenceConfig).anyTimes();
+
+        CloudCenter cloudCenter = EasyMock.mock(CloudCenter.class);
+        EasyMock.expect(cloudCenter.acquireUrlByKind("upm.script.version")).andReturn("2.0").anyTimes();
+        PowerMock.mockStatic(CloudCenter.class);
+        EasyMock.expect(CloudCenter.getInstance()).andReturn(cloudCenter).anyTimes();
+
+        EasyMock.replay(serverPreferenceConfig, cloudCenter);
+        PowerMock.replayAll();
+
+        GeneralContext.setLocale(Locale.CHINA);
+
+        Map<String, String> map4Tpl = UpmUtils.renderMap();
+        Assert.assertEquals(map4Tpl.size(), 3);
+        Assert.assertEquals(map4Tpl.get("version"), "1.0");
+        Assert.assertEquals(map4Tpl.get("new_version"), "2.0");
+        Assert.assertEquals(map4Tpl.get("language"), "zh_CN");
+
+
+        EasyMock.verify(serverPreferenceConfig, cloudCenter);
+        PowerMock.verifyAll();
+
+    }
+}