Updated Tests. (#1)

cryptic-game · Oct 30, 2019 · 4ad0dcd · 4ad0dcd
1 parent f60d43b
commit 4ad0dcd
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/src/test/java/net/cryptic_game/microservice/chat/channel/ChannelTests.java b/src/test/java/net/cryptic_game/microservice/chat/channel/ChannelTests.java
@@ -34,11 +34,19 @@ public void testSetName() {
     public void testAddUser() {
         final Channel channel = new Channel("test");
         final User testUser = new User(UUID.randomUUID(), "test", "test@test.test", new Date(), new Date());
-        channel.addUser(testUser);
 
+        assertTrue(channel.addUser(testUser));
         assertTrue(channel.getUsers().contains(testUser));
     }
 
+    @Test
+    public void testDupedAddUser() {
+        final Channel channel = new Channel("test");
+        final User testUser = new User(UUID.randomUUID(), "test", "test@test.test", new Date(), new Date());
+        channel.addUser(testUser);
+        assertFalse(channel.addUser(testUser));
+    }
+
     @Test
     public void testRemoveUser() {
         final Channel channel = new Channel("test");
@@ -49,8 +57,15 @@ public void testRemoveUser() {
             fail();
         }
 
-        channel.removeUser(testUser);
-
+        assertTrue(channel.removeUser(testUser));
         assertFalse(channel.getUsers().contains(testUser));
     }
+
+    @Test
+    public void testRemoveUserWithoutExist() {
+        final Channel channel = new Channel("test");
+        final User testUser = new User(UUID.randomUUID(), "test", "test@test.test", new Date(), new Date());
+
+        assertFalse(channel.removeUser(testUser));
+    }
 }