fix testGetServices ut

pull/715/head
lepdou 3 years ago
parent b8520a93d4
commit 232ee3281d

@ -32,7 +32,6 @@ import static com.tencent.polaris.test.common.Consts.SERVICE_PROVIDER;
import static java.util.Collections.singletonList; import static java.util.Collections.singletonList;
import static org.assertj.core.api.Assertions.assertThat; import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.anyString; import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
@ -63,14 +62,12 @@ public class PolarisDiscoveryClientTest {
@Test @Test
public void testGetServices() { public void testGetServices() {
when(polarisServiceDiscovery.getServices()).thenReturn(singletonList(SERVICE_PROVIDER));
// when(polarisServiceDiscovery.getServices()).thenReturn(singletonList(SERVICE_PROVIDER));
doReturn(singletonList(SERVICE_PROVIDER)).when(polarisServiceDiscovery).getServices();
List<String> services = client.getServices(); List<String> services = client.getServices();
assertThat(services).contains(SERVICE_PROVIDER).size().isEqualTo(1); assertThat(services).size().isEqualTo(1);
assertThat(services.contains(SERVICE_PROVIDER)).isTrue();
} }
@Test @Test

Loading…
Cancel
Save