TableViewRoleTest.java

/*******************************************************************************
 * Mission Control Technologies, Copyright (c) 2009-2012, United States Government
 * as represented by the Administrator of the National Aeronautics and Space 
 * Administration. All rights reserved.
 *
 * The MCT platform is 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.
 *
 * MCT includes source code licensed under additional open source licenses. See 
 * the MCT Open Source Licenses file included with this distribution or the About 
 * MCT Licenses dialog available at runtime from the MCT Help menu for additional 
 * information. 
 *******************************************************************************/
package gov.nasa.arc.mct.table.view;

import gov.nasa.arc.mct.components.AbstractComponent;
import gov.nasa.arc.mct.components.ExtendedProperties;
import gov.nasa.arc.mct.gui.NamingContext;
import gov.nasa.arc.mct.gui.View;
import gov.nasa.arc.mct.policy.ExecutionResult;
import gov.nasa.arc.mct.policy.PolicyContext;
import gov.nasa.arc.mct.roles.events.PropertyChangeEvent;
import gov.nasa.arc.mct.services.component.PolicyManager;
import gov.nasa.arc.mct.services.component.ViewInfo;
import gov.nasa.arc.mct.services.component.ViewType;
import gov.nasa.arc.mct.table.access.ServiceAccess;
import gov.nasa.arc.mct.table.gui.LabeledTable;
import gov.nasa.arc.mct.table.model.AbbreviatingTableLabelingAlgorithm;

import java.beans.PropertyChangeListener;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

import javax.swing.JTable;
import javax.swing.ListSelectionModel;

import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.testng.Assert;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

@SuppressWarnings("javadoc")
public class TableViewRoleTest {
	
	private TableViewManifestation manifestation;
	@Mock
	private AbstractComponent component;
	@Mock
	private PolicyManager policyManager;
	@Mock
	private AbstractComponent row0, row1;
	private List<AbstractComponent> allComponents = new ArrayList<AbstractComponent>();
	
	
	@SuppressWarnings("serial")
	@BeforeMethod
	void init() {
		MockitoAnnotations.initMocks(this);
		allComponents.clear();
		Mockito.when(component.getDisplayName()).thenReturn("test");
		List<AbstractComponent> rootChildren = new ArrayList<AbstractComponent>();
		Mockito.when(component.getComponents()).thenReturn(rootChildren);
		ExecutionResult trueResult = new ExecutionResult(null, true, "");
		Mockito.when(policyManager.execute(Mockito.anyString(), Mockito.any(PolicyContext.class))).thenReturn(trueResult);
		(new ServiceAccess()).bind(policyManager);
		List<AbstractComponent> row0Children = new ArrayList<AbstractComponent>();
		List<AbstractComponent> row1Children = new ArrayList<AbstractComponent>();
		
		Mockito.when(row0.getDisplayName()).thenReturn("row0");
		Mockito.when(row0.getComponents()).thenReturn(row0Children);
		rootChildren.add(row0);
		addComponents(2, row0Children, allComponents);
		
		Mockito.when(row1.getDisplayName()).thenReturn("row1");
		Mockito.when(row1.getComponents()).thenReturn(row1Children);
		addComponents(2, row1Children, allComponents);
		rootChildren.add(row1);
		final ExtendedProperties viewProps = new ExtendedProperties();
		manifestation =  new TableViewManifestation(component,new ViewInfo(TableViewManifestation.class,"",ViewType.CENTER)) {
			@Override
			public ExtendedProperties getViewProperties() {
				return viewProps;
			}
		};
	}
	
	private void addComponents(int components, List<AbstractComponent> childList, List<AbstractComponent> createdComponents) {
		for (int i = 0; i < components; i++) {
			AbstractComponent ac = Mockito.mock(AbstractComponent.class);
			Mockito.when(ac.getComponents()).thenReturn(Collections.<AbstractComponent>emptyList());
			Mockito.when(ac.isLeaf()).thenReturn(true);
			final View mvm = Mockito.mock(View.class);
			ViewInfo vi = new ViewInfo(TableViewManifestation.class, "", ViewType.EMBEDDED) {
				@Override
				public View createView(AbstractComponent component) {
					return mvm;
				}
			};
			Mockito.when(ac.getViewInfos(ViewType.NODE)).thenReturn(Collections.singleton(vi));
			Mockito.when(ac.getViewInfos(ViewType.EMBEDDED)).thenReturn(Collections.singleton(vi));
			Mockito.when(ac.getComponentTypeID()).thenReturn("MockComponent");
			Mockito.when(mvm.getManifestedComponent()).thenReturn(ac);
			Mockito.when(mvm.getInfo()).thenReturn(vi);
			childList.add(ac);
			createdComponents.add(ac);
		}
	}
	
	@AfterMethod
	void tearDown() {
		(new ServiceAccess()).unbind(policyManager);
	}
	
	
	@Test
	public void testSelectionListener() throws Exception {
		final AtomicInteger eventsFired = new AtomicInteger(0);
		PropertyChangeListener pcl = new PropertyChangeListener() {
			
			@Override
			public void propertyChange(java.beans.PropertyChangeEvent evt) {
				eventsFired.incrementAndGet();
			}
		};
		
		manifestation.addSelectionChangeListener(pcl);
		Field f = TableViewManifestation.class.getDeclaredField("table");
		f.setAccessible(true);
		LabeledTable labeledTable = (LabeledTable) f.get(manifestation);
		JTable jTable = labeledTable.getTable();
		
		ListSelectionModel rsm = jTable.getSelectionModel();
		ListSelectionModel csm = jTable.getColumnModel().getSelectionModel();
		
		// try combinations to ensure that each selection generates one and only one selection event
		for (int i = 0; i < 2; i++) {
			rsm.setSelectionInterval(i, i);
			eventsFired.set(0);
			for (int j = 0; j < 2; j++) {
				csm.setSelectionInterval(j, j);
				Assert.assertEquals(eventsFired.get(), 1);
				eventsFired.set(0);
				Collection<View> selected = manifestation.getSelectedManifestations();
				Assert.assertEquals(selected.size(), 1);
				Assert.assertSame(selected.iterator().next().getManifestedComponent(), allComponents.get(i*2+j));
			}
		}
	}
	
	String naming; 
	
	@Test
	public void testNamingContext() throws Exception {
		Field f = TableViewManifestation.class.getDeclaredField("labelingAlgorithm");
		f.setAccessible(true);
		f.set(manifestation, new AbbreviatingTableLabelingAlgorithm() {
			@Override
			public void setContextLabels(String... s) { naming = s[0]; }
		});
		
		NamingContext nullContext = Mockito.mock(NamingContext.class);
		NamingContext blankContext = Mockito.mock(NamingContext.class);
		NamingContext specificContext = Mockito.mock(NamingContext.class);
		
		PropertyChangeEvent event = Mockito.mock(PropertyChangeEvent.class);
		
		Mockito.when(nullContext.getContextualName()).thenReturn(null);
		Mockito.when(blankContext.getContextualName()).thenReturn("");
		Mockito.when(specificContext.getContextualName()).thenReturn("Specific");
		
		manifestation.setNamingContext(nullContext);
		manifestation.updateMonitoredGUI(event);
		Assert.assertEquals(naming, "");
		
		manifestation.setNamingContext(blankContext);
		manifestation.updateMonitoredGUI(event);
		Assert.assertEquals(naming, manifestation.getManifestedComponent().getDisplayName());
		
		manifestation.setNamingContext(specificContext);
		manifestation.updateMonitoredGUI(event);
		Assert.assertEquals(naming, "Specific");
		
		manifestation.setNamingContext(null);
		manifestation.updateMonitoredGUI(event);
		Assert.assertEquals(naming, manifestation.getManifestedComponent().getDisplayName());
	}
	
}