Java tutorial
/* * Cloud Foundry 2012.02.03 Beta * Copyright (c) [2009-2012] VMware, Inc. All Rights Reserved. * * This product is licensed to you under the Apache License, Version 2.0 (the "License"). * You may not use this product except in compliance with the License. * * This product includes a number of subcomponents with * separate copyright notices and license terms. Your use of these * subcomponents is subject to the terms and conditions of the * subcomponent's license, as noted in the LICENSE file. */ package org.cloudfoundry.identity.uaa.scim.bootstrap; import org.cloudfoundry.identity.uaa.rest.jdbc.DefaultLimitSqlAdapter; import org.cloudfoundry.identity.uaa.rest.jdbc.JdbcPagingListFactory; import org.cloudfoundry.identity.uaa.scim.endpoints.ScimUserEndpoints; import org.cloudfoundry.identity.uaa.scim.ScimUser; import org.cloudfoundry.identity.uaa.scim.jdbc.JdbcScimGroupMembershipManager; import org.cloudfoundry.identity.uaa.scim.jdbc.JdbcScimGroupProvisioning; import org.cloudfoundry.identity.uaa.scim.jdbc.JdbcScimUserProvisioning; import org.cloudfoundry.identity.uaa.scim.validate.NullPasswordValidator; import org.cloudfoundry.identity.uaa.user.UaaUser; import org.junit.After; import org.junit.Before; import org.junit.Test; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.jdbc.datasource.embedded.EmbeddedDatabase; import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder; import org.springframework.security.core.authority.AuthorityUtils; import java.util.Arrays; import java.util.Collection; import java.util.Map; import static org.junit.Assert.assertEquals; /** * @author Luke Taylor * @author Dave Syer */ public class ScimUserBootstrapTests { private JdbcScimUserProvisioning db; private JdbcScimGroupProvisioning gdb; private JdbcScimGroupMembershipManager mdb; private ScimUserEndpoints userEndpoints; private EmbeddedDatabase database; @Before public void setUp() { EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder(); builder.addScript("classpath:/org/cloudfoundry/identity/uaa/schema-hsqldb.sql"); builder.addScript("classpath:/org/cloudfoundry/identity/uaa/scim/schema-hsqldb.sql"); database = builder.build(); JdbcTemplate jdbcTemplate = new JdbcTemplate(database); JdbcPagingListFactory pagingListFactory = new JdbcPagingListFactory(jdbcTemplate, new DefaultLimitSqlAdapter()); db = new JdbcScimUserProvisioning(jdbcTemplate, pagingListFactory); db.setPasswordValidator(new NullPasswordValidator()); gdb = new JdbcScimGroupProvisioning(jdbcTemplate, pagingListFactory); mdb = new JdbcScimGroupMembershipManager(jdbcTemplate); mdb.setScimUserProvisioning(db); mdb.setScimGroupProvisioning(gdb); userEndpoints = new ScimUserEndpoints(); userEndpoints.setScimGroupMembershipManager(mdb); userEndpoints.setScimUserProvisioning(db); } @After public void shutdownDb() throws Exception { database.shutdown(); } @Test public void canAddUsers() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); UaaUser mabel = new UaaUser("mabel", "password", "mabel@blah.com", "Mabel", "User"); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe, mabel)); bootstrap.afterPropertiesSet(); Collection<ScimUser> users = db.retrieveAll(); assertEquals(2, users.size()); } @Test public void canAddUserWithAuthorities() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid,read")); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); @SuppressWarnings("unchecked") Collection<Map<String, Object>> users = (Collection<Map<String, Object>>) userEndpoints .findUsers("id", "id pr", "id", "ascending", 1, 100).getResources(); assertEquals(1, users.size()); String id = (String) users.iterator().next().get("id"); ScimUser user = userEndpoints.getUser(id); // uaa.user is always added assertEquals(3, user.getGroups().size()); } @Test public void noOverrideByDefault() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid,read")); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); joe = new UaaUser("joe", "password", "joe@test.org", "Joel", "User"); bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); @SuppressWarnings("unchecked") Collection<Map<String, Object>> users = (Collection<Map<String, Object>>) userEndpoints .findUsers("id", "id pr", "id", "ascending", 1, 100).getResources(); assertEquals(1, users.size()); String id = (String) users.iterator().next().get("id"); ScimUser user = userEndpoints.getUser(id); // uaa.user is always added assertEquals("Joe", user.getGivenName()); } @Test public void canOverride() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid,read")); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); joe = new UaaUser("joe", "password", "joe@test.org", "Joel", "User"); bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.setOverride(true); bootstrap.afterPropertiesSet(); @SuppressWarnings("unchecked") Collection<Map<String, Object>> users = (Collection<Map<String, Object>>) userEndpoints .findUsers("id", "id pr", "id", "ascending", 1, 100).getResources(); assertEquals(1, users.size()); String id = (String) users.iterator().next().get("id"); ScimUser user = userEndpoints.getUser(id); // uaa.user is always added assertEquals("Joel", user.getGivenName()); } @Test public void canOverrideAuthorities() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid,read")); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid,read,write")); bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.setOverride(true); bootstrap.afterPropertiesSet(); @SuppressWarnings("unchecked") Collection<Map<String, Object>> users = (Collection<Map<String, Object>>) userEndpoints .findUsers("id", "id pr", "id", "ascending", 1, 100).getResources(); assertEquals(1, users.size()); String id = (String) users.iterator().next().get("id"); ScimUser user = userEndpoints.getUser(id); // uaa.user is always added assertEquals(4, user.getGroups().size()); } @Test public void canRemoveAuthorities() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid,read")); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); joe = joe.authorities(AuthorityUtils.commaSeparatedStringToAuthorityList("openid")); JdbcTemplate jdbcTemplate = new JdbcTemplate(database); System.err.println(jdbcTemplate.queryForList("SELECT * FROM group_membership")); bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.setOverride(true); bootstrap.afterPropertiesSet(); @SuppressWarnings("unchecked") Collection<Map<String, Object>> users = (Collection<Map<String, Object>>) userEndpoints .findUsers("id", "id pr", "id", "ascending", 1, 100).getResources(); assertEquals(1, users.size()); String id = (String) users.iterator().next().get("id"); ScimUser user = userEndpoints.getUser(id); // uaa.user is always added assertEquals(2, user.getGroups().size()); } @Test public void canUpdateUsers() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); joe = new UaaUser("joe", "new", "joe@test.org", "Joe", "Bloggs"); bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.setOverride(true); bootstrap.afterPropertiesSet(); Collection<ScimUser> users = db.retrieveAll(); assertEquals(1, users.size()); assertEquals("Bloggs", users.iterator().next().getFamilyName()); } @Test public void failedAttemptToUpdateUsersNotFatal() throws Exception { UaaUser joe = new UaaUser("joe", "password", "joe@test.org", "Joe", "User"); ScimUserBootstrap bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.afterPropertiesSet(); joe = new UaaUser("joe", "new", "joe@test.org", "Joe", "Bloggs"); bootstrap = new ScimUserBootstrap(db, gdb, mdb, Arrays.asList(joe)); bootstrap.setOverride(false); bootstrap.afterPropertiesSet(); Collection<ScimUser> users = db.retrieveAll(); assertEquals(1, users.size()); assertEquals("User", users.iterator().next().getFamilyName()); } }