From e72cd6a78b47684cf1431e406e0390162ab5103d Mon Sep 17 00:00:00 2001
From: j2blake <j2blake@440791e1-c2bf-4c1e-ad7c-caf3f8b0ebe8>
Date: Tue, 20 Dec 2011 21:53:27 +0000
Subject: [PATCH] NIHVIVO-3523 Implement more functionality for unit tests.

---
 .../vitro/webapp/dao/UserAccountsDaoStub.java    | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/webapp/test/stubs/edu/cornell/mannlib/vitro/webapp/dao/UserAccountsDaoStub.java b/webapp/test/stubs/edu/cornell/mannlib/vitro/webapp/dao/UserAccountsDaoStub.java
index 9c782a90b..e404b348e 100644
--- a/webapp/test/stubs/edu/cornell/mannlib/vitro/webapp/dao/UserAccountsDaoStub.java
+++ b/webapp/test/stubs/edu/cornell/mannlib/vitro/webapp/dao/UserAccountsDaoStub.java
@@ -20,6 +20,7 @@ public class UserAccountsDaoStub implements UserAccountsDao {
 	private static final Log log = LogFactory.getLog(UserAccountsDaoStub.class);
 
 	private final Map<String, UserAccount> userAccountsByUri = new HashMap<String, UserAccount>();
+	private final Map<String, PermissionSet> permissionSetsByUri = new HashMap<String, PermissionSet>();
 
 	// ----------------------------------------------------------------------
 	// Stub infrastructure
@@ -28,6 +29,10 @@ public class UserAccountsDaoStub implements UserAccountsDao {
 	public void addUser(UserAccount user) {
 		userAccountsByUri.put(user.getUri(), user);
 	}
+	
+	public void addPermissionSet(PermissionSet ps) {
+		permissionSetsByUri.put(ps.getUri(), ps);
+	}
 
 	// ----------------------------------------------------------------------
 	// Stub methods
@@ -38,6 +43,11 @@ public class UserAccountsDaoStub implements UserAccountsDao {
 		return userAccountsByUri.get(uri);
 	}
 
+	@Override
+	public PermissionSet getPermissionSetByUri(String uri) {
+		return permissionSetsByUri.get(uri);
+	}
+
 	// ----------------------------------------------------------------------
 	// Un-implemented methods
 	// ----------------------------------------------------------------------
@@ -66,12 +76,6 @@ public class UserAccountsDaoStub implements UserAccountsDao {
 				"UserAccountsDaoStub.deleteUserAccount() not implemented.");
 	}
 
-	@Override
-	public PermissionSet getPermissionSetByUri(String uri) {
-		throw new RuntimeException(
-				"UserAccountsDaoStub.getPermissionSetByUri() not implemented.");
-	}
-
 	@Override
 	public Collection<PermissionSet> getAllPermissionSets() {
 		throw new RuntimeException(