diff --git a/CHANGELOG.md b/CHANGELOG.md index 7d293bab..c7075f87 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -38,3 +38,4 @@ - [docs:update docs](https://github.com/Tencent/spring-cloud-tencent/pull/378) - [docs:optimize example](https://github.com/Tencent/spring-cloud-tencent/pull/385) - [Feature: format code](https://github.com/Tencent/spring-cloud-tencent/pull/394) +- [test: add PostInitPolarisSDKContextTest](https://github.com/Tencent/spring-cloud-tencent/pull/397) 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 00000000..13a0f6f7 --- /dev/null +++ b/spring-cloud-tencent-polaris-context/src/test/java/com/tencent/cloud/polaris/context/PostInitPolarisSDKContextTest.java @@ -0,0 +1,50 @@ +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; + + +/** + * Test for {@link PostInitPolarisSDKContext}. + * + * @author wh + */ +@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)); + } +}