SessionRegistryTest.java 文件源码

@Test
public void fromBroker() throws IOException, EncodeException {
    final Message msg = mock(Message.class);
    when(msg.sessionId()).thenReturn("sessionId");
    final Frame frame = mock(Frame.class);
    when(msg.frame()).thenReturn(frame);
    when(frame.command()).thenReturn(Command.MESSAGE);
    final Session session = Mockito.mock(Session.class);
    getSessionMap().put("sessionId", session);
    getPrincipalSessionMap().put(NULL_PRINCIPLE, new HashSet<>(singleton(session)));
    final Basic basic = mock(Basic.class);
    when(session.getBasicRemote()).thenReturn(basic);

    this.registry.fromBroker(msg);

    verify(msg).sessionId();
    verify(msg).frame();
    verify(frame, times(2)).command();
    verify(this.log).debug("Sending message to client. [sessionId={},command={}]", "sessionId", Command.MESSAGE);
    verify(session).getBasicRemote();
    verify(basic).sendObject(frame);
    verifyNoMoreInteractions(msg, frame, session, basic);
}