Examples with SessionListeners io.undertow.server.session.SessionListeners used on opensource projects

Example 11 with SessionListeners

use of io.undertow.server.session.SessionListeners in project wildfly by wildfly.

the class DistributableSessionTestCase method setAttribute.

@Test
public void setAttribute() {
    when(this.session.getMetaData()).thenReturn(this.metaData);
    when(this.metaData.isNew()).thenReturn(false);
    io.undertow.server.session.Session session = new DistributableSession(this.manager, this.session, this.config, this.batch, this.closeTask);
    String name = "name";
    Integer value = 1;
    SessionManager<Map<String, Object>, Batch> manager = mock(SessionManager.class);
    Batcher<Batch> batcher = mock(Batcher.class);
    BatchContext context = mock(BatchContext.class);
    SessionAttributes attributes = mock(SessionAttributes.class);
    SessionListener listener = mock(SessionListener.class);
    SessionListeners listeners = new SessionListeners();
    listeners.addSessionListener(listener);
    Object expected = new Object();
    when(this.session.getAttributes()).thenReturn(attributes);
    when(attributes.setAttribute(name, value)).thenReturn(expected);
    when(this.manager.getSessionListeners()).thenReturn(listeners);
    when(this.manager.getSessionManager()).thenReturn(manager);
    when(manager.getBatcher()).thenReturn(batcher);
    when(batcher.resumeBatch(this.batch)).thenReturn(context);
    Object result = session.setAttribute(name, value);
    assertSame(expected, result);
    verify(listener, never()).attributeAdded(session, name, value);
    verify(listener).attributeUpdated(session, name, value, expected);
    verify(listener, never()).attributeRemoved(same(session), same(name), any());
    verify(context).close();
    verify(this.session, never()).close();
    verify(this.closeTask, never()).accept(null);
    reset(context);
    doThrow(IllegalStateException.class).when(this.session).getAttributes();
    assertThrows(IllegalStateException.class, () -> session.setAttribute(name, value));
    verify(context).close();
    verify(this.session).close();
    verify(this.closeTask).accept(null);
}