mirror of
https://github.com/signalapp/Signal-Server
synced 2026-04-25 00:18:04 +01:00
Account for websocket-resources changes.
// FREEBIE
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
package org.whispersystems.textsecuregcm.tests.websocket;
|
||||
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.google.common.base.Optional;
|
||||
import com.google.common.util.concurrent.SettableFuture;
|
||||
import org.eclipse.jetty.websocket.api.UpgradeRequest;
|
||||
@@ -79,7 +78,7 @@ public class WebSocketConnectionTest {
|
||||
}});
|
||||
|
||||
Optional<Account> account = webSocketAuthenticator.authenticate(upgradeRequest);
|
||||
when(sessionContext.getAuthenticated()).thenReturn(account.orNull());
|
||||
when(sessionContext.getAuthenticated(Account.class)).thenReturn(account);
|
||||
|
||||
connectListener.onWebSocketConnect(sessionContext);
|
||||
|
||||
@@ -98,7 +97,7 @@ public class WebSocketConnectionTest {
|
||||
}});
|
||||
|
||||
account = webSocketAuthenticator.authenticate(upgradeRequest);
|
||||
when(sessionContext.getAuthenticated()).thenReturn(account.orNull());
|
||||
when(sessionContext.getAuthenticated(Account.class)).thenReturn(account);
|
||||
|
||||
WebSocketClient client = mock(WebSocketClient.class);
|
||||
when(sessionContext.getClient()).thenReturn(client);
|
||||
|
||||
Reference in New Issue
Block a user