/*
 * Copyright 2000-2006 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package jetbrains.communicator.idea.toolWindow;

import com.intellij.ui.treeStructure.Tree;
import jetbrains.communicator.BaseTestCase;
import jetbrains.communicator.core.TestFactory;
import jetbrains.communicator.core.impl.dispatcher.LocalMessageDispatcherImpl;
import jetbrains.communicator.core.users.User;
import jetbrains.communicator.core.users.UserEvent;
import jetbrains.communicator.core.users.UserModel;
import jetbrains.communicator.mock.MockIDEFacade;
import jetbrains.communicator.mock.MockUser;
import jetbrains.communicator.util.TreeUtils;

import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.TreePath;

/**
 * @author kir
 */
public class UsersTreeModelTest extends BaseTestCase {
  private UserModel myUserModel;
  private UsersTreeModel myUsersTreeModel;
  private JTree myTree;
  private LocalMessageDispatcherImpl myLocalMessageDispatcher;

  @Override
  protected void setUp() throws Exception {
    super.setUp();

    myUserModel = TestFactory.createUserListWithUsers(this);
    myTree = new Tree();
    myLocalMessageDispatcher = new LocalMessageDispatcherImpl(getBroadcaster(), new MockIDEFacade(getClass()), myUserModel);
    disposeOnTearDown(myLocalMessageDispatcher);
    myUsersTreeModel = new UsersTreeModel(myTree, myUserModel, myLocalMessageDispatcher);
    disposeOnTearDown(myUsersTreeModel);
    myTree.setModel(myUsersTreeModel);
  }

  public void testUserDeleted() throws Exception {
    myUserModel.removeGroup("group2");
    assertEquals(0, myTree.getSelectionRows()[0]);

    myUserModel.removeGroup("group1");
    assertEquals(0, myTree.getSelectionRows()[0]);
  }

  public void testUserNameChanged() throws Exception {

    final User user = myUserModel.getAllUsers()[0];
    assertEquals("aaa", user.getName());

    TreePath path = new TreePath(new DefaultMutableTreeNode(user));
    myUsersTreeModel.valueForPathChanged(path, "new name");

    assertEquals("new name", user.getDisplayName());
  }

  public void testGroupNameChanged() throws Exception {
    TreePath path = new TreePath(new DefaultMutableTreeNode("group1"));

    myUsersTreeModel.valueForPathChanged(path, "new name");

    assertEquals(0, myUserModel.getUsers("group1").length);
    assertEquals("user groups should be changed from group1 to 'new name'",
        2, myUserModel.getUsers("new name").length);
    assertEquals("new name",
        ((DefaultMutableTreeNode) path.getLastPathComponent()).getUserObject());
  }

  public void testGroupNameChanged_EmptyGroup() throws Exception {
    TreePath path = new TreePath(new DefaultMutableTreeNode("group1"));

    myUsersTreeModel.valueForPathChanged(path, "");

    assertEquals(0, myUserModel.getUsers("group1").length);
    assertEquals("user groups should be changed from group1 to 'General'",
        2, myUserModel.getUsers(UserModel.DEFAULT_GROUP).length);
    assertEquals(UserModel.DEFAULT_GROUP,
        ((DefaultMutableTreeNode) path.getLastPathComponent()).getUserObject());
  }

  public void testRebuildTreeOnUserNameChange() throws Exception {
    myTree.expandRow(2);
    myTree.expandRow(1);

    assertEquals("All expanded, root node is hidden", 8, myTree.getRowCount());

    // rename user aaa
    myUserModel.getUsers("group1")[0].setDisplayName("xxxx", myUserModel);

    assertEquals("Nodes expansion should be preserved", 8, myTree.getRowCount());

    assertEquals("Tree rebuild expected", "ccc", getUserForRow(myTree, 2).getDisplayName());
    assertEquals("Tree rebuild expected", "xxxx", getUserForRow(myTree, 3).getDisplayName());
  }

  public void testRebuildTreeOnOnlineStatusChange() throws Exception {
    myTree.expandRow(2);
    myTree.expandRow(1);

    assertEquals("All expanded, root node is hidden", 8, myTree.getRowCount());

    // rename user aaa
    MockUser mockUser = ((MockUser) myUserModel.getUsers("group1")[1]);
    mockUser.setOnline(true);
    getBroadcaster().fireEvent(new UserEvent.Online(mockUser));

    assertEquals("Nodes expansion should be preserved", 8, myTree.getRowCount());

    assertEquals("Online user should go to the top", "ccc", getUserForRow(myTree, 2).getDisplayName());
  }

  public void testRebuildTreeOnGroupChange() throws Exception {
    myTree.expandRow(2);
    myTree.expandRow(1);

    assertEquals("All expanded, root node is hidden", 8, myTree.getRowCount());

    myUserModel.renameGroup("group1", "new name");

    assertEquals("Nodes expansion should be preserved", 8, myTree.getRowCount());

    assertEquals("Tree rebuild expected, group2 should be the first in the list",
        "zzz", getUserForRow(myTree, 2).getDisplayName());
  }

  public void testExplicitGroup() throws Exception {
    myUserModel.addGroup("aaaaaaaa");

    assertEquals("aaaaaaaa", TreeUtils.getUserObject(myTree.getPathForRow(1)));
    myUserModel.renameGroup("aaaaaaaa", "aaaaaaaa1");

    assertEquals("aaaaaaaa1", TreeUtils.getUserObject(myTree.getPathForRow(1)));
  }

  private User getUserForRow(final JTree tree, int row) {
    return ((User) ((DefaultMutableTreeNode) tree.getPathForRow(row).getLastPathComponent()).getUserObject());
  }

}