android/guiTestSrc/com/android/tools/idea/tests/gui/framework/fixture/FindToolWindowFixture.java

/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * 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 com.android.tools.idea.tests.gui.framework.fixture;

import com.android.annotations.Nullable;
import com.google.common.collect.Lists;
import com.intellij.ide.OccurenceNavigatorSupport;
import com.intellij.ui.content.Content;
import com.intellij.ui.treeStructure.Tree;
import com.intellij.usageView.UsageViewManager;
import com.intellij.usages.impl.GroupNode;
import org.fest.swing.edt.GuiQuery;
import org.jetbrains.annotations.NotNull;

import javax.swing.*;
import java.util.List;

import static org.fest.reflect.core.Reflection.field;
import static org.fest.swing.edt.GuiActionRunner.execute;
import static org.junit.Assert.assertNotNull;

public class FindToolWindowFixture {
  @NotNull private final IdeFrameFixture myParent;

  public FindToolWindowFixture(@NotNull IdeFrameFixture parent) {
    myParent = parent;
  }

  @NotNull
  public ContentFixture getSelectedContext() {
    return new ContentFixture(myParent);
  }

  public static class ContentFixture {
    @NotNull private final Content myContent;

    ContentFixture(@NotNull IdeFrameFixture parent) {
      UsageViewManager usageViewManager = UsageViewManager.getInstance(parent.getProject());
      myContent = usageViewManager.getSelectedContent();
      assertNotNull(myContent);
    }


    public void findUsagesInGeneratedCodeGroup() {
      findUsageGroup("Usages in generated code");
    }

    public void findUsageGroup(@NotNull final String groupText) {
      final Tree tree = getContentsTree();
      final List<String> groupNames = Lists.newArrayList();
      GroupNode foundGroup = execute(new GuiQuery<GroupNode>() {
        @Override
        @Nullable
        protected GroupNode executeInEDT() throws Throwable {
          GroupNode rootNode = (GroupNode)tree.getModel().getRoot();
          GroupNode found = null;
          for (GroupNode subGroup : rootNode.getSubGroups()) {
            String subGroupText = subGroup.getGroup().getText(null);
            groupNames.add(subGroupText);
            if (groupText.equals(subGroupText)) {
              found = subGroup;
            }
          }
          return found;
        }
      });
      String msg = String.format("Failed to find usage group '%1$s' in %2$s", groupText, groupNames);
      assertNotNull(msg, foundGroup);
    }

    @NotNull
    private Tree getContentsTree() {
      JComponent component = myContent.getComponent();
      OccurenceNavigatorSupport navigatorSupport = field("mySupport").ofType(OccurenceNavigatorSupport.class).in(component).get();
      assertNotNull(navigatorSupport);
      JTree tree = field("myTree").ofType(JTree.class).in(navigatorSupport).get();
      assertNotNull(tree);
      return (Tree)tree;
    }
  }
}