From fac6ecbee5fa03edc89a3e978c272b6a9a3afd20 Mon Sep 17 00:00:00 2001 From: weihu Date: Wed, 6 Jul 2022 13:35:49 +0800 Subject: [PATCH] add PostInitPolarisSDKContextTest --- .../PostInitPolarisSDKContextTest.java | 51 +++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 spring-cloud-tencent-polaris-context/src/test/java/com/tencent/cloud/polaris/context/PostInitPolarisSDKContextTest.java diff --git a/spring-cloud-tencent-polaris-context/src/test/java/com/tencent/cloud/polaris/context/PostInitPolarisSDKContextTest.java b/spring-cloud-tencent-polaris-context/src/test/java/com/tencent/cloud/polaris/context/PostInitPolarisSDKContextTest.java new file mode 100644 index 000000000..0a6d06c34 --- /dev/null +++ b/spring-cloud-tencent-polaris-context/src/test/java/com/tencent/cloud/polaris/context/PostInitPolarisSDKContextTest.java @@ -0,0 +1,51 @@ +package com.tencent.cloud.polaris.context; + +import com.tencent.cloud.common.metadata.StaticMetadataManager; +import com.tencent.polaris.api.plugin.common.ValueContext; +import com.tencent.polaris.api.plugin.route.LocationLevel; +import com.tencent.polaris.client.api.SDKContext; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import static org.hamcrest.MatcherAssert.assertThat; +import static org.hamcrest.Matchers.is; +import static org.mockito.Mockito.when; + + +/** + *@author : wh + *@date : 2022/7/6 11:41 + *@description: + */ +@RunWith(MockitoJUnitRunner.class) +public class PostInitPolarisSDKContextTest { + + @Mock + private SDKContext sdkContext; + @Mock + private StaticMetadataManager staticMetadataManager; + + private static final String region = "region"; + + private static final String zone = "zone"; + + private static final String campus = "campus"; + + @Test + public void PostInitPolarisSDKContextTest() { + ValueContext valueContext = new ValueContext(); + + when(sdkContext.getValueContext()).thenReturn(valueContext); + when(staticMetadataManager.getRegion()).thenReturn(region); + when(staticMetadataManager.getZone()).thenReturn(zone); + when(staticMetadataManager.getCampus()).thenReturn(campus); + + new PostInitPolarisSDKContext(sdkContext, staticMetadataManager); + assertThat(valueContext.getValue(LocationLevel.region.name()), is(region)); + assertThat(valueContext.getValue(LocationLevel.zone.name()), is(zone)); + assertThat(valueContext.getValue(LocationLevel.campus.name()), is(campus)); + } + +}