1 files changed, 15 insertions, 0 deletions
diff --git a/src/test/java/org/onap/pomba/common/datatypes/VfTests.java b/src/test/java/org/onap/pomba/common/datatypes/VfTests.java
index 8b2a255..9c17589 100644
--- a/src/test/java/org/onap/pomba/common/datatypes/VfTests.java
+++ b/src/test/java/org/onap/pomba/common/datatypes/VfTests.java
@@ -15,6 +15,7 @@
  * limitations under the License.
  * ============LICENSE_END=====================================================
  */
+
 package org.onap.pomba.common.datatypes;
 
 import static org.junit.Assert.assertEquals;
@@ -24,6 +25,7 @@ import java.util.ArrayList;
 import java.util.List;
 
 import org.junit.Test;
+import org.onap.pomba.common.datatypes.DataQuality.Status;
 
 public class VfTests {
     @Test
@@ -34,11 +36,24 @@ public class VfTests {
         aVF.setInvariantUuid("Invariant Uuid");
         aVF.setUuid("Uuid");
         aVF.setNfNamingCode("NF Naming Code");
+        DataQuality dataQuality = new DataQuality();
+        dataQuality.setStatus(Status.error);
+        dataQuality.setErrorText("Test");
+        aVF.setDataQuality(dataQuality);
+        Attribute attribute = new Attribute();
+        attribute.setName("Attribute");
+        aVF.addAttribute(attribute);
         assertTrue("VF name doesn't match", aVF.getName().equals("VF name"));
         assertTrue("VF Type doesn't match", aVF.getType().equals("Type"));
         assertTrue("VF Invariant Uuid doesn't match", aVF.getInvariantUuid().equals("Invariant Uuid"));
         assertTrue("VF Uuid doesn't match", aVF.getUuid().equals("Uuid"));
         assertTrue("VF NF Naming Code doesn't match", aVF.getNfNamingCode().equals("NF Naming Code"));
+        assertTrue("VF data quality status doesn't match", aVF.getDataQuality().getStatus().equals(Status.error));
+        assertTrue("VF data quality error text doesn't match", aVF.getDataQuality().getErrorText().equals("Test"));
+        assertTrue("VF attribute name doesn't match", aVF.getAttribute().get(0).getName().equals("Attribute"));
+        List<Attribute> attributeList = aVF.getAttribute();
+        aVF.setAttribute(attributeList);
+        assertEquals(aVF.getAttribute().size(), 1);
     }
 
     @Test