Skip to content

Commit c8da34b

Browse files
committed
Fix test
Signed-off-by: Zixuan Liu <nodeces@gmail.com>
1 parent 4790279 commit c8da34b

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyIdleTimeoutTest.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import static java.util.concurrent.TimeUnit.SECONDS;
2222
import static org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
2323
import static org.assertj.core.api.Assertions.assertThat;
24+
import static org.mockito.ArgumentMatchers.anyBoolean;
2425
import static org.mockito.ArgumentMatchers.anyInt;
2526
import static org.mockito.ArgumentMatchers.anyString;
2627
import static org.mockito.Mockito.doReturn;
@@ -65,7 +66,7 @@ public void setup() throws Exception {
6566
config.setWebSocketSessionIdleTimeoutMillis(3 * 1000);
6667
service = spyWithClassAndConstructorArgs(WebSocketService.class, config);
6768
doReturn(new ZKMetadataStore(mockZooKeeperGlobal)).when(service)
68-
.createConfigMetadataStore(anyString(), anyInt());
69+
.createConfigMetadataStore(anyString(), anyInt(), anyBoolean());
6970
proxyServer = new ProxyServer(config);
7071
WebSocketServiceStarter.start(proxyServer, service);
7172
log.info("Proxy Server Started");

pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPingTest.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import static java.util.concurrent.TimeUnit.SECONDS;
2222
import static org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
2323
import static org.assertj.core.api.Assertions.assertThat;
24+
import static org.mockito.ArgumentMatchers.anyBoolean;
2425
import static org.mockito.ArgumentMatchers.anyInt;
2526
import static org.mockito.ArgumentMatchers.anyString;
2627
import static org.mockito.Mockito.doReturn;
@@ -67,7 +68,7 @@ public void setup() throws Exception {
6768
config.setWebSocketPingDurationSeconds(2);
6869
service = spyWithClassAndConstructorArgs(WebSocketService.class, config);
6970
doReturn(new ZKMetadataStore(mockZooKeeperGlobal)).when(service)
70-
.createConfigMetadataStore(anyString(), anyInt());
71+
.createConfigMetadataStore(anyString(), anyInt(), anyBoolean());
7172
proxyServer = new ProxyServer(config);
7273
WebSocketServiceStarter.start(proxyServer, service);
7374
log.info("Proxy Server Started");

0 commit comments

Comments
 (0)