From f1e98db84f93dead8f7cbe262ccc3ae9414fc16a Mon Sep 17 00:00:00 2001
From: andrew shan <45474304+andrewshan@users.noreply.github.com>
Date: Wed, 19 Jun 2024 11:53:53 +0800
Subject: [PATCH] fix: fix test case unstable issues

---
 .../lossless/LosslessRegistryAspectTest.java      |  2 +-
 .../polaris/context/PolarisSDKContextManager.java | 15 +++++++++------
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/plugin/lossless/LosslessRegistryAspectTest.java b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/plugin/lossless/LosslessRegistryAspectTest.java
index 67d0e98af..e594f8a3d 100644
--- a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/plugin/lossless/LosslessRegistryAspectTest.java
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/plugin/lossless/LosslessRegistryAspectTest.java
@@ -141,7 +141,7 @@ public class LosslessRegistryAspectTest {
 				assertThat(OkHttpUtil.checkUrl(HOST, LOSSLESS_PORT_1, "/online", Collections.EMPTY_MAP)).isFalse();
 			}).doesNotThrowAnyException();
 			// delay register after 5s
-			Thread.sleep(5000);
+			Thread.sleep(10000);
 			PolarisServiceRegistry registry = context.getBean(PolarisServiceRegistry.class);
 			PolarisRegistration registration = context.getBean(PolarisRegistration.class);
 
diff --git a/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/PolarisSDKContextManager.java b/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/PolarisSDKContextManager.java
index 8d4fd852a..26fe59e6b 100644
--- a/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/PolarisSDKContextManager.java
+++ b/spring-cloud-tencent-polaris-context/src/main/java/com/tencent/cloud/polaris/context/PolarisSDKContextManager.java
@@ -22,6 +22,7 @@ import java.util.List;
 import java.util.Objects;
 
 import com.tencent.cloud.polaris.context.config.PolarisContextProperties;
+import com.tencent.polaris.api.config.Configuration;
 import com.tencent.polaris.api.control.Destroyable;
 import com.tencent.polaris.api.core.ConsumerAPI;
 import com.tencent.polaris.api.core.LosslessAPI;
@@ -229,9 +230,10 @@ public class PolarisSDKContextManager {
 					}
 				}
 				// init SDKContext
-				serviceSdkContext = SDKContext.initContextByConfig(properties.configuration(serviceModifierList,
+				Configuration configuration = properties.configuration(serviceModifierList,
 						() -> environment.getProperty("spring.cloud.client.ip-address"),
-						() -> environment.getProperty("spring.cloud.polaris.local-port", Integer.class, 0)));
+						() -> environment.getProperty("spring.cloud.polaris.local-port", Integer.class, 0));
+				serviceSdkContext = SDKContext.initContextByConfig(configuration);
 				serviceSdkContext.init();
 
 				// init ProviderAPI
@@ -269,7 +271,7 @@ public class PolarisSDKContextManager {
 						}
 					}
 				}));
-				LOG.info("create Polaris SDK context successfully. properties: {}, ", properties);
+				LOG.info("create Polaris SDK context successfully. properties: {}, configuration: {}", properties, configuration);
 			}
 			catch (Throwable throwable) {
 				LOG.error("create Polaris SDK context failed. properties: {}, ", properties, throwable);
@@ -289,14 +291,15 @@ public class PolarisSDKContextManager {
 		if (null == configSDKContext && CollectionUtils.isNotEmpty(configModifierList)) {
 			try {
 				// init config SDKContext
-				configSDKContext = SDKContext.initContextByConfig(properties.configuration(configModifierList,
+				Configuration configuration = properties.configuration(configModifierList,
 						() -> environment.getProperty("spring.cloud.client.ip-address"),
-						() -> environment.getProperty("spring.cloud.polaris.local-port", Integer.class, 0)));
+						() -> environment.getProperty("spring.cloud.polaris.local-port", Integer.class, 0));
+				configSDKContext = SDKContext.initContextByConfig(configuration);
 				configSDKContext.init();
 
 				// add shutdown hook
 				Runtime.getRuntime().addShutdownHook(new Thread(PolarisSDKContextManager::innerConfigDestroy));
-				LOG.info("create Polaris config SDK context successfully. properties: {}, ", properties);
+				LOG.info("create Polaris config SDK context successfully. properties: {}, configuration: {}", properties, configuration);
 			}
 			catch (Throwable throwable) {
 				LOG.error("create Polaris config SDK context failed. properties: {}, ", properties, throwable);