diff --git a/build.gradle b/build.gradle index 2b6ed72..98f5661 100644 --- a/build.gradle +++ b/build.gradle @@ -18,9 +18,9 @@ apply plugin: 'org.springframework.boot' apply plugin: 'io.spring.dependency-management' apply plugin: 'application' -application { - mainClassName = 'greenify.server.Application' -} +//application { +// mainClassName = 'greenify.server.Application' +//} repositories { mavenCentral() diff --git a/src/Server/src/main/java/greenify/server/data/model/User.java b/src/Server/src/main/java/greenify/server/data/model/User.java index 9f1860b..9227fde 100644 --- a/src/Server/src/main/java/greenify/server/data/model/User.java +++ b/src/Server/src/main/java/greenify/server/data/model/User.java @@ -208,6 +208,19 @@ public class User { } } + /** + * Removes a friend from the friendslist of the user. + * @param user the friend you want to remove. + */ + public void removeFriend(User user) { + if (!friends.contains(user)) { + throw new ApplicationException("This user is not your friend!"); + } else { + friends.remove(user); + System.out.print("Friend removed"); + } + } + /** * This method gets a human readable (JSON) object. * @return the JSON form of the object. diff --git a/src/Server/src/test/java/greenify/server/data/model/UserTest.java b/src/Server/src/test/java/greenify/server/data/model/UserTest.java index f95b2c4..dcfd702 100644 --- a/src/Server/src/test/java/greenify/server/data/model/UserTest.java +++ b/src/Server/src/test/java/greenify/server/data/model/UserTest.java @@ -125,6 +125,27 @@ public class UserTest { }); } + @Test + public void removeFriendValidTest() { + User test = new User(1L, "greenify", "password"); + List friendList = new ArrayList<>(); + friendList.add(test); + User user = new User(1L, "green", "pass"); + user.setFriends(friendList); + assertEquals(user.getFriends(), friendList); + user.removeFriend(test); + assertEquals(user.getFriends(), new ArrayList()); + } + + @Test + public void removeFriendInvalidTest() { + User user = new User(1L, "greenify", "password"); + User test = new User(2L, "user", "pass"); + assertThrows(ApplicationException.class, () -> { + user.removeFriend(test); + }); + } + @Test public void setFriendTest() { List friends = new ArrayList();