Increase code coverage

test ZusammenConnector
Issue-ID: SDC-2690

Signed-off-by: shrek2000 <oren.kleks@amdocs.com>
Change-Id: Ide319e4dda496666711cb4b5597cc166ffdd4e2c

1 files changed, 31 insertions, 0 deletions

diff --git a/zusammen-lib/src/test/java/org/onap/sdc/common/zusammen/persistence/ZusammenConnectorTest.java b/zusammen-lib/src/test/java/org/onap/sdc/common/zusammen/persistence/ZusammenConnectorTest.java
index ead9b98..cbf41ab 100644
--- a/zusammen-lib/src/test/java/org/onap/sdc/common/zusammen/persistence/ZusammenConnectorTest.java
+++ b/zusammen-lib/src/test/java/org/onap/sdc/common/zusammen/persistence/ZusammenConnectorTest.java
@@ -21,6 +21,8 @@ import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.fail;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
 import com.amdocs.zusammen.adaptor.inbound.api.health.HealthAdaptor;
@@ -38,6 +40,7 @@ import com.amdocs.zusammen.datatypes.Space;
 import com.amdocs.zusammen.datatypes.item.Info;
 import com.amdocs.zusammen.datatypes.item.Item;
 import com.amdocs.zusammen.datatypes.item.ItemVersion;
+import com.amdocs.zusammen.datatypes.item.ItemVersionData;
 import com.amdocs.zusammen.datatypes.response.ErrorCode;
 import com.amdocs.zusammen.datatypes.response.Module;
 import com.amdocs.zusammen.datatypes.response.Response;
@@ -237,6 +240,34 @@ public class ZusammenConnectorTest {
         assertEquals(myId, itemVersion.getId().getValue());
     }
 
+    @Test
+    public void testCreateVersionVersion() {
+        ItemVersionAdaptor itemAdaptor = mock(ItemVersionAdaptor.class);
+        when(versionAdaptorFactoryMock.createInterface(sessionContext)).thenReturn(itemAdaptor);
+        ItemVersionData itemVersionData = new ItemVersionData();
+        Id id = new Id();
+        String myId = "myId";
+        id.setValue(myId);
+        Response<Id> response = new Response<>(id);
+        when(itemAdaptor.create(sessionContext, id, id, itemVersionData)).thenReturn(response);
+        Id outId = zusammenConnector.createVersion(sessionContext, id, id, itemVersionData);
+        assertEquals(myId, outId.getValue());
+    }
+
+    @Test
+    public void testUpdateVersionVersion() {
+        ItemVersionAdaptor itemAdaptor = mock(ItemVersionAdaptor.class);
+        when(versionAdaptorFactoryMock.createInterface(sessionContext)).thenReturn(itemAdaptor);
+        ItemVersionData itemVersionData = new ItemVersionData();
+        Id id = new Id();
+        Response<Void> response = new Response<>(null);
+        Response<Void> spyResponse = spy(response);
+        when(itemAdaptor.update(sessionContext, id, id, itemVersionData)).thenReturn(spyResponse);
+        zusammenConnector.updateVersion(sessionContext, id, id, itemVersionData);
+        verify(spyResponse).getValue();
+
+    }
+
     private void setResponseErrorReturnCode(Response response) {
         response.setReturnCode(new ReturnCode(ErrorCode.CL_ELEMENT_GET, Module.ZSTM, "bla bla", null));
     }