1 files changed, 36 insertions, 3 deletions
diff --git a/catalog-model/src/test/java/org/openecomp/sdc/be/model/operations/impl/AdditionalInformationOperationTest.java b/catalog-model/src/test/java/org/openecomp/sdc/be/model/operations/impl/AdditionalInformationOperationTest.java
index d59b810b9e..8bb4a2940d 100644
--- a/catalog-model/src/test/java/org/openecomp/sdc/be/model/operations/impl/AdditionalInformationOperationTest.java
+++ b/catalog-model/src/test/java/org/openecomp/sdc/be/model/operations/impl/AdditionalInformationOperationTest.java
@@ -7,9 +7,9 @@
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *      http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -20,12 +20,18 @@
 
 package org.openecomp.sdc.be.model.operations.impl;
 
+import com.thinkaurelius.titan.core.TitanVertex;
+import fj.data.Either;
+import static org.assertj.core.api.Assertions.assertThat;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.Mock;
 import org.openecomp.sdc.be.dao.titan.TitanGenericDao;
+import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
+import org.openecomp.sdc.be.model.AdditionalInformationDefinition;
 import org.openecomp.sdc.be.model.ModelTestBase;
 import org.openecomp.sdc.be.model.operations.api.IAdditionalInformationOperation;
 import org.openecomp.sdc.be.model.operations.impl.util.OperationTestsUtil;
@@ -35,13 +41,18 @@ import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
 @RunWith(SpringJUnit4ClassRunner.class)
 @ContextConfiguration("classpath:application-context-test.xml")
 public class AdditionalInformationOperationTest extends ModelTestBase {
-
+    private static final TitanGenericDao titanGenericDao = mock(TitanGenericDao.class);
     private static String USER_ID = "muUserId";
     private static String CATEGORY_NAME = "category/mycategory";
+    @Mock
+    private TitanVertex titanVertex;
 
     @javax.annotation.Resource(name = "titan-generic-dao")
     private TitanGenericDao titanDao;
@@ -70,6 +81,28 @@ public class AdditionalInformationOperationTest extends ModelTestBase {
 
     }
 
+    @Test
+    public void testAddInfoParameter_InvalidId(){
+        Either<AdditionalInformationDefinition, TitanOperationStatus> result;
+        String uid = "uid";
+        String componentId = "componentId";
+        when(titanGenericDao.getVertexByProperty(eq(uid),eq(componentId))).thenReturn(Either.left(titanVertex));
+        result = additionalInformationOperation.addAdditionalInformationParameter
+                (NodeTypeEnum.Resource,componentId,"key","value");
+        assertThat(result.isRight());
+    }
+
+    @Test
+    public void testUpdateInfoParameter_InvalidId(){
+        Either<AdditionalInformationDefinition, TitanOperationStatus> result;
+        String uid = "uid";
+        String componentId = "componentId";
+        when(titanGenericDao.getVertexByProperty(eq(uid),eq(componentId))).thenReturn(Either.left(titanVertex));
+        result = additionalInformationOperation.updateAdditionalInformationParameter
+                (NodeTypeEnum.Resource,componentId,"id","key","value");
+        assertTrue(result.isRight());
+    }
+
     private UserData deleteAndCreateUser(String userId, String firstName, String lastName) {
         UserData userData = new UserData();
         userData.setUserId(userId);