add important tests

FAForever · Nov 8, 2020 · 2faa7dd · 2faa7dd
1 parent cfde386
commit 2faa7dd
Showing 1 changed file with 54 additions and 0 deletions.
diff --git a/src/test/java/com/faforever/client/chat/ChannelTabControllerTest.java b/src/test/java/com/faforever/client/chat/ChannelTabControllerTest.java
@@ -659,4 +659,58 @@ public void testEnemyPlayerBecomesOther() {
     assertEquals(0, enemies.size());
     assertEquals(2, otherUsers.size());
   }
+
+  @Test
+  public void testWhenUserAddPlayerToFriendlyListByAccidentallyAndCancelsActionImmediately() {
+    String username1 = "player1";
+    Player player1 = PlayerBuilder.create(username1).socialStatus(OTHER).get();
+    ChatChannelUser user1 = ChatChannelUserBuilder.create(username1).get();
+    String username2 = "player2";
+    Player player2 = PlayerBuilder.create(username2).socialStatus(OTHER).get();
+    ChatChannelUser user2 = ChatChannelUserBuilder.create(username2).get();
+    defaultChannel.addUsers(Arrays.asList(user1, user2));
+
+    when(playerService.getPlayerForUsername(username1)).thenReturn(Optional.of(player1));
+    when(playerService.getPlayerForUsername(username2)).thenReturn(Optional.of(player2));
+    Platform.runLater(() -> instance.setChannel(defaultChannel));
+    WaitForAsyncUtils.waitForFxEvents();
+
+    player1.setSocialStatus(FRIEND);
+    WaitForAsyncUtils.waitForFxEvents();
+    player1.setSocialStatus(OTHER);
+    WaitForAsyncUtils.waitForFxEvents();
+
+    List<CategoryOrChatUserListItem> friends = instance.getChatUserItemsByCategory(ChatUserCategory.FRIEND);
+    List<CategoryOrChatUserListItem> otherUsers = instance.getChatUserItemsByCategory(ChatUserCategory.OTHER);
+    assertTrue(otherUsers.stream().anyMatch(userItem -> userItem.getUser().equals(user1)));
+    assertEquals(0, friends.size());
+    assertEquals(2, otherUsers.size());
+  }
+
+  @Test
+  public void testWhenUserAddPlayerToFoeListByAccidentallyAndCancelsActionImmediately() {
+    String username1 = "player1";
+    Player player1 = PlayerBuilder.create(username1).socialStatus(OTHER).get();
+    ChatChannelUser user1 = ChatChannelUserBuilder.create(username1).get();
+    String username2 = "player2";
+    Player player2 = PlayerBuilder.create(username2).socialStatus(OTHER).get();
+    ChatChannelUser user2 = ChatChannelUserBuilder.create(username2).get();
+    defaultChannel.addUsers(Arrays.asList(user1, user2));
+
+    when(playerService.getPlayerForUsername(username1)).thenReturn(Optional.of(player1));
+    when(playerService.getPlayerForUsername(username2)).thenReturn(Optional.of(player2));
+    Platform.runLater(() -> instance.setChannel(defaultChannel));
+    WaitForAsyncUtils.waitForFxEvents();
+
+    player1.setSocialStatus(FOE);
+    WaitForAsyncUtils.waitForFxEvents();
+    player1.setSocialStatus(OTHER);
+    WaitForAsyncUtils.waitForFxEvents();
+
+    List<CategoryOrChatUserListItem> enemies = instance.getChatUserItemsByCategory(ChatUserCategory.FOE);
+    List<CategoryOrChatUserListItem> otherUsers = instance.getChatUserItemsByCategory(ChatUserCategory.OTHER);
+    assertTrue(otherUsers.stream().anyMatch(userItem -> userItem.getUser().equals(user1)));
+    assertEquals(0, enemies.size());
+    assertEquals(2, otherUsers.size());
+  }
 }