55import com .google .protobuf .ByteString ;
66import org .eclipse .jetty .websocket .api .UpgradeRequest ;
77import org .junit .Test ;
8+ import org .mockito .ArgumentMatchers ;
89import org .mockito .invocation .InvocationOnMock ;
910import org .mockito .stubbing .Answer ;
1011import org .whispersystems .textsecuregcm .auth .AccountAuthenticator ;
@@ -130,7 +131,7 @@ public void testOpen() throws Exception {
130131 final List <SettableFuture <WebSocketResponseMessage >> futures = new LinkedList <>();
131132 final WebSocketClient client = mock (WebSocketClient .class );
132133
133- when (client .sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), any (List .class ), any (Optional . class )))
134+ when (client .sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), ArgumentMatchers . nullable (List .class ), ArgumentMatchers .< Optional < byte []>> any ()))
134135 .thenAnswer (new Answer <SettableFuture <WebSocketResponseMessage >>() {
135136 @ Override
136137 public SettableFuture <WebSocketResponseMessage > answer (InvocationOnMock invocationOnMock ) throws Throwable {
@@ -145,7 +146,7 @@ public SettableFuture<WebSocketResponseMessage> answer(InvocationOnMock invocati
145146 account , device , client );
146147
147148 connection .onDispatchSubscribed (websocketAddress .serialize ());
148- verify (client , times (3 )).sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), anyList ( ), any (Optional . class ));
149+ verify (client , times (3 )).sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), ArgumentMatchers . nullable ( List . class ), ArgumentMatchers .< Optional < byte []>> any ());
149150
150151 assertTrue (futures .size () == 3 );
151152
@@ -214,7 +215,7 @@ public void testOnlineSend() throws Exception {
214215 final List <SettableFuture <WebSocketResponseMessage >> futures = new LinkedList <>();
215216 final WebSocketClient client = mock (WebSocketClient .class );
216217
217- when (client .sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), anyList ( ), any (Optional . class )))
218+ when (client .sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), ArgumentMatchers . nullable ( List . class ), ArgumentMatchers .< Optional < byte []>> any ()))
218219 .thenAnswer (new Answer <SettableFuture <WebSocketResponseMessage >>() {
219220 @ Override
220221 public SettableFuture <WebSocketResponseMessage > answer (InvocationOnMock invocationOnMock ) throws Throwable {
@@ -239,7 +240,7 @@ public SettableFuture<WebSocketResponseMessage> answer(InvocationOnMock invocati
239240 .setContent (ByteString .copyFrom (secondMessage .toByteArray ()))
240241 .build ().toByteArray ());
241242
242- verify (client , times (2 )).sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), anyList ( ), any (Optional . class ));
243+ verify (client , times (2 )).sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), ArgumentMatchers . nullable ( List . class ), ArgumentMatchers .< Optional < byte []>> any ());
243244
244245 assertEquals (futures .size (), 2 );
245246
@@ -320,7 +321,7 @@ public void testPendingSend() throws Exception {
320321 final List <SettableFuture <WebSocketResponseMessage >> futures = new LinkedList <>();
321322 final WebSocketClient client = mock (WebSocketClient .class );
322323
323- when (client .sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), anyList ( ), any (Optional . class )))
324+ when (client .sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), ArgumentMatchers . nullable ( List . class ), ArgumentMatchers .< Optional < byte []>> any ()))
324325 .thenAnswer (new Answer <SettableFuture <WebSocketResponseMessage >>() {
325326 @ Override
326327 public SettableFuture <WebSocketResponseMessage > answer (InvocationOnMock invocationOnMock ) throws Throwable {
@@ -336,7 +337,7 @@ public SettableFuture<WebSocketResponseMessage> answer(InvocationOnMock invocati
336337
337338 connection .onDispatchSubscribed (websocketAddress .serialize ());
338339
339- verify (client , times (2 )).sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), anyList ( ), any (Optional . class ));
340+ verify (client , times (2 )).sendRequest (eq ("PUT" ), eq ("/api/v1/message" ), ArgumentMatchers . nullable ( List . class ), ArgumentMatchers .< Optional < byte []>> any ());
340341
341342 assertEquals (futures .size (), 2 );
342343
0 commit comments