feat:issue 818

3 years ago · 303b1c3724
parent b1df53b4f7
commit 303b1c3724
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
@ -124,6 +124,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