platform/smRunner/src/com/intellij/execution/testframework/sm/runner/ui/statistics/StatisticsPanel.java - platform/tools/idea - Git at Google

 /*
  * Copyright 2000-2009 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 com.intellij.execution.testframework.sm.runner.ui.statistics;

 import com.intellij.execution.testframework.TestFrameworkRunningModel;
 import com.intellij.execution.testframework.TestsUIUtil;
 import com.intellij.execution.testframework.sm.SMRunnerUtil;
 import com.intellij.execution.testframework.sm.runner.SMTRunnerEventsAdapter;
 import com.intellij.execution.testframework.sm.runner.SMTRunnerEventsListener;
 import com.intellij.execution.testframework.sm.runner.SMTestProxy;
 import com.intellij.execution.testframework.sm.runner.ui.PropagateSelectionHandler;
 import com.intellij.ide.DataManager;
 import com.intellij.openapi.actionSystem.ActionPlaces;
 import com.intellij.openapi.actionSystem.DataKey;
 import com.intellij.openapi.actionSystem.DataProvider;
 import com.intellij.openapi.actionSystem.IdeActions;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.wm.IdeFocusManager;
 import com.intellij.ui.DoubleClickListener;
 import com.intellij.ui.PopupHandler;
 import com.intellij.ui.TableUtil;
 import com.intellij.ui.table.BaseTableView;
 import com.intellij.ui.table.TableView;
 import com.intellij.util.config.Storage;
 import com.intellij.util.containers.ContainerUtil;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;

 import javax.swing.*;
 import java.awt.event.InputEvent;
 import java.awt.event.KeyEvent;
 import java.awt.event.MouseEvent;
 import java.util.Collection;
 import java.util.List;

 /**
  * @author Roman Chernyatchik
  */
 public class StatisticsPanel implements DataProvider {
   public static final DataKey<StatisticsPanel> SM_TEST_RUNNER_STATISTICS  = DataKey.create("SM_TEST_RUNNER_STATISTICS");

   private TableView<SMTestProxy> myStatisticsTableView;
   private JPanel myContentPane;
   private final Storage.PropertiesComponentStorage myStorage = new Storage.PropertiesComponentStorage("sm_test_statistics_table_columns");

   private final StatisticsTableModel myTableModel;
   private final List<PropagateSelectionHandler> myPropagateSelectionHandlers = ContainerUtil.createLockFreeCopyOnWriteList();
   private final Project myProject;
   private final TestFrameworkRunningModel myFrameworkRunningModel;

   public StatisticsPanel(final Project project, final TestFrameworkRunningModel model) {
     myProject = project;
     myTableModel = new StatisticsTableModel();
     myStatisticsTableView.setModelAndUpdateColumns(myTableModel);
     myFrameworkRunningModel = model;

     final Runnable gotoSuiteOrParentAction = createGotoSuiteOrParentAction();
     new DoubleClickListener() {
       @Override
       protected boolean onDoubleClick(MouseEvent e) {
         gotoSuiteOrParentAction.run();
         return true;
       }
     }.installOn(myStatisticsTableView);

     // Fire selection changed and move focus on SHIFT+ENTER
     final KeyStroke shiftEnterKey = KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, InputEvent.SHIFT_MASK);
     SMRunnerUtil.registerAsAction(shiftEnterKey, "select-test-proxy-in-test-view",
                             new Runnable() {
                               public void run() {
                                 showSelectedProxyInTestsTree();
                               }
                             },
                             myStatisticsTableView);

     // Expand selected or go to parent on ENTER
     final KeyStroke enterKey = KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, 0);
     SMRunnerUtil.registerAsAction(enterKey, "go-to-selected-suite-or-parent",
                             gotoSuiteOrParentAction,
                             myStatisticsTableView);
     // Contex menu in Table
     PopupHandler.installPopupHandler(myStatisticsTableView, IdeActions.GROUP_TESTTREE_POPUP, ActionPlaces.TESTTREE_VIEW_POPUP);
     // set this statistic tab as dataprovider for test's table view
     DataManager.registerDataProvider(myStatisticsTableView, this);
   }

   public void addPropagateSelectionListener(final PropagateSelectionHandler handler) {
     myPropagateSelectionHandlers.add(handler);
   }

   public JPanel getContentPane() {
     return myContentPane;
   }

   public SMTRunnerEventsListener createTestEventsListener() {
     return new SMTRunnerEventsAdapter() {
       @Override
       public void onSuiteStarted(@NotNull final SMTestProxy suite) {
         if (myTableModel.shouldUpdateModelBySuite(suite)) {
           updateAndRestoreSelection();
         }
       }

       @Override
       public void onSuiteFinished(@NotNull final SMTestProxy suite) {
         if (myTableModel.shouldUpdateModelBySuite(suite)) {
           updateAndRestoreSelection();
         }
       }

       @Override
       public void onTestStarted(@NotNull final SMTestProxy test) {
         if (myTableModel.shouldUpdateModelByTest(test)) {
           updateAndRestoreSelection();
         }
       }

       @Override
       public void onTestFinished(@NotNull final SMTestProxy test) {
         if (myTableModel.shouldUpdateModelByTest(test)) {
           updateAndRestoreSelection();
         }
       }

       private void updateAndRestoreSelection() {
         SMRunnerUtil.addToInvokeLater(new Runnable() {
           public void run() {
             BaseTableView.restore(myStorage, myStatisticsTableView);
             // statisticsTableView can be null in JUnit tests
             final SMTestProxy oldSelection = myStatisticsTableView.getSelectedObject();

             // update module
             myTableModel.updateModel();

             // restore selection if it is possible
             if (oldSelection != null) {
               final int newRow = myTableModel.getIndexOf(oldSelection);
               if (newRow > -1) {
                 myStatisticsTableView.setRowSelectionInterval(newRow, newRow);
               }
             }
           }
         });
       }
     };
   }

   public Object getData(@NonNls final String dataId) {
     if (SM_TEST_RUNNER_STATISTICS.is(dataId)) {
       return this;
     }
     return TestsUIUtil.getData(getSelectedItem(), dataId, myFrameworkRunningModel);
   }

   /**
    * On event - change selection and probably requests focus. Is used when we want
    * navigate from other component to this
    * @return Listener
    */
   public PropagateSelectionHandler createSelectMeListener() {
     return new PropagateSelectionHandler() {
       public void handlePropagateSelectionRequest(@Nullable final SMTestProxy selectedTestProxy,
                                     @NotNull final Object sender,
                                     final boolean requestFocus) {
         selectProxy(selectedTestProxy, sender, requestFocus);
       }
     };
   }

   public void selectProxy(@Nullable final SMTestProxy selectedTestProxy,
                           @NotNull final Object sender,
                           final boolean requestFocus) {
     SMRunnerUtil.addToInvokeLater(new Runnable() {
       public void run() {
         // Select tab if focus was requested
         if (requestFocus) {
           IdeFocusManager.getInstance(myProject).requestFocus(myStatisticsTableView, true);
         }

         // Select proxy in table
         selectProxy(selectedTestProxy);
       }
     });
   }

   protected void showSelectedProxyInTestsTree() {
     final Collection<SMTestProxy> proxies = myStatisticsTableView.getSelection();
     if (proxies.isEmpty()) {
       return;
     }
     final SMTestProxy proxy = proxies.iterator().next();
     myStatisticsTableView.clearSelection();
     fireOnPropagateSelection(proxy);
   }

   protected Runnable createGotoSuiteOrParentAction() {
     // Expand selected or go to parent
     return new Runnable() {
       public void run() {
         final SMTestProxy selectedProxy = getSelectedItem();
         if (selectedProxy == null) {
           return;
         }

         final int i = myStatisticsTableView.getSelectedRow();
         assert i >= 0; //because something is selected

         // if selected element is suite - we should expand it
         if (selectedProxy.isSuite()) {
           // expand and select first (Total) row
           showInTableAndSelectRow(selectedProxy, selectedProxy);
         }
       }
     };
   }

   protected void selectProxy(@Nullable final SMTestProxy selectedTestProxy) {
     // Send event to model
     myTableModel.updateModelOnProxySelected(selectedTestProxy);

     // Now we want to select proxy in table (if it is possible)
     if (selectedTestProxy != null) {
       findAndSelectInTable(selectedTestProxy);
     }
   }
   /**
    * Selects row in table
    * @param rowIndex Row's index
    */
   protected void selectRow(final int rowIndex) {
     SMRunnerUtil.addToInvokeLater(new Runnable() {
       public void run() {
         // updates model
         myStatisticsTableView.setRowSelectionInterval(rowIndex, rowIndex);

         // Scroll to visible
         TableUtil.scrollSelectionToVisible(myStatisticsTableView);
       }
     });
   }

   /**
    * Selects row in table
    */
   protected void selectRowOf(final SMTestProxy proxy) {
     SMRunnerUtil.addToInvokeLater(new Runnable() {
       public void run() {
         final int indexOf = myTableModel.getIndexOf(proxy);
         if (indexOf > -1) {
           final int rowIndex = myStatisticsTableView.convertRowIndexToView(indexOf);
           myStatisticsTableView.setRowSelectionInterval(rowIndex, rowIndex >= 0 ? rowIndex : 0);
           // Scroll to visible
           TableUtil.scrollSelectionToVisible(myStatisticsTableView);
         }
       }
     });
   }

   @Nullable
   protected SMTestProxy getSelectedItem() {
     return myStatisticsTableView.getSelectedObject();
   }

   protected List<SMTestProxy> getTableItems() {
     return myTableModel.getItems();
   }

   private void findAndSelectInTable(final SMTestProxy proxy) {
     SMRunnerUtil.addToInvokeLater(new Runnable() {
       public void run() {
         final int rowIndex = myTableModel.getIndexOf(proxy);
         if (rowIndex >= 0) {
           final int rowIndexToView = myStatisticsTableView.convertRowIndexToView(rowIndex);
           myStatisticsTableView.setRowSelectionInterval(rowIndexToView, rowIndexToView);
         }
       }
     });
   }

   private void fireOnPropagateSelection(final SMTestProxy selectedTestProxy) {
     for (PropagateSelectionHandler handler : myPropagateSelectionHandlers) {
       handler.handlePropagateSelectionRequest(selectedTestProxy, this, true);
     }
   }

   private void createUIComponents() {
     myStatisticsTableView = new TableView<SMTestProxy>();
   }

   private void showInTableAndSelectRow(final SMTestProxy suite, final SMTestProxy suiteProxy) {
     selectProxy(suite);
     selectRowOf(suiteProxy);
   }

   public void doDispose() {
     BaseTableView.store(myStorage, myStatisticsTableView);
   }
 }
	/*
	* Copyright 2000-2009 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 com.intellij.execution.testframework.sm.runner.ui.statistics;

	import com.intellij.execution.testframework.TestFrameworkRunningModel;
	import com.intellij.execution.testframework.TestsUIUtil;
	import com.intellij.execution.testframework.sm.SMRunnerUtil;
	import com.intellij.execution.testframework.sm.runner.SMTRunnerEventsAdapter;
	import com.intellij.execution.testframework.sm.runner.SMTRunnerEventsListener;
	import com.intellij.execution.testframework.sm.runner.SMTestProxy;
	import com.intellij.execution.testframework.sm.runner.ui.PropagateSelectionHandler;
	import com.intellij.ide.DataManager;
	import com.intellij.openapi.actionSystem.ActionPlaces;
	import com.intellij.openapi.actionSystem.DataKey;
	import com.intellij.openapi.actionSystem.DataProvider;
	import com.intellij.openapi.actionSystem.IdeActions;
	import com.intellij.openapi.project.Project;
	import com.intellij.openapi.wm.IdeFocusManager;
	import com.intellij.ui.DoubleClickListener;
	import com.intellij.ui.PopupHandler;
	import com.intellij.ui.TableUtil;
	import com.intellij.ui.table.BaseTableView;
	import com.intellij.ui.table.TableView;
	import com.intellij.util.config.Storage;
	import com.intellij.util.containers.ContainerUtil;
	import org.jetbrains.annotations.NonNls;
	import org.jetbrains.annotations.NotNull;
	import org.jetbrains.annotations.Nullable;

	import javax.swing.*;
	import java.awt.event.InputEvent;
	import java.awt.event.KeyEvent;
	import java.awt.event.MouseEvent;
	import java.util.Collection;
	import java.util.List;

	/**
	* @author Roman Chernyatchik
	*/
	public class StatisticsPanel implements DataProvider {
	public static final DataKey<StatisticsPanel> SM_TEST_RUNNER_STATISTICS = DataKey.create("SM_TEST_RUNNER_STATISTICS");

	private TableView<SMTestProxy> myStatisticsTableView;
	private JPanel myContentPane;
	private final Storage.PropertiesComponentStorage myStorage = new Storage.PropertiesComponentStorage("sm_test_statistics_table_columns");

	private final StatisticsTableModel myTableModel;
	private final List<PropagateSelectionHandler> myPropagateSelectionHandlers = ContainerUtil.createLockFreeCopyOnWriteList();
	private final Project myProject;
	private final TestFrameworkRunningModel myFrameworkRunningModel;

	public StatisticsPanel(final Project project, final TestFrameworkRunningModel model) {
	myProject = project;
	myTableModel = new StatisticsTableModel();
	myStatisticsTableView.setModelAndUpdateColumns(myTableModel);
	myFrameworkRunningModel = model;

	final Runnable gotoSuiteOrParentAction = createGotoSuiteOrParentAction();
	new DoubleClickListener() {
	@Override
	protected boolean onDoubleClick(MouseEvent e) {
	gotoSuiteOrParentAction.run();
	return true;
	}
	}.installOn(myStatisticsTableView);

	// Fire selection changed and move focus on SHIFT+ENTER
	final KeyStroke shiftEnterKey = KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, InputEvent.SHIFT_MASK);
	SMRunnerUtil.registerAsAction(shiftEnterKey, "select-test-proxy-in-test-view",
	new Runnable() {
	public void run() {
	showSelectedProxyInTestsTree();
	}
	},
	myStatisticsTableView);

	// Expand selected or go to parent on ENTER
	final KeyStroke enterKey = KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, 0);
	SMRunnerUtil.registerAsAction(enterKey, "go-to-selected-suite-or-parent",
	gotoSuiteOrParentAction,
	myStatisticsTableView);
	// Contex menu in Table
	PopupHandler.installPopupHandler(myStatisticsTableView, IdeActions.GROUP_TESTTREE_POPUP, ActionPlaces.TESTTREE_VIEW_POPUP);
	// set this statistic tab as dataprovider for test's table view
	DataManager.registerDataProvider(myStatisticsTableView, this);
	}

	public void addPropagateSelectionListener(final PropagateSelectionHandler handler) {
	myPropagateSelectionHandlers.add(handler);
	}

	public JPanel getContentPane() {
	return myContentPane;
	}

	public SMTRunnerEventsListener createTestEventsListener() {
	return new SMTRunnerEventsAdapter() {
	@Override
	public void onSuiteStarted(@NotNull final SMTestProxy suite) {
	if (myTableModel.shouldUpdateModelBySuite(suite)) {
	updateAndRestoreSelection();
	}
	}

	@Override
	public void onSuiteFinished(@NotNull final SMTestProxy suite) {
	if (myTableModel.shouldUpdateModelBySuite(suite)) {
	updateAndRestoreSelection();
	}
	}

	@Override
	public void onTestStarted(@NotNull final SMTestProxy test) {
	if (myTableModel.shouldUpdateModelByTest(test)) {
	updateAndRestoreSelection();
	}
	}

	@Override
	public void onTestFinished(@NotNull final SMTestProxy test) {
	if (myTableModel.shouldUpdateModelByTest(test)) {
	updateAndRestoreSelection();
	}
	}

	private void updateAndRestoreSelection() {
	SMRunnerUtil.addToInvokeLater(new Runnable() {
	public void run() {
	BaseTableView.restore(myStorage, myStatisticsTableView);
	// statisticsTableView can be null in JUnit tests
	final SMTestProxy oldSelection = myStatisticsTableView.getSelectedObject();

	// update module
	myTableModel.updateModel();

	// restore selection if it is possible
	if (oldSelection != null) {
	final int newRow = myTableModel.getIndexOf(oldSelection);
	if (newRow > -1) {
	myStatisticsTableView.setRowSelectionInterval(newRow, newRow);
	}
	}
	}
	});
	}
	};
	}

	public Object getData(@NonNls final String dataId) {
	if (SM_TEST_RUNNER_STATISTICS.is(dataId)) {
	return this;
	}
	return TestsUIUtil.getData(getSelectedItem(), dataId, myFrameworkRunningModel);
	}

	/**
	* On event - change selection and probably requests focus. Is used when we want
	* navigate from other component to this
	* @return Listener
	*/
	public PropagateSelectionHandler createSelectMeListener() {
	return new PropagateSelectionHandler() {
	public void handlePropagateSelectionRequest(@Nullable final SMTestProxy selectedTestProxy,
	@NotNull final Object sender,
	final boolean requestFocus) {
	selectProxy(selectedTestProxy, sender, requestFocus);
	}
	};
	}

	public void selectProxy(@Nullable final SMTestProxy selectedTestProxy,
	@NotNull final Object sender,
	final boolean requestFocus) {
	SMRunnerUtil.addToInvokeLater(new Runnable() {
	public void run() {
	// Select tab if focus was requested
	if (requestFocus) {
	IdeFocusManager.getInstance(myProject).requestFocus(myStatisticsTableView, true);
	}

	// Select proxy in table
	selectProxy(selectedTestProxy);
	}
	});
	}

	protected void showSelectedProxyInTestsTree() {
	final Collection<SMTestProxy> proxies = myStatisticsTableView.getSelection();
	if (proxies.isEmpty()) {
	return;
	}
	final SMTestProxy proxy = proxies.iterator().next();
	myStatisticsTableView.clearSelection();
	fireOnPropagateSelection(proxy);
	}

	protected Runnable createGotoSuiteOrParentAction() {
	// Expand selected or go to parent
	return new Runnable() {
	public void run() {
	final SMTestProxy selectedProxy = getSelectedItem();
	if (selectedProxy == null) {
	return;
	}

	final int i = myStatisticsTableView.getSelectedRow();
	assert i >= 0; //because something is selected

	// if selected element is suite - we should expand it
	if (selectedProxy.isSuite()) {
	// expand and select first (Total) row
	showInTableAndSelectRow(selectedProxy, selectedProxy);
	}
	}
	};
	}

	protected void selectProxy(@Nullable final SMTestProxy selectedTestProxy) {
	// Send event to model
	myTableModel.updateModelOnProxySelected(selectedTestProxy);

	// Now we want to select proxy in table (if it is possible)
	if (selectedTestProxy != null) {
	findAndSelectInTable(selectedTestProxy);
	}
	}
	/**
	* Selects row in table
	* @param rowIndex Row's index
	*/
	protected void selectRow(final int rowIndex) {
	SMRunnerUtil.addToInvokeLater(new Runnable() {
	public void run() {
	// updates model
	myStatisticsTableView.setRowSelectionInterval(rowIndex, rowIndex);

	// Scroll to visible
	TableUtil.scrollSelectionToVisible(myStatisticsTableView);
	}
	});
	}

	/**
	* Selects row in table
	*/
	protected void selectRowOf(final SMTestProxy proxy) {
	SMRunnerUtil.addToInvokeLater(new Runnable() {
	public void run() {
	final int indexOf = myTableModel.getIndexOf(proxy);
	if (indexOf > -1) {
	final int rowIndex = myStatisticsTableView.convertRowIndexToView(indexOf);
	myStatisticsTableView.setRowSelectionInterval(rowIndex, rowIndex >= 0 ? rowIndex : 0);
	// Scroll to visible
	TableUtil.scrollSelectionToVisible(myStatisticsTableView);
	}
	}
	});
	}

	@Nullable
	protected SMTestProxy getSelectedItem() {
	return myStatisticsTableView.getSelectedObject();
	}

	protected List<SMTestProxy> getTableItems() {
	return myTableModel.getItems();
	}

	private void findAndSelectInTable(final SMTestProxy proxy) {
	SMRunnerUtil.addToInvokeLater(new Runnable() {
	public void run() {
	final int rowIndex = myTableModel.getIndexOf(proxy);
	if (rowIndex >= 0) {
	final int rowIndexToView = myStatisticsTableView.convertRowIndexToView(rowIndex);
	myStatisticsTableView.setRowSelectionInterval(rowIndexToView, rowIndexToView);
	}
	}
	});
	}

	private void fireOnPropagateSelection(final SMTestProxy selectedTestProxy) {
	for (PropagateSelectionHandler handler : myPropagateSelectionHandlers) {
	handler.handlePropagateSelectionRequest(selectedTestProxy, this, true);
	}
	}

	private void createUIComponents() {
	myStatisticsTableView = new TableView<SMTestProxy>();
	}

	private void showInTableAndSelectRow(final SMTestProxy suite, final SMTestProxy suiteProxy) {
	selectProxy(suite);
	selectRowOf(suiteProxy);
	}

	public void doDispose() {
	BaseTableView.store(myStorage, myStatisticsTableView);
	}
	}