From 15e110915e86bb8680abf38e9f05068fd6d1c3a9 Mon Sep 17 00:00:00 2001 From: chuntaojun Date: Tue, 4 Apr 2023 10:41:03 +0800 Subject: [PATCH] feat:issue 818 --- .../rpc/enhancement/AbstractPolarisReporterAdapterTest.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git 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 index afd5c27ab..b84d23d1f 100644 --- 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