vid-app-common/src/test/java/org/onap/vid/services/BulkInstantiationServiceImplTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * VID
   4  * ================================================================================
   5  * Copyright (C) 2018 Nokia. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.vid.services;
  21
  22 import static org.mockito.Mockito.doReturn;
  23 import static org.mockito.Mockito.reset;
  24 import static org.mockito.Mockito.when;
  25 import static org.mockito.MockitoAnnotations.initMocks;
  26
  27 import java.util.ArrayList;
  28 import java.util.HashMap;
  29 import java.util.List;
  30 import java.util.Map;
  31 import java.util.UUID;
  32 import javax.ws.rs.NotFoundException;
  33 import org.mockito.InjectMocks;
  34 import org.mockito.Mock;
  35 import org.onap.vid.job.Job;
  36 import org.onap.vid.job.JobAdapter;
  37 import org.onap.vid.job.JobsBrokerService;
  38 import org.onap.vid.job.impl.JobDaoImpl;
  39 import org.onap.vid.model.JobBulk;
  40 import org.onap.vid.model.JobModel;
  41 import org.testng.Assert;
  42 import org.testng.annotations.BeforeMethod;
  43 import org.testng.annotations.BeforeSuite;
  44 import org.testng.annotations.Test;
  45
  46 public class BulkInstantiationServiceImplTest {
  47
  48   @Mock
  49   private JobsBrokerService jobsBrokerService;
  50
  51   @Mock
  52   private JobAdapter jobAdapter;
  53
  54   @InjectMocks
  55   private BulkInstantiationServiceImpl testSubject;
  56
  57   @BeforeSuite
  58   public void before() {
  59     initMocks(this);
  60   }
  61
  62   @BeforeMethod
  63   public void resetMocks() {
  64     reset(jobsBrokerService);
  65     reset(jobAdapter);
  66   }
  67
  68   @Test
  69   public void saveBulkTest() {
  70     UUID uuid = UUID.randomUUID();
  71     Map<String, Object> bulkRequest = new HashMap<>();
  72     List<Job> jobList = new ArrayList<>();
  73     jobList.add(createJob(uuid));
  74     when(jobAdapter.createBulkOfJobs(bulkRequest)).thenReturn(jobList);
  75
  76     JobBulk jobBulk = createJobBulk(jobList);
  77     when(jobAdapter.toModelBulk(jobList)).thenReturn(jobBulk);
  78
  79     JobBulk result = testSubject.saveBulk(bulkRequest);
  80
  81     Assert.assertEquals(result.getJobs().size(), jobList.size());
  82     Assert.assertEquals(result.getJobs().get(0).getUuid(), uuid);
  83   }
  84
  85   @Test
  86   public void getJobTest() {
  87     UUID uuid = UUID.randomUUID();
  88     Job job = createJob(uuid);
  89     doReturn(job).when(jobsBrokerService).peek(uuid);
  90     JobModel jobModel = createJobModel(uuid);
  91     when(jobAdapter.toModel(job)).thenReturn(jobModel);
  92
  93     JobModel response = testSubject.getJob(uuid);
  94     Assert.assertEquals(response.getUuid(), uuid);
  95   }
  96
  97   @Test(expectedExceptions = {NotFoundException.class})
  98   public void getJobTest_throwsExceptionOnEmptyUUID() {
  99     UUID uuid = null;
 100     Job job = createJob(uuid);
 101     doReturn(job).when(jobsBrokerService).peek(uuid);
 102     JobModel response = testSubject.getJob(uuid);
 103     Assert.fail();
 104   }
 105
 106   @Test(expectedExceptions = {NotFoundException.class})
 107   public void getJobTest_throwsExceptionCauseJobDoesNotExists() {
 108     UUID uuid = UUID.randomUUID();
 109     doReturn(null).when(jobsBrokerService).peek(uuid);
 110     JobModel response = testSubject.getJob(uuid);
 111     Assert.fail();
 112   }
 113
 114   private Job createJob(UUID uuid) {
 115     Job job = new JobDaoImpl();
 116     job.setUuid(uuid);
 117     return job;
 118   }
 119
 120   private JobModel createJobModel(UUID uuid) {
 121     JobModel jobModel = new JobModel();
 122     jobModel.setUuid(uuid);
 123     return jobModel;
 124   }
 125
 126   private JobBulk createJobBulk(List<Job> jobList) {
 127     List<JobModel> jobBulkList = new ArrayList<>();
 128     jobList.stream().forEach(job -> {
 129       JobModel jm = new JobModel();
 130       jm.setUuid(job.getUuid());
 131       jobBulkList.add(jm);
 132     });
 133     return new JobBulk(jobBulkList);
 134   }
 135 }