Merge pull request #638 from Nikhil1Reddy/mydevelop4

MOSIP-28140 saveDataShareToken testcases
mosip · Jul 12, 2023 · 25be895 · 25be895
2 parents 3b3ccaa + 4ab99f1
commit 25be895
Show file tree

Hide file tree

Showing 2 changed files with 71 additions and 0 deletions.
diff --git a/...it/src/test/java/io/mosip/compliance/toolkit/controllers/ABISDataShareControllerTest.java b/...it/src/test/java/io/mosip/compliance/toolkit/controllers/ABISDataShareControllerTest.java
@@ -3,6 +3,7 @@
 import io.mosip.compliance.toolkit.dto.abis.DataShareExpireRequest;
 import io.mosip.compliance.toolkit.dto.abis.DataShareRequestDto;
 import io.mosip.compliance.toolkit.dto.abis.DataShareResponseWrapperDto;
+import io.mosip.compliance.toolkit.dto.abis.DataShareSaveTokenRequest;
 import io.mosip.compliance.toolkit.service.ABISDataShareService;
 import io.mosip.compliance.toolkit.util.RequestValidator;
 import io.mosip.kernel.core.http.RequestWrapper;
@@ -68,4 +69,14 @@ public void expireDataShareUrlTest() {
         Mockito.when(abisDataShareService.expireDataShareUrl(dataShareExpireRequest)).thenReturn(response);
         abisDataShareController.expireDataShareUrl(value, errors);
     }
+
+    @Test
+    public void saveDataShareTokenTest() {
+        RequestWrapper<DataShareSaveTokenRequest> request = new RequestWrapper<>();
+        DataShareSaveTokenRequest dataShareSaveTokenRequest = new DataShareSaveTokenRequest();
+        request.setRequest(dataShareSaveTokenRequest);
+        ResponseWrapper<String> response = new ResponseWrapper<>();
+        Mockito.when(abisDataShareService.saveDataShareToken(request)).thenReturn(response);
+        abisDataShareController.saveDataShareToken(request, errors);
+    }
 }
diff --git a/...e-toolkit/src/test/java/io/mosip/compliance/toolkit/service/ABISDataShareServiceTest.java b/...e-toolkit/src/test/java/io/mosip/compliance/toolkit/service/ABISDataShareServiceTest.java
@@ -2,14 +2,19 @@
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.mosip.commons.khazana.spi.ObjectStoreAdapter;
+import io.mosip.compliance.toolkit.constants.AppConstants;
 import io.mosip.compliance.toolkit.dto.abis.DataShareExpireRequest;
 import io.mosip.compliance.toolkit.dto.abis.DataShareRequestDto;
 import io.mosip.compliance.toolkit.dto.abis.DataShareResponseWrapperDto;
+import io.mosip.compliance.toolkit.dto.abis.DataShareSaveTokenRequest;
+import io.mosip.compliance.toolkit.entity.AbisDataShareTokenEntity;
+import io.mosip.compliance.toolkit.repository.AbisDataShareTokenRepository;
 import io.mosip.compliance.toolkit.repository.BiometricTestDataRepository;
 import io.mosip.compliance.toolkit.util.KeyManagerHelper;
 import io.mosip.compliance.toolkit.util.ObjectMapperConfig;
 import io.mosip.kernel.core.authmanager.authadapter.model.AuthUserDetails;
 import io.mosip.kernel.core.authmanager.authadapter.model.MosipUserDto;
+import io.mosip.kernel.core.http.RequestWrapper;
 import io.mosip.kernel.core.http.ResponseWrapper;
 import org.junit.Assert;
 import org.junit.Before;
@@ -29,6 +34,7 @@
 import org.springframework.web.context.WebApplicationContext;
 
 import java.io.InputStream;
+import java.util.Optional;
 
 import static org.junit.Assert.assertNotNull;
 import static org.mockito.Mockito.mock;
@@ -62,6 +68,9 @@ public class ABISDataShareServiceTest {
     @Mock
     private ObjectMapperConfig objectMapperConfig;
 
+    @Mock
+    private AbisDataShareTokenRepository abisDataShareTokenRepository;
+
     @Mock
     private ObjectMapper mapper;
 
@@ -122,6 +131,57 @@ public void expireDataShareUrlTest() {
         dataShareExpireRequest.setTransactionsAllowed(1);
         abisDataShareService.expireDataShareUrl(dataShareExpireRequest);
     }
+
+    /*
+     * These functions test the saveDataShareToken method
+     */
+    @Test
+    public void testSaveDataShareToken_existingToken() {
+
+        RequestWrapper<DataShareSaveTokenRequest> requestWrapper = new RequestWrapper<>();
+        DataShareSaveTokenRequest dataShareSaveTokenRequest = new DataShareSaveTokenRequest();
+        dataShareSaveTokenRequest.setPartnerId("1234");
+        dataShareSaveTokenRequest.setCtkTestCaseId("5678");
+        dataShareSaveTokenRequest.setCtkTestRunId("9012");
+        dataShareSaveTokenRequest.setToken("token");
+        requestWrapper.setRequest(dataShareSaveTokenRequest);
+
+        AbisDataShareTokenEntity savedEntity = new AbisDataShareTokenEntity();
+        savedEntity.setToken("token");
+        savedEntity.setTestRunId("9012");
+        savedEntity.setTestCaseId("5678");
+        savedEntity.setPartnerId("1234");
+
+        Mockito.when(abisDataShareTokenRepository.findByAllIds(
+                Mockito.any(), Mockito.any(), Mockito.any()))
+                .thenReturn(Optional.of(savedEntity));
+
+        ResponseWrapper<String> responseWrapper = abisDataShareService.saveDataShareToken(requestWrapper);
+
+        Mockito.verify(abisDataShareTokenRepository, Mockito.times(1)).updateResultByToken(
+                AppConstants.SUCCESS, savedEntity.getToken());
+        Assert.assertEquals(AppConstants.SUCCESS, responseWrapper.getResponse());
+    }
+
+    @Test
+    public void testSaveDataShareToken_newToken() {
+        RequestWrapper<DataShareSaveTokenRequest> requestWrapper = new RequestWrapper<>();
+        DataShareSaveTokenRequest dataShareSaveTokenRequest = new DataShareSaveTokenRequest();
+        dataShareSaveTokenRequest.setPartnerId("1234");
+        dataShareSaveTokenRequest.setCtkTestCaseId("5678");
+        dataShareSaveTokenRequest.setCtkTestRunId("9012");
+        dataShareSaveTokenRequest.setToken("token");
+        requestWrapper.setRequest(dataShareSaveTokenRequest);
+
+        Mockito.when(abisDataShareTokenRepository.findByAllIds(
+                Mockito.any(), Mockito.any(), Mockito.any()))
+                .thenReturn(Optional.empty());
+
+        ResponseWrapper<String> responseWrapper = abisDataShareService.saveDataShareToken(requestWrapper);
+
+        Mockito.verify(abisDataShareTokenRepository, Mockito.times(1)).save(
+                Mockito.any(AbisDataShareTokenEntity.class));
+    }
 
     /*
      * This method is used to get MosipUserDto in class