From 232ee3281d597313ed46b4b7c55275fa91bf9ef6 Mon Sep 17 00:00:00 2001 From: lepdou Date: Mon, 14 Nov 2022 11:15:20 +0800 Subject: [PATCH] fix testGetServices ut --- .../polaris/discovery/PolarisDiscoveryClientTest.java | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/discovery/PolarisDiscoveryClientTest.java b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/discovery/PolarisDiscoveryClientTest.java index 41da53413..bfd67879c 100644 --- a/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/discovery/PolarisDiscoveryClientTest.java +++ b/spring-cloud-starter-tencent-polaris-discovery/src/test/java/com/tencent/cloud/polaris/discovery/PolarisDiscoveryClientTest.java @@ -32,7 +32,6 @@ import static com.tencent.polaris.test.common.Consts.SERVICE_PROVIDER; import static java.util.Collections.singletonList; import static org.assertj.core.api.Assertions.assertThat; import static org.mockito.ArgumentMatchers.anyString; -import static org.mockito.Mockito.doReturn; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; @@ -63,14 +62,12 @@ public class PolarisDiscoveryClientTest { @Test public void testGetServices() { - -// when(polarisServiceDiscovery.getServices()).thenReturn(singletonList(SERVICE_PROVIDER)); - doReturn(singletonList(SERVICE_PROVIDER)).when(polarisServiceDiscovery).getServices(); + when(polarisServiceDiscovery.getServices()).thenReturn(singletonList(SERVICE_PROVIDER)); List services = client.getServices(); - assertThat(services).contains(SERVICE_PROVIDER).size().isEqualTo(1); - + assertThat(services).size().isEqualTo(1); + assertThat(services.contains(SERVICE_PROVIDER)).isTrue(); } @Test