Examples with ModelTest org.keycloak.testsuite.arquillian.annotation.ModelTest used on opensource projects

Example 76 with ModelTest

use of org.keycloak.testsuite.arquillian.annotation.ModelTest in project keycloak by keycloak.

the class UserSessionProviderOfflineTest method testOnUserRemoved.

@Test
@ModelTest
public void testOnUserRemoved(KeycloakSession session) {
    KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionUR) -> {
        try {
            int started = Time.currentTime();
            AtomicReference<String> userSessionID = new AtomicReference<>();
            KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionUR1) -> {
                currentSession = sessionUR1;
                RealmModel fooRealm = currentSession.realms().createRealm("foo", "foo");
                fooRealm.setDefaultRole(currentSession.roles().addRealmRole(fooRealm, Constants.DEFAULT_ROLES_ROLE_PREFIX + "-" + fooRealm.getName()));
                fooRealm.setSsoSessionIdleTimeout(1800);
                fooRealm.setSsoSessionMaxLifespan(36000);
                fooRealm.setOfflineSessionIdleTimeout(2592000);
                fooRealm.setOfflineSessionMaxLifespan(5184000);
                fooRealm.addClient("foo-app");
                currentSession.users().addUser(fooRealm, "user3");
                UserSessionModel userSession = currentSession.sessions().createUserSession(fooRealm, currentSession.users().getUserByUsername(fooRealm, "user3"), "user3", "127.0.0.1", "form", true, null, null);
                userSessionID.set(userSession.getId());
                createClientSession(currentSession, fooRealm.getClientByClientId("foo-app"), userSession, "http://redirect", "state");
            });
            KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionUR2) -> {
                currentSession = sessionUR2;
                // Create offline session
                RealmModel fooRealm = currentSession.realms().getRealm("foo");
                UserSessionModel userSession = currentSession.sessions().getUserSession(fooRealm, userSessionID.get());
                createOfflineSessionIncludeClientSessions(currentSession, userSession);
            });
            KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionUR3) -> {
                currentSession = sessionUR3;
                RealmManager realmMgr = new RealmManager(currentSession);
                RealmModel fooRealm = realmMgr.getRealm("foo");
                UserModel user3 = currentSession.users().getUserByUsername(fooRealm, "user3");
                // Assert session was persisted with both clientSessions
                UserSessionModel offlineSession = currentSession.sessions().getOfflineUserSession(fooRealm, userSessionID.get());
                assertSession(offlineSession, user3, "127.0.0.1", started, started, "foo-app");
            });
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionTearDown) -> {
                currentSession = sessionTearDown;
                RealmManager realmMgr = new RealmManager(currentSession);
                RealmModel fooRealm = realmMgr.getRealm("foo");
                UserModel user3 = currentSession.users().getUserByUsername(fooRealm, "user3");
                // Remove user3
                new UserManager(currentSession).removeUser(fooRealm, user3);
                // Cleanup
                realmMgr = new RealmManager(currentSession);
                realmMgr.removeRealm(realmMgr.getRealm("foo"));
            });
        }
    });
}

Example 77 with ModelTest

use of org.keycloak.testsuite.arquillian.annotation.ModelTest in project keycloak by keycloak.

the class MultipleRealmsTest method testUsers.

@Test
@ModelTest
public void testUsers(KeycloakSession session) {
    AtomicReference<UserModel> r1user1Atomic = new AtomicReference<>();
    KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionTestUser1) -> {
        KeycloakSession currentSession = sessionTestUser1;
        RealmModel realm1 = currentSession.realms().createRealm("id1", "realm1");
        RealmModel realm2 = currentSession.realms().createRealm("id2", "realm2");
        realm1.setDefaultRole(currentSession.roles().addRealmRole(realm1, Constants.DEFAULT_ROLES_ROLE_PREFIX + "-" + realm1.getName()));
        realm2.setDefaultRole(currentSession.roles().addRealmRole(realm2, Constants.DEFAULT_ROLES_ROLE_PREFIX + "-" + realm2.getName()));
        createObjects(currentSession, realm1);
        createObjects(currentSession, realm2);
        UserModel r1user1 = currentSession.users().getUserByUsername(realm1, "user1");
        UserModel r2user1 = currentSession.users().getUserByUsername(realm2, "user1");
        r1user1Atomic.set(r1user1);
        Assert.assertEquals(r1user1.getUsername(), r2user1.getUsername());
        Assert.assertNotEquals(r1user1.getId(), r2user1.getId());
        // Test password
        currentSession.userCredentialManager().updateCredential(realm1, r1user1, UserCredentialModel.password("pass1"));
        currentSession.userCredentialManager().updateCredential(realm2, r2user1, UserCredentialModel.password("pass2"));
        Assert.assertTrue(currentSession.userCredentialManager().isValid(realm1, r1user1, UserCredentialModel.password("pass1")));
        Assert.assertFalse(currentSession.userCredentialManager().isValid(realm1, r1user1, UserCredentialModel.password("pass2")));
        Assert.assertFalse(currentSession.userCredentialManager().isValid(realm2, r2user1, UserCredentialModel.password("pass1")));
        Assert.assertTrue(currentSession.userCredentialManager().isValid(realm2, r2user1, UserCredentialModel.password("pass2")));
        // Test searching
        Assert.assertEquals(2, currentSession.users().searchForUserStream(realm1, "user").count());
    });
    KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionTestUser2) -> {
        KeycloakSession currentSession = sessionTestUser2;
        RealmModel realm1 = currentSession.realms().getRealm("id1");
        RealmModel realm2 = currentSession.realms().getRealm("id2");
        UserModel r1user1 = r1user1Atomic.get();
        currentSession.users().removeUser(realm1, r1user1);
        UserModel user2 = currentSession.users().getUserByUsername(realm1, "user2");
        currentSession.users().removeUser(realm1, user2);
        Assert.assertEquals(0, currentSession.users().searchForUserStream(realm1, "user").count());
        Assert.assertEquals(2, currentSession.users().searchForUserStream(realm2, "user").count());
        UserModel user1 = currentSession.users().getUserByUsername(realm1, "user1");
        UserModel user1a = currentSession.users().getUserByUsername(realm2, "user1");
        UserManager um = new UserManager(currentSession);
        if (user1 != null) {
            um.removeUser(realm1, user1);
        }
        if (user1a != null) {
            um.removeUser(realm2, user1a);
        }
    });
    KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionTestUser3) -> {
        KeycloakSession currentSession = sessionTestUser3;
        currentSession.realms().removeRealm("id1");
        currentSession.realms().removeRealm("id2");
    });
}

Example 78 with ModelTest

use of org.keycloak.testsuite.arquillian.annotation.ModelTest in project keycloak by keycloak.

the class MultipleRealmsTest method testGetById.

@Test
@ModelTest
public void testGetById(KeycloakSession session) {
    KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession sessionById) -> {
        KeycloakSession currentSession = sessionById;
        RealmModel realm1 = currentSession.realms().createRealm("id1", "realm1");
        RealmModel realm2 = currentSession.realms().createRealm("id2", "realm2");
        realm1.setDefaultRole(currentSession.roles().addRealmRole(realm1, Constants.DEFAULT_ROLES_ROLE_PREFIX + "-" + realm1.getName()));
        realm2.setDefaultRole(currentSession.roles().addRealmRole(realm2, Constants.DEFAULT_ROLES_ROLE_PREFIX + "-" + realm2.getName()));
        createObjects(currentSession, realm1);
        createObjects(currentSession, realm2);
        Assert.assertEquals(realm1, currentSession.realms().getRealm("id1"));
        Assert.assertEquals(realm1, currentSession.realms().getRealmByName("realm1"));
        Assert.assertEquals(realm2, currentSession.realms().getRealm("id2"));
        Assert.assertEquals(realm2, currentSession.realms().getRealmByName("realm2"));
        ClientModel r1app1 = realm1.getClientByClientId("app1");
        Assert.assertNotNull(realm1.getClientByClientId("app2"));
        Assert.assertNotNull(realm2.getClientByClientId("app1"));
        Assert.assertNotNull(realm2.getClientByClientId("app2"));
        Assert.assertEquals(r1app1, realm1.getClientById(r1app1.getId()));
        Assert.assertNull(realm2.getClientById(r1app1.getId()));
        ClientModel r2cl1 = realm2.getClientByClientId("cl1");
        Assert.assertEquals(r2cl1.getId(), realm2.getClientById(r2cl1.getId()).getId());
        Assert.assertNull(realm1.getClientByClientId(r2cl1.getId()));
        RoleModel r1App1Role = r1app1.getRole("app1Role1");
        Assert.assertEquals(r1App1Role, realm1.getRoleById(r1App1Role.getId()));
        Assert.assertNull(realm2.getRoleById(r1App1Role.getId()));
        RoleModel r2Role1 = realm2.getRole("role2");
        Assert.assertNull(realm1.getRoleById(r2Role1.getId()));
        Assert.assertEquals(r2Role1, realm2.getRoleById(r2Role1.getId()));
        UserModel user1 = currentSession.users().getUserByUsername(realm1, "user1");
        UserModel user1a = currentSession.users().getUserByUsername(realm2, "user1");
        UserManager um = new UserManager(currentSession);
        if (user1 != null) {
            um.removeUser(realm1, user1);
        }
        if (user1a != null) {
            um.removeUser(realm2, user1a);
        }
        currentSession.realms().removeRealm("id1");
        currentSession.realms().removeRealm("id2");
    });
}

Example 79 with ModelTest

use of org.keycloak.testsuite.arquillian.annotation.ModelTest in project keycloak by keycloak.

the class SimpleModelTest method simpleModelTestWithAssertionError.

// Just for the test that AssertionError is correctly propagated
@Test(expected = AssertionError.class)
@ModelTest
public void simpleModelTestWithAssertionError(KeycloakSession session) {
    log.infof("simpleModelTestWithAssertionError");
    RealmModel realm = session.realms().getRealmByName("masterr");
    // This should fail and throw the AssertionError
    Assert.assertNotNull("Master realm was not found!", realm);
}

Example 80 with ModelTest

use of org.keycloak.testsuite.arquillian.annotation.ModelTest in project keycloak by keycloak.

the class UserConsentWithUserStorageModelTest method getAllConsentTest.

@Test
@ModelTest
public void getAllConsentTest(KeycloakSession session) {
    KeycloakModelUtils.runJobInTransaction(session.getKeycloakSessionFactory(), (KeycloakSession currentSessionACT) -> {
        KeycloakSession currentSession = currentSessionACT;
        RealmModel realm = currentSession.realms().getRealmByName("original");
        ClientModel fooClient = realm.getClientByClientId("foo-client");
        UserModel john = currentSessionACT.users().getUserByUsername(realm, "john");
        UserModel mary = currentSessionACT.users().getUserByUsername(realm, "mary");
        Assert.assertEquals(2, currentSession.users().getConsentsStream(realm, john.getId()).count());
        ClientModel hardcodedClient = currentSessionACT.clients().getClientByClientId(realm, "hardcoded-client");
        List<UserConsentModel> maryConsents = currentSession.users().getConsentsStream(realm, mary.getId()).collect(Collectors.toList());
        Assert.assertEquals(2, maryConsents.size());
        UserConsentModel maryConsent = maryConsents.get(0);
        UserConsentModel maryHardcodedConsent = maryConsents.get(1);
        if (maryConsents.get(0).getClient().getId().equals(hardcodedClient.getId())) {
            maryConsent = maryConsents.get(1);
            maryHardcodedConsent = maryConsents.get(0);
        }
        Assert.assertEquals(maryConsent.getClient().getId(), fooClient.getId());
        Assert.assertEquals(maryConsent.getGrantedClientScopes().size(), 1);
        Assert.assertTrue(isClientScopeGranted(realm, "foo", maryConsent));
        Assert.assertEquals(maryHardcodedConsent.getClient().getId(), hardcodedClient.getId());
        Assert.assertEquals(maryHardcodedConsent.getGrantedClientScopes().size(), 0);
    });
}