1 files changed, 6 insertions, 5 deletions

diff --git a/src/test/java/org/onap/pomba/common/datatypes/ServiceTests.java b/src/test/java/org/onap/pomba/common/datatypes/ServiceTests.java
index a4ebc74..10f9339 100644
--- a/src/test/java/org/onap/pomba/common/datatypes/ServiceTests.java
+++ b/src/test/java/org/onap/pomba/common/datatypes/ServiceTests.java
@@ -22,6 +22,7 @@ import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;

 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 ServiceTests {

@@ -36,16 +37,16 @@ public class ServiceTests {
         dataQuality.setErrorText("Test");

         aService.setDataQuality(dataQuality);

         Attribute attribute = new Attribute();

-        attribute.setName("Attribute");

+        attribute.setName(Name.hostName);

         aService.addAttribute(attribute);

         assertTrue("Service Name doesn't match", aService.getName().equals("new service"));

         assertTrue("Invariant Uuid doesn't match", aService.getInvariantUuid().equals("Invariant Uuid"));

         assertTrue("Uuid doesn't match", aService.getUuid().equals("Uuid"));

         assertTrue("Service data quality status doesn't match", aService.getDataQuality().getStatus().equals(Status.error));

         assertTrue("Service data quality error text doesn't match", aService.getDataQuality().getErrorText().equals("Test"));

-        assertTrue("Service attribute name doesn't match", aService.getAttribute().get(0).getName().equals("Attribute"));

-        List<Attribute> attributeList = aService.getAttribute();

-        aService.setAttribute(attributeList);

-        assertEquals(aService.getAttribute().size(), 1);

+        assertTrue("Service attribute name doesn't match", aService.getAttributes().get(0).getName().equals(Name.hostName));

+        List<Attribute> attributeList = aService.getAttributes();

+        aService.setAttributes(attributeList);

+        assertEquals(aService.getAttributes().size(), 1);

     }

 }