Create new tests for adding Widgets to VfModule
[aai/babel.git] / src / test / java / org / onap / aai / babel / xml / generator / model / TestVfModule.java
index 5572fe2..a07a68f 100644 (file)
  * limitations under the License.
  * ============LICENSE_END=========================================================
  */
+
 package org.onap.aai.babel.xml.generator.model;
 
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.CoreMatchers.notNullValue;
+import static org.hamcrest.collection.IsCollectionWithSize.hasSize;
 import static org.junit.Assert.assertThat;
 
 import java.io.FileNotFoundException;
@@ -31,24 +33,29 @@ import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Properties;
-import org.junit.Before;
+
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.onap.aai.babel.xml.generator.data.WidgetConfigurationUtil;
 import org.onap.aai.babel.xml.generator.model.Widget.Type;
 
 /**
- * Direct tests of the Model class VfModule so as to improve code coverage
+ * Direct tests of the VfModule Model class to improve code coverage.
  */
 public class TestVfModule {
 
     static {
-        if (System.getProperty("APP_HOME") == null) {
-            System.setProperty("APP_HOME", ".");
-        }
+        System.setProperty("APP_HOME", ".");
     }
 
-    @Before
-    public void setup() throws FileNotFoundException, IOException {
+    /**
+     * Load the Widget to UUID mappings from the Artifact Generator properties.
+     * 
+     * @throws FileNotFoundException if the properties file is missing
+     * @throws IOException if the properties file is not loaded
+     */
+    @BeforeClass
+    public static void setup() throws FileNotFoundException, IOException {
         InputStream in = TestVfModule.class.getClassLoader().getResourceAsStream("artifact-generator.properties");
         Properties properties = new Properties();
         properties.load(in);
@@ -56,66 +63,266 @@ public class TestVfModule {
         WidgetConfigurationUtil.setConfig(properties);
     }
 
+    /**
+     * Call equals() and hashCode() methods for code coverage.
+     */
     @Test
-    public void testCreateVfModule() {
-        VfModule vf = new VfModule();
-        Map<String, String> modelIdentInfo = new HashMap<>();
-        modelIdentInfo.put("UUID", "dummy_uuid");
-        vf.populateModelIdentificationInformation(modelIdentInfo);
-        assertThat(vf.hashCode(), is(notNullValue()));
-        assertThat(vf.equals(vf), is(true));
+    public void testEqualsHashCode() {
+        VfModule vfModule = createNewVfModule();
+        populateIdentInfo(vfModule);
+        assertThat(vfModule.hashCode(), is(notNullValue()));
+        assertThat(vfModule.equals(vfModule), is(true));
         // Tests that the overridden equals() method correctly returns false for a different type of Object
         // This is necessary to achieve complete code coverage
-        assertThat(vf.equals("string"), is(false)); // NOSONAR
+        assertThat(vfModule.equals("string"), is(false)); // NOSONAR
     }
 
     @Test
-    public void testNonMemberWidgetToVf() {
-        VfModule vf = new VfModule();
-        Widget widget = Widget.getWidget(Type.SERVICE);
-        vf.setMembers(Collections.singletonList(widget.getId()));
-        vf.addWidget(widget);
+    public void testAddVServerWidgetToVf() {
+        assertAddWidget(createNewVfModule(), Type.VSERVER);
     }
 
     @Test
     public void testAddServiceWidgetToVf() {
-        VfModule vf = new VfModule();
-        addWidgetToModule(vf, Type.SERVICE);
+        assertAddWidget(createNewVfModule(), Type.SERVICE);
     }
 
+    /**
+     * Add a new Widget to a VF Module, where the Widget is NOT set as a member. N.B. For the current VF Module
+     * implementation the actual Widget type is not important.
+     */
     @Test
-    public void testAddVServerWidgetToVf() {
-        VfModule vf = new VfModule();
-        addWidgetToModule(vf, Type.VSERVER);
+    public void testNonMemberWidgetToVf() {
+        VfModule vfModule = createNewVfModule();
+        assertThat(vfModule.addWidget(createNewWidget(Type.SERVICE)), is(false));
+        assertNumberOfWidgets(vfModule, 0);
     }
 
+    /**
+     * OAM Network is specifically excluded from a VF Module.
+     */
     @Test
-    public void testAddLIntfWidgetToVf() {
-        VfModule vf = new VfModule();
-        addWidgetToModule(vf, Type.LINT);
-        addWidgetToModule(vf, Type.VSERVER);
-        addWidgetToModule(vf, Type.LINT);
+    public void testAddOamNetworkWidgetToVf() {
+        VfModule vfModule = createNewVfModule();
+        assertThat(createNewWidgetForModule(vfModule, Type.OAM_NETWORK), is(false));
+        assertNumberOfWidgets(vfModule, 0);
     }
 
+    /**
+     * Add a Volume Widget to a VF Module via a vserver Widget.
+     * 
+     * <li>Create a VF Module</li>
+     * <li>Add a Volume Widget</li>
+     * <li>Add a vserver Widget</li>
+     * <li>Check that the Volume Widget appears under the vserver</li>
+     */
     @Test
     public void testAddVolumeWidgetToVf() {
-        VfModule vf = new VfModule();
-        addWidgetToModule(vf, Type.VOLUME);
-        addWidgetToModule(vf, Type.VSERVER);
-        addWidgetToModule(vf, Type.VOLUME);
+        VfModule vfModule = createNewVfModule();
+
+        // Adding a Volume widget has no effect until a vserver widget is added.
+        assertAddWidget(vfModule, Type.VOLUME);
+        assertNumberOfWidgets(vfModule, 0);
+
+        final int vserverBaseWidgetCount = createVserverForVf(vfModule);
+
+        // The vserver now has Volume as well.
+        assertNumberOfWidgets(vfModule.vserver, vserverBaseWidgetCount + 1);
+
+        // Adding another instance of a vserver widget fails.
+        assertFailToAddWidget(vfModule, Type.VSERVER);
+        assertNumberOfWidgets(vfModule, 1);
+
+        // Adding another Volume widget is always treated as successful.
+        assertAddWidget(vfModule, Type.VOLUME);
+        // Assert that no additional Widgets are actually present.
+        assertNumberOfWidgets(vfModule, 1);
+        assertNumberOfWidgets(vfModule.vserver, vserverBaseWidgetCount + 1);
     }
 
+    /**
+     * Add an L-Interface Widget to a VF Module via a vserver Widget.
+     * 
+     * <li>Create a VF Module</li>
+     * <li>Add an L-Interface Widget</li>
+     * <li>Add a vserver Widget</li>
+     * <li>Check that the L-Interface Widget appears under the vserver</li>
+     */
     @Test
-    public void testAddOamNetworkWidgetToVf() {
-        VfModule vf = new VfModule();
-        addWidgetToModule(vf, Type.OAM_NETWORK);
+    public void testAddLinterfaceWidgetToVf() {
+        VfModule vfModule = createNewVfModule();
+
+        // Adding an L-Interface widget has no effect until a vserver widget is added.
+        assertFailToAddWidget(vfModule, Type.LINT);
+        assertNumberOfWidgets(vfModule, 0);
+
+        final int vserverBaseWidgetCount = createVserverForVf(vfModule);
+
+        // The vserver now has an L-Interface as well.
+        assertNumberOfWidgets(vfModule.vserver, vserverBaseWidgetCount + 1);
+
+        // Adding another instance of a vserver widget fails.
+        assertFailToAddWidget(vfModule, Type.VSERVER);
+        assertNumberOfWidgets(vfModule, 1);
+
+        // Adding an L-Interface widget is always treated as successful when a vserver exists.
+        assertAddWidget(vfModule, Type.LINT);
+        // Assert that no additional Widgets are actually present.
+        assertNumberOfWidgets(vfModule, 1);
+        assertNumberOfWidgets(vfModule.vserver, vserverBaseWidgetCount + 1);
     }
 
-    private void addWidgetToModule(VfModule vfModule, Type widgeType) {
-        Widget widget = Widget.getWidget(widgeType);
+    /**
+     * Add a Volume and an L-Interface Widget to a VF Module via a vserver Widget.
+     * 
+     * <li>Create a VF Module</li>
+     * <li>Add a Volume Widget</li>
+     * <li>Add an L-Interface Widget</li>
+     * <li>Add a vserver Widget</li>
+     * <li>Check that both Widgets appear under the vserver</li>
+     */
+    @Test
+    public void testAddVolumeAndLinterfaceWidgetToVf() {
+        VfModule vfModule = createNewVfModule();
+
+        // Adding a Volume widget has no effect until a vserver widget is added.
+        assertAddWidget(vfModule, Type.VOLUME);
+        assertNumberOfWidgets(vfModule, 0);
+
+        // Adding an L-Interface widget has no effect until a vserver widget is added.
+        assertFailToAddWidget(vfModule, Type.LINT);
+        assertNumberOfWidgets(vfModule, 0);
+
+        final int vserverBaseWidgetCount = createVserverForVf(vfModule);
+
+        // The vserver now has both Volume and L-Interface.
+        assertNumberOfWidgets(vfModule.vserver, vserverBaseWidgetCount + 2);
+
+        // Adding another instance of a vserver widget fails.
+        assertFailToAddWidget(vfModule, Type.VSERVER);
+        assertNumberOfWidgets(vfModule, 1);
+
+        // Add new instances (with no effect).
+        assertAddWidget(vfModule, Type.VOLUME);
+        assertAddWidget(vfModule, Type.LINT);
+        // Assert that no additional Widgets are in fact present.
+        assertNumberOfWidgets(vfModule, 1);
+        assertNumberOfWidgets(vfModule.vserver, vserverBaseWidgetCount + 2);
+    }
+
+    private void assertNumberOfWidgets(Model model, int numberOfWidgets) {
+        assertThat(model.getWidgets(), hasSize(numberOfWidgets));
+    }
+
+    /**
+     * Use the static Factory method to create a new Widget.
+     *
+     * @param widgetType type of Widget to create
+     * @return a new Widget
+     */
+    private Widget createNewWidget(Type widgetType) {
+        return Widget.getWidget(widgetType);
+    }
+
+    /**
+     * Create a new VF Module that contains zero widgets and has no members.
+     *
+     * @return new VF Module resource
+     */
+    private VfModule createNewVfModule() {
+        VfModule vfModule = new VfModule();
+        assertNumberOfWidgets(vfModule, 0);
+        return vfModule;
+    }
+
+    /**
+     * Set up some dummy Model Identification properties.
+     *
+     * @param vfModule to be populated
+     */
+    private void populateIdentInfo(VfModule vfModule) {
+        Map<String, String> modelIdentInfo = new HashMap<>();
+        modelIdentInfo.put("UUID", "dummy_uuid");
+        vfModule.populateModelIdentificationInformation(modelIdentInfo);
+    }
+
+    /**
+     * Create a new Widget and assert that it is successfully added to the VF Module.
+     *
+     * @param vfModule the VF Module to update
+     * @param widgetType the type of Widget to create and add
+     */
+    private void assertAddWidget(VfModule vfModule, Type widgetType) {
+        assertThat(createNewWidgetForModule(vfModule, widgetType), is(true));
+    }
+
+    /**
+     * Create a new Widget and assert that it cannot be added to the VF Module.
+     *
+     * @param vfModule the VF Module
+     * @param widgetType the type of Widget to create and attempt to add
+     */
+    private void assertFailToAddWidget(VfModule vfModule, Type widgetType) {
+        assertThat(createNewWidgetForModule(vfModule, widgetType), is(false));
+    }
+
+    /**
+     * Create a new widget, make it a member of the VF Module, then try to add it.
+     *
+     * @param vfModule the VF Module to update
+     * @param widgetType the type of Widget to create and attempt to add
+     * @return whether or not the Widget was added to the module
+     */
+    private boolean createNewWidgetForModule(VfModule vfModule, Type widgetType) {
+        Widget widget = createNewWidget(widgetType);
+        setWidgetAsMember(vfModule, widget);
+        return vfModule.addWidget(widget);
+    }
+
+    /**
+     * Make the specified Widget the sole member of the VF Module. This is achieved by first adding the Widget's own ID
+     * to its set of keys, and by then setting the VF Module's members to a Singleton List comprised of this ID. These
+     * updates allow the Widget to be successfully added to the VF Module. (Non-member Widgets cannot be added.)
+     *
+     * @param vfModule the module for which members are overwritten
+     * @param widget the widget to be set as the member
+     */
+    private void setWidgetAsMember(VfModule vfModule, Widget widget) {
         String id = widget.getId();
         widget.addKey(id);
         vfModule.setMembers(Collections.singletonList(id));
-        vfModule.addWidget(widget);
+    }
+
+    /**
+     * Create a vserver widget and add it to the specified VF Module.
+     *
+     * @param vfModule the VF Module to update
+     * @return the number of Widgets present in the vserver on creation
+     */
+    private int createVserverForVf(VfModule vfModule) {
+        VServerWidget vserverWidget = (VServerWidget) createNewWidget(Type.VSERVER);
+        assertNumberOfWidgets(vfModule, 0);
+        final int initialWidgetCount = addVserverToVf(vfModule, vserverWidget);
+        assertNumberOfWidgets(vfModule, 1);
+        return initialWidgetCount;
+    }
+
+    /**
+     * Add the specified vserver to the specified VF Module.
+     * 
+     * @param vfModule the VF Module to update
+     * @param vserverWidget the Widget to add
+     * @return initial widget count for the vserver Widget
+     */
+    private int addVserverToVf(VfModule vfModule, VServerWidget vserverWidget) {
+        // A vserver (initially) has Flavor, Image, Tenant and Vfc.
+        final int initialWidgetCount = 4;
+        assertNumberOfWidgets(vserverWidget, initialWidgetCount);
+
+        // Add the vserver to the VF Module.
+        setWidgetAsMember(vfModule, vserverWidget);
+        assertThat(vfModule.addWidget(vserverWidget), is(true));
+
+        return initialWidgetCount;
     }
 }