Examples with UserSessionModel org.keycloak.models.UserSessionModel used on opensource projects

Example 61 with UserSessionModel

use of org.keycloak.models.UserSessionModel in project keycloak by keycloak.

the class OfflineSessionPersistenceTest method createOfflineSession.

private UserSessionModel createOfflineSession(KeycloakSession session, RealmModel realm, String userId, int sessionIndex) {
    final UserModel user = session.users().getUserById(realm, userId);
    UserSessionModel us = session.sessions().createUserSession(realm, user, "un" + sessionIndex, "ip1", "auth", false, null, null);
    return session.sessions().createOfflineUserSession(us);
}

Example 62 with UserSessionModel

use of org.keycloak.models.UserSessionModel in project keycloak by keycloak.

the class UserSessionPersisterProviderTest method loadPersistedSessionsPaginated.

private List<UserSessionModel> loadPersistedSessionsPaginated(KeycloakSession session, boolean offline, int sessionsPerPage, int expectedPageCount, int expectedSessionsCount) {
    UserSessionPersisterProvider persister = session.getProvider(UserSessionPersisterProvider.class);
    int count = persister.getUserSessionsCount(offline);
    int pageCount = 0;
    boolean next = true;
    List<UserSessionModel> result = new ArrayList<>();
    String lastSessionId = "00000000-0000-0000-0000-000000000000";
    while (next) {
        List<UserSessionModel> sess = persister.loadUserSessionsStream(0, sessionsPerPage, offline, lastSessionId).collect(Collectors.toList());
        if (sess.size() < sessionsPerPage) {
            next = false;
            // We had at least some session
            if (sess.size() > 0) {
                pageCount++;
            }
        } else {
            pageCount++;
            UserSessionModel lastSession = sess.get(sess.size() - 1);
            lastSessionId = lastSession.getId();
        }
        result.addAll(sess);
    }
    Assert.assertEquals(expectedPageCount, pageCount);
    Assert.assertEquals(expectedSessionsCount, result.size());
    return result;
}

Example 63 with UserSessionModel

use of org.keycloak.models.UserSessionModel in project keycloak by keycloak.

the class UserSessionPersisterProviderTest method testOnRealmRemoved.

@Test
public void testOnRealmRemoved() {
    AtomicReference<String> userSessionID = new AtomicReference<>();
    inComittedTransaction(session -> {
        RealmModel fooRealm = session.realms().createRealm("foo", "foo");
        fooRealm.setDefaultRole(session.roles().addRealmRole(fooRealm, Constants.DEFAULT_ROLES_ROLE_PREFIX + "-" + fooRealm.getName()));
        fooRealm.addClient("foo-app");
        session.users().addUser(fooRealm, "user3");
        UserSessionModel userSession = session.sessions().createUserSession(fooRealm, session.users().getUserByUsername(fooRealm, "user3"), "user3", "127.0.0.1", "form", true, null, null);
        userSessionID.set(userSession.getId());
        createClientSession(session, realmId, fooRealm.getClientByClientId("foo-app"), userSession, "http://redirect", "state");
    });
    inComittedTransaction(session -> {
        // Persist offline session
        RealmModel fooRealm = session.realms().getRealm("foo");
        UserSessionModel userSession = session.sessions().getUserSession(fooRealm, userSessionID.get());
        persistUserSession(session, userSession, true);
    });
    inComittedTransaction(session -> {
        // Assert session was persisted
        loadPersistedSessionsPaginated(session, true, 10, 1, 1);
        // Remove realm
        RealmManager realmMgr = new RealmManager(session);
        realmMgr.removeRealm(realmMgr.getRealm("foo"));
    });
    inComittedTransaction(session -> {
        // Assert nothing loaded
        loadPersistedSessionsPaginated(session, true, 10, 0, 0);
    });
}

Example 64 with UserSessionModel

use of org.keycloak.models.UserSessionModel in project keycloak by keycloak.

the class UserSessionPersisterProviderTest method createSessions.

protected static UserSessionModel[] createSessions(KeycloakSession session, String realmId) {
    RealmModel realm = session.realms().getRealm(realmId);
    UserSessionModel[] sessions = new UserSessionModel[3];
    sessions[0] = session.sessions().createUserSession(realm, session.users().getUserByUsername(realm, "user1"), "user1", "127.0.0.1", "form", true, null, null);
    createClientSession(session, realmId, realm.getClientByClientId("test-app"), sessions[0], "http://redirect", "state");
    createClientSession(session, realmId, realm.getClientByClientId("third-party"), sessions[0], "http://redirect", "state");
    sessions[1] = session.sessions().createUserSession(realm, session.users().getUserByUsername(realm, "user1"), "user1", "127.0.0.2", "form", true, null, null);
    createClientSession(session, realmId, realm.getClientByClientId("test-app"), sessions[1], "http://redirect", "state");
    sessions[2] = session.sessions().createUserSession(realm, session.users().getUserByUsername(realm, "user2"), "user2", "127.0.0.3", "form", true, null, null);
    createClientSession(session, realmId, realm.getClientByClientId("test-app"), sessions[2], "http://redirect", "state");
    return sessions;
}

Example 65 with UserSessionModel

use of org.keycloak.models.UserSessionModel in project keycloak by keycloak.

the class UserSessionInitializerTest method testUserSessionInitializer.

@Test
public void testUserSessionInitializer() {
    UserSessionModel[] origSessionIds = createSessionsInPersisterOnly();
    int started = origSessionIds[0].getStarted();
    inComittedTransaction(session -> {
        RealmModel realm = session.realms().getRealm(realmId);
        // Assert sessions are in
        ClientModel testApp = realm.getClientByClientId("test-app");
        ClientModel thirdparty = realm.getClientByClientId("third-party");
        assertThat("Count of offline sesions for client 'test-app'", session.sessions().getOfflineSessionsCount(realm, testApp), is((long) 3));
        assertThat("Count of offline sesions for client 'third-party'", session.sessions().getOfflineSessionsCount(realm, thirdparty), is((long) 1));
        List<UserSessionModel> loadedSessions = session.sessions().getOfflineUserSessionsStream(realm, testApp, 0, 10).collect(Collectors.toList());
        assertSessionLoaded(loadedSessions, origSessionIds[0].getId(), session.users().getUserByUsername(realm, "user1"), "127.0.0.1", started, started, "test-app", "third-party");
        assertSessionLoaded(loadedSessions, origSessionIds[1].getId(), session.users().getUserByUsername(realm, "user1"), "127.0.0.2", started, started, "test-app");
        assertSessionLoaded(loadedSessions, origSessionIds[2].getId(), session.users().getUserByUsername(realm, "user2"), "127.0.0.3", started, started, "test-app");
    });
}