Unit test to test the policy associated with the multi-column view

multiColumnTables/src/test/java/org/acme/example/MultiColPolicyTest.java

@@ -0,0 +1,100 @@
+package org.acme.example;
+
+import gov.nasa.arc.mct.components.AbstractComponent;
+import gov.nasa.arc.mct.gui.View;
+import gov.nasa.arc.mct.policy.ExecutionResult;
+import gov.nasa.arc.mct.policy.Policy;
+import gov.nasa.arc.mct.policy.PolicyContext;
+import gov.nasa.arc.mct.services.component.ViewInfo;
+
+import org.acme.example.policy.MultiColViewPolicy;
+import org.acme.example.view.MultiColView;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import org.testng.Assert;
+import org.testng.annotations.BeforeTest;
+import org.testng.annotations.Test;
+
+
+/**
+ * We test the execution of the policy associated with the multi-column view
+ * 
+ * @author Harleigh
+ *
+ */
+public class MultiColPolicyTest {
+
+	private Policy policy;			//This policy will be a MultiColView policy
+	private PolicyContext context;	//Note: PolicyContext is 'final' and so can not be mocked
+
+	@BeforeTest
+	public void setup() {
+		MockitoAnnotations.initMocks(this);
+		policy = new MultiColViewPolicy();
+		context = new PolicyContext();
+	}
+
+	/*Testing the execution of the MultiColView policy.  We are testing for the following behavior:
+	 * *When the viewClass context is MultiColView.class:
+	 *      -Components which are leaves may not be viewed by MultiColView; hence
+	 *       the policy should return false.
+	 *      -Components which are not leaves are viewable by MultiColView; hence
+	 *       the policy should return true.
+	 * *When the viewClass context is not MultiColView.class:
+	 *      -The policy should always return true (regardless of whether the component
+	 *       is or is not a leaf.
+	 */
+	@SuppressWarnings("rawtypes")
+	@Test
+	public void testExecute() {
+		ExecutionResult result = null;
+		AbstractComponent mockComp = Mockito.mock(AbstractComponent.class);
+		ViewInfo mockViewInfo = Mockito.mock(ViewInfo.class);
+
+		/*
+		 * From (the below two lines), calling getProperty with the key 'PolicyContext.PropertyName.TARGET_COMPONENT.getName()'
+		 * returns our mocked component. Similarly calling getProperty with the key
+		 * 'PolicyContext.PropertyName.TARGET_VIEW_INFO.getName()' returns our mocked viewInfo.
+		 */
+		context.setProperty(PolicyContext.PropertyName.TARGET_COMPONENT.getName(), mockComp);
+		context.setProperty(PolicyContext.PropertyName.TARGET_VIEW_INFO.getName(), mockViewInfo);
+
+		//Now we test:
+
+		/*Two tests (for leaf and non-leaf components) when the ViewInfo is from the MultiColView class*/
+		Mockito.when((Class)mockViewInfo.getViewClass()).thenReturn(MultiColView.class);
+			//(1) Test for leaf-components: 
+		    Mockito.when(mockComp.isLeaf()).thenReturn(true);	//mocked component is a leaf
+			result = policy.execute(context);	//context is: the component is a leaf, and the viewInfo is from MultiColView
+			Assert.assertEquals(result.getStatus(), false);		//components which are leaves may not be viewed wrt MultiColView
+			//---end test on leaf-components
+
+			//(2) Test for non-leaf components:
+		    Mockito.when(mockComp.isLeaf()).thenReturn(false);	//mocked component is not a leaf
+			result = policy.execute(context);					//context is: the component is a leaf, and the viewInfo is from MultiColView
+			Assert.assertEquals(result.getStatus(), true);		//components which are leaves are viewable wrt MultiColView
+			//---end test on non-leaf components
+
+		/*Two tests (for leaf and non-leaf components) when the ViewInfo is not from the MultiColView class*/
+		Mockito.when((Class)mockViewInfo.getViewClass()).thenReturn(OnStiltsView.class);
+			//(1) Test for leaf-components: 
+		    Mockito.when(mockComp.isLeaf()).thenReturn(true);	//mocked component is a leaf
+			result = policy.execute(context);	//context is: the component is a leaf, and the viewInfo is not from MultiColView
+			Assert.assertEquals(result.getStatus(), true);		//policy should return true for views that are not MultiCol views
+			//---end test on leaf-components
+
+			//(2) Test for non-leaf components:
+		    Mockito.when(mockComp.isLeaf()).thenReturn(false);	//mocked component is not a leaf
+			result = policy.execute(context);					//context is: the component is a leaf, and the viewInfo is not from MultiColView
+			Assert.assertEquals(result.getStatus(), true);		//policy should return true for views that are not MultiCol views
+			//---end test on non-leaf components
+	}
+
+	/*
+	 * This exists so we can test the MultiColView policy when executing the policy when the context has the viewInfo not being-from
+	 * the MultiCol class.  That is  we just needed a view which is not the view of the MultiColView class (Also, when you are on stilts,
+	 * your view is extended)
+	 */
+	@SuppressWarnings("serial")
+	private class OnStiltsView extends View {}
+}