feat:issue 818

3 years ago · 15e110915e
parent e44a70c2c9
commit 15e110915e
1 changed files with 4 additions and 0 deletions
--- a/spring-cloud-tencent-rpc-enhancement/src/test/java/com/tencent/cloud/rpc/enhancement/AbstractPolarisReporterAdapterTest.java
+++ b/spring-cloud-tencent-rpc-enhancement/src/test/java/com/tencent/cloud/rpc/enhancement/AbstractPolarisReporterAdapterTest.java
@ -125,6 +125,10 @@ public class AbstractPolarisReporterAdapterTest {
 		headers.set(HeaderConstant.INTERNAL_CALLEE_RET_STATUS, RetStatus.RetFlowControl.getDesc());
 		ret = adapter.getRetStatusFromRequest(headers, RetStatus.RetFail);
 		Assertions.assertThat(ret).isEqualTo(RetStatus.RetFlowControl);
+
+		headers.set(HeaderConstant.INTERNAL_CALLEE_RET_STATUS, RetStatus.RetReject.getDesc());
+		ret = adapter.getRetStatusFromRequest(headers, RetStatus.RetFail);
+		Assertions.assertThat(ret).isEqualTo(RetStatus.RetReject);
 	}

 	@Test