1 files changed, 25 insertions, 4 deletions
diff --git a/src/test/java/org/onap/pomba/common/datatypes/ModelContextTests.java b/src/test/java/org/onap/pomba/common/datatypes/ModelContextTests.java
index bad0523..ee04a08 100644
--- a/src/test/java/org/onap/pomba/common/datatypes/ModelContextTests.java
+++ b/src/test/java/org/onap/pomba/common/datatypes/ModelContextTests.java
@@ -15,6 +15,7 @@
  * limitations under the License.
  * ============LICENSE_END=====================================================
  */
+
 package org.onap.pomba.common.datatypes;
 
 import static org.junit.Assert.assertEquals;
@@ -24,9 +25,29 @@ import java.util.ArrayList;
 import java.util.List;
 
 import org.junit.Test;
+import org.onap.pomba.common.datatypes.Attribute.Name;
+import org.onap.pomba.common.datatypes.DataQuality.Status;
 
 public class ModelContextTests {
     @Test
+    public void testModelContext() {
+        ModelContext modelContext = new ModelContext();
+        DataQuality dataQuality = new DataQuality();
+        dataQuality.setStatus(Status.error);
+        dataQuality.setErrorText("Test");
+        modelContext.setDataQuality(dataQuality);
+        assertTrue("ModelContext data quality status doesn't match", modelContext.getDataQuality().getStatus().equals(Status.error));
+        assertTrue("ModelContext data quality error text doesn't match", modelContext.getDataQuality().getErrorText().equals("Test"));
+        Attribute attribute = new Attribute();
+        attribute.setName(Name.hostName);
+        modelContext.addAttribute(attribute);
+        assertTrue("ModelContext attribute name doesn't match", modelContext.getAttributes().get(0).getName().equals(Name.hostName));
+        List<Attribute> attributeList = modelContext.getAttributes();
+        modelContext.setAttributes(attributeList);
+        assertEquals(modelContext.getAttributes().size(), 1);
+    }
+
+    @Test
     public void testSetService() {
         ModelContext modelContext = new ModelContext();
         Service aService = new Service();
@@ -43,8 +64,8 @@ public class ModelContextTests {
         VF aVF = new VF();
         aVF.setName("VF name");
         vfList.add(aVF);
-        modelContext.setVf(vfList);
-        assertEquals(modelContext.getVf().size(), 1);
+        modelContext.setVfs(vfList);
+        assertEquals(modelContext.getVfs().size(), 1);
     }
 
     @Test
@@ -53,8 +74,8 @@ public class ModelContextTests {
         VF aVF = new VF();
         aVF.setName("VF name");
         modelContext.addVf(aVF);
-        assertEquals(modelContext.getVf().size(), 1);
-        assertTrue("ModelContext VF name doesn't match", modelContext.getVf().get(0).getName().equals("VF name"));
+        assertEquals(modelContext.getVfs().size(), 1);
+        assertTrue("ModelContext VF name doesn't match", modelContext.getVfs().get(0).getName().equals("VF name"));
     }
 
 }