1 files changed, 14 insertions, 6 deletions

diff --git a/ecomp-sdk/epsdk-domain/src/test/java/org/onap/portalsdk/core/domain/UserTest.java b/ecomp-sdk/epsdk-domain/src/test/java/org/onap/portalsdk/core/domain/UserTest.java
index 598ee900..b4fb2ced 100644
--- a/ecomp-sdk/epsdk-domain/src/test/java/org/onap/portalsdk/core/domain/UserTest.java
+++ b/ecomp-sdk/epsdk-domain/src/test/java/org/onap/portalsdk/core/domain/UserTest.java
@@ -2,7 +2,7 @@
  * ============LICENSE_START==========================================
  * ONAP Portal SDK
  * ===================================================================
- * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
  * ===================================================================
  *
  * Unless otherwise specified, all software contained herein is licensed
@@ -37,17 +37,15 @@
  */
 package org.onap.portalsdk.core.domain;
 
-import static org.junit.Assert.*;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
 
 import java.util.Set;
 import java.util.SortedSet;
 import java.util.TreeSet;
 
 import org.junit.Test;
-import org.onap.portalsdk.core.domain.App;
-import org.onap.portalsdk.core.domain.Role;
-import org.onap.portalsdk.core.domain.User;
-import org.onap.portalsdk.core.domain.UserApp;
 
 public class UserTest {
 
@@ -165,8 +163,14 @@ public class UserTest {
 	public void addAppRolesIfRolesNullTest()
 	{
 		User user = mockUser();
+		App appTest = new App();
+		appTest.setId(1l);
+		SortedSet<Role>  roles = new TreeSet<>();
 		Set userApps = new TreeSet();
 		UserApp userapp = new UserApp();
+		userapp.setApp(appTest);
+		userapp.setRole(new Role());
+		user.setRoles(roles);
 		userApps.add(userapp);
 		user.setUserApps(userApps);
 		user.setUserApps(userApps);
@@ -195,11 +199,15 @@ public class UserTest {
 		User user = mockUser();
 		Set userApps = new TreeSet();
 		UserApp userapp = new UserApp();
+		App app = new App();
+		app.setId(1l);
 		SortedSet<Role> roles = new TreeSet();
 		Role role = new Role();
 		role.setId((long) 1);
+		role.setName("test");
 		roles.add(role);
 		userapp.setRole(role);
+		userapp.setApp(app);
 		userApps.add(userapp);
 		user.setUserApps(userApps);
 		user.removeRole((long) 1);