Redirect sofa runtime log to console in test env (#1031)

sofastack · Oct 26, 2022 · 620f9e0 · 620f9e0
1 parent 186ba07
commit 620f9e0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/...sofa-boot-core/log-sofa-boot/src/main/java/com/alipay/sofa/boot/logging/LocalEnvUtil.java b/...sofa-boot-core/log-sofa-boot/src/main/java/com/alipay/sofa/boot/logging/LocalEnvUtil.java
@@ -22,6 +22,7 @@
  */
 public class LocalEnvUtil {
     private static boolean LOCAL_ENV;
+    private static boolean TEST_ENV;
 
     static {
         // Currently supports for detection of IDEA IntelliJ
@@ -31,9 +32,23 @@ public class LocalEnvUtil {
         } catch (ClassNotFoundException e) {
             LOCAL_ENV = false;
         }
+        // Detection of test environment
+        StackTraceElement[] stackTrace = new RuntimeException().getStackTrace();
+        for (StackTraceElement stackTraceElement : stackTrace) {
+            if ("loadContext".equals(stackTraceElement.getMethodName())
+                && "org.springframework.boot.test.context.SpringBootContextLoader"
+                    .equals(stackTraceElement.getClassName())) {
+                TEST_ENV = true;
+                break;
+            }
+        }
     }
 
     public static boolean isLocalEnv() {
         return LOCAL_ENV;
     }
+
+    public static boolean isTestEnv() {
+        return TEST_ENV;
+    }
 }
diff --git a/...sofa-boot/src/main/java/com/alipay/sofa/boot/logging/LogEnvironmentPreparingListener.java b/...sofa-boot/src/main/java/com/alipay/sofa/boot/logging/LogEnvironmentPreparingListener.java
@@ -80,7 +80,7 @@ private void prepare(ConfigurableEnvironment environment) {
     }
 
     private void defaultConsoleLoggers() {
-        if (LocalEnvUtil.isLocalEnv()) {
+        if (LocalEnvUtil.isLocalEnv() || LocalEnvUtil.isTestEnv()) {
             CommonLoggingConfigurations.loadExternalConfiguration(
                 Constants.SOFA_MIDDLEWARE_ALL_LOG_CONSOLE_SWITCH, "true");
         }