From 303b1c3724448c4f5ef4883e347f65e9ba6ca377 Mon Sep 17 00:00:00 2001 From: chuntaojun Date: Tue, 4 Apr 2023 10:44:31 +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 ec638d2ac..351c278ce 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 @@ -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