From aa6325779e252e702066386bfcdb2091f6f05054 Mon Sep 17 00:00:00 2001
From: Haotian Zhang <928016560@qq.com>
Date: Tue, 1 Aug 2023 19:07:54 +0800
Subject: [PATCH] fix:add deregister judgment.

---
 .../PolarisAutoServiceRegistrationTest.java   | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisAutoServiceRegistrationTest.java b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisAutoServiceRegistrationTest.java
index 9d78c9033..e1c04a29b 100644
--- a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisAutoServiceRegistrationTest.java
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/registry/PolarisAutoServiceRegistrationTest.java
@@ -71,6 +71,7 @@ public class PolarisAutoServiceRegistrationTest {
 	@BeforeEach
 	void setUp() {
 		doNothing().when(serviceRegistry).register(nullable(PolarisRegistration.class));
+		doNothing().when(serviceRegistry).deregister(nullable(PolarisRegistration.class));
 
 		polarisAutoServiceRegistration =
 				new PolarisAutoServiceRegistration(serviceRegistry, autoServiceRegistrationProperties, registration,
@@ -111,6 +112,32 @@ public class PolarisAutoServiceRegistrationTest {
 		}).doesNotThrowAnyException();
 	}
 
+	@Test
+	public void testDeregister() {
+		doReturn(false).when(registration).isRegisterEnabled();
+		assertThatCode(() -> {
+			polarisAutoServiceRegistration.registerManagement();
+		}).doesNotThrowAnyException();
+
+		doReturn(true).when(registration).isRegisterEnabled();
+		assertThatCode(() -> {
+			polarisAutoServiceRegistration.deregister();
+		}).doesNotThrowAnyException();
+	}
+
+	@Test
+	public void testDeregisterManagement() {
+		doReturn(false).when(registration).isRegisterEnabled();
+		assertThatCode(() -> {
+			polarisAutoServiceRegistration.registerManagement();
+		}).doesNotThrowAnyException();
+
+		doReturn(true).when(registration).isRegisterEnabled();
+		assertThatCode(() -> {
+			polarisAutoServiceRegistration.deregisterManagement();
+		}).doesNotThrowAnyException();
+	}
+
 	@Test
 	public void testGetAppName() {
 		doReturn("application").when(environment).getProperty(anyString(), anyString());