示例#1
0
  @OnOpen
  public void onOpen(Session session) {
    LOG.debug("@OnOpen: " + session.getId());
    thisSession = session;

    if (thisSession.getUserPrincipal() != null) {
      authToken = (UsernamePasswordAuthenticationToken) thisSession.getUserPrincipal();
      if (authToken.isAuthenticated()) {
        isHttpSessionValid = true;
        httpSession = (HttpSession) thisSession.getUserProperties().get("httpSession");
        httpSessionId = httpSession.getId();
        LOG.debug("HTTP Session id: " + httpSessionId);
        if (httpSession.getAttribute("CHAT_OPEN") != null) {
          activeClose(new CloseReason(CloseReason.CloseCodes.VIOLATED_POLICY, "ALREADY_OPEN"));
        } else {
          httpSession.setAttribute("CHAT_OPEN", true);
        }
        return;
      }
    }

    activeClose(new CloseReason(CloseReason.CloseCodes.VIOLATED_POLICY, "Unauthorized"));
  }
  @BeforeClass
  public static void init() {
    Mockito.when(master.getUserPrincipal()).thenReturn(principalMaster);
    Mockito.when(slave1.getUserPrincipal()).thenReturn(principal1);
    Mockito.when(slave2.getUserPrincipal()).thenReturn(principal2);

    Mockito.when(master.getUserPrincipal().getName()).thenReturn("master1");
    Mockito.when(slave1.getUserPrincipal().getName()).thenReturn("slave1");
    Mockito.when(slave2.getUserPrincipal().getName()).thenReturn("slave2");
  }