Account for websocket-resources changes.

// FREEBIE
This commit is contained in:
Moxie Marlinspike
2014-11-15 09:48:09 -08:00
parent fdb35d4f77
commit 2525304215
2 changed files with 3 additions and 4 deletions

View File

@@ -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);