--- /dev/null
+/*
+ * Copyright 2005-2010 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.wamblee.security.authentication.jpa;
+
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.util.Set;
+
+import org.wamblee.cache.EhCache;
+import org.wamblee.io.ClassPathResource;
+import org.wamblee.security.authentication.Group;
+import org.wamblee.security.authentication.GroupSet;
+import org.wamblee.security.authentication.InMemoryUserSetTest;
+import org.wamblee.security.authentication.NameValidator;
+import org.wamblee.security.authentication.RegexpNameValidator;
+import org.wamblee.security.authentication.User;
+import org.wamblee.security.authentication.UserSet;
+import org.wamblee.security.authentication.jpa.JpaGroupSet;
+import org.wamblee.security.authentication.jpa.JpaUserSet;
+import org.wamblee.security.encryption.Md5HexMessageDigester;
+import org.wamblee.security.encryption.MessageDigester;
+import org.wamblee.support.persistence.JpaTester;
+import org.wamblee.support.persistence.TransactionProxyFactory;
+import org.wamblee.support.persistence.DatabaseUtils.JdbcUnitOfWork;
+
+/**
+ * Tests for {@link org.wamblee.usermgt.hibernate.HibernateGroupSet}
+ *
+ * @author Erik Brakkee
+ */
+public class JpaUserSetTest extends InMemoryUserSetTest {
+ private static final String USER_TABLE = "SEC_USER";
+
+ private static final String GROUP_TABLE = "SEC_GROUP";
+
+ private static final String USER_QUERY = "select * from " + USER_TABLE +
+ " where name = ?";
+
+ private static final String GROUP_QUERY = "select * from " + GROUP_TABLE +
+ " where name = ?";
+
+ private UserSet userset;
+
+ private GroupSet groupset;
+
+ private EhCache<String, User> userCache;
+
+ private JpaTester jpaTester;
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see org.wamblee.usermgt.InMemoryUserSetTest#setUp()
+ */
+ @Override
+ protected void setUp() throws Exception {
+ jpaTester = new JpaTester(new SecurityPersistenceUnit());
+ jpaTester.start();
+
+ userCache = new EhCache<String, User>(
+ new ClassPathResource("properties/org.wamblee.security.ehcache.xml"), "users");
+
+ userset = createUserSetImpl();
+ groupset = createGroupSetImpl();
+
+ clearUserCache();
+
+ super.setUp();
+ }
+
+ private UserSet createUserSetImpl() {
+ NameValidator passwordValidator = new RegexpNameValidator(
+ ".{5,}", "INVALID_PASSWORD", "Password must have at least 5 characters");
+
+ MessageDigester passwordDigester = new Md5HexMessageDigester();
+ TransactionProxyFactory<UserSet> factory = new TransactionProxyFactory<UserSet>(
+ jpaTester.getJpaBuilder(), UserSet.class);
+ UserSet jpaUserset = new JpaUserSet(userCache, passwordValidator, passwordDigester,
+ factory.getTransactionScopedEntityManager());
+ return factory.getProxy(jpaUserset);
+ }
+
+ private GroupSet createGroupSetImpl() {
+ TransactionProxyFactory<GroupSet> factory = new TransactionProxyFactory<GroupSet>(
+ jpaTester.getJpaBuilder(), GroupSet.class);
+ GroupSet groupset = new JpaGroupSet(factory.getTransactionScopedEntityManager());
+ GroupSet proxy = factory.getProxy(groupset);
+ return proxy;
+ }
+
+ @Override
+ protected void tearDown() throws Exception {
+ jpaTester.stop();
+ super.tearDown();
+ }
+
+ /**
+ * Clears the user cache.
+ */
+ private void clearUserCache() {
+ userCache.clear();
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupCount(int)
+ */
+ @Override
+ protected void checkUserCount(int aSize) throws Exception {
+ super.checkUserCount(aSize);
+ assertEquals(aSize, jpaTester.getDbUtils().getTableSize(USER_TABLE));
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see
+ * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupExists(java.lang.String
+ * )
+ */
+ @Override
+ protected void checkUserExists(final String aUser) throws Exception {
+ assertEquals(1, countUser(aUser));
+ }
+
+ private int countUser(final String aUser) throws Exception {
+ int count = jpaTester.getDbUtils().executeInTransaction(new JdbcUnitOfWork<Integer>() {
+ @Override
+ public Integer execute(Connection aConnection) throws Exception {
+ ResultSet res = jpaTester.getDbUtils().executeQuery(aConnection, USER_QUERY, aUser);
+ return jpaTester.getDbUtils().countResultSet(res);
+ }
+ });
+ return count;
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see
+ * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupNotExists(java.lang
+ * .String)
+ */
+ @Override
+ protected void checkUserNotExists(String aUser) throws Exception {
+ assertEquals(0, countUser(aUser));
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupCount(int)
+ */
+ @Override
+ protected void checkGroupCount(int aSize) throws Exception {
+ assertEquals(aSize, jpaTester.getDbUtils().getTableSize(GROUP_TABLE));
+ }
+
+ private int countGroup(final String aGroup) throws Exception {
+ int count = jpaTester.getDbUtils().executeInTransaction(new JdbcUnitOfWork<Integer>() {
+ @Override
+ public Integer execute(Connection aConnection) throws Exception {
+ ResultSet res = jpaTester.getDbUtils().executeQuery(aConnection, GROUP_QUERY, aGroup);
+ return jpaTester.getDbUtils().countResultSet(res);
+ }
+ });
+ return count;
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see
+ * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupExists(java.lang.String
+ * )
+ */
+ @Override
+ protected void checkGroupExists(String aGroup) throws Exception {
+ assertEquals(1, countGroup(aGroup));
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see
+ * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupNotExists(java.lang
+ * .String)
+ */
+ @Override
+ protected void checkGroupNotExists(String aGroup) throws Exception {
+ assertEquals(0, countGroup(aGroup));
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see org.wamblee.usermgt.InMemoryGroupSetTest#createGroupSet()
+ */
+ @Override
+ protected UserSet createUserSet() {
+ return userset;
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see org.wamblee.usermgt.InMemoryUserSetTest#createGroupSet()
+ */
+ @Override
+ protected GroupSet createGroupSet() {
+ return groupset;
+ }
+
+ /**
+ * Reproduction of a bug. Create a user which is in group1 Add it to a
+ * second group group2. Remove the user from group1. Verify the user is in
+ * group2.
+ *
+ */
+ public void testVerifyAddRemove() throws Exception {
+ jpaTester.getDbUtils().cleanDatabase(); // super class setup always creates one group.
+
+ GroupSet groups = getGroups();
+ assertEquals(0, groups.size());
+
+ Group group1 = createGroup("group1");
+ Group group2 = createGroup("group2");
+ groups.add(group1);
+ groups.add(group2);
+ checkGroupExists("group1");
+ checkGroupExists("group2");
+
+ User user = createUser("user", PASSWORD, group1);
+ getUsers().add(user);
+ checkUserExists("user");
+
+ addUserToGroup(user, group2);
+ getUsers().userModified(user);
+ clearUserCache();
+
+ User user2 = getUsers().find("user");
+
+ Set<Group> userGroups = user2.getGroups();
+ assertTrue(user2.isInGroup("group1"));
+ assertTrue(user2.isInGroup("group2"));
+ assertEquals(2, userGroups.size());
+
+ removeUserFromGroup(user, group1);
+ getUsers().userModified(user);
+ clearUserCache();
+ user2 = getUsers().find("user");
+ userGroups = user2.getGroups();
+ assertFalse(user2.isInGroup("group1"));
+ assertTrue(user2.isInGroup("group2"));
+ assertEquals(1, userGroups.size());
+ }
+}
git://wamblee.org / utils / blobdiff