1 package org.openecomp.sdc.healing.healers;
3 import org.junit.Before;
5 import org.mockito.Mock;
6 import org.mockito.Mockito;
7 import org.mockito.MockitoAnnotations;
8 import org.openecomp.sdc.common.utils.SdcCommon;
9 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductInfoDao;
10 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspDetails;
11 import org.openecomp.sdc.versioning.dao.types.Version;
13 import java.util.HashMap;
16 import static org.junit.Assert.assertEquals;
17 import static org.mockito.Matchers.any;
19 public class VspOnboardingMethodHealerTest{
22 private VendorSoftwareProductInfoDao vendorSoftwareProductInfoDao;
25 public void setUp() throws Exception {
26 MockitoAnnotations.initMocks(VspOnboardingMethodHealerTest.this);
30 public void checkHealingWithNullOnboarding() throws Exception{
31 VspOnboardingMethodHealer vspOnboardingMethodHealer = new VspOnboardingMethodHealer(vendorSoftwareProductInfoDao);
32 Map<String,Object> params = new HashMap<>();
33 params.put(SdcCommon.VSP_ID,"1");
34 params.put(SdcCommon.VERSION, new Version(1,1));
35 VspDetails vspDetails = new VspDetails();
36 vspDetails.setOnboardingMethod(null);
37 Mockito.when(vendorSoftwareProductInfoDao.get(any())).thenReturn(vspDetails);
38 vspOnboardingMethodHealer.heal(params);
39 assertEquals(vspDetails.getOnboardingMethod(),"NetworkPackage");
40 assertEquals(vspDetails.getOnboardingOrigin(),"zip");
44 public void checkHealingWithHEATOnboarding() throws Exception{
45 VspOnboardingMethodHealer vspOnboardingMethodHealer = new VspOnboardingMethodHealer(vendorSoftwareProductInfoDao);
46 Map<String,Object> params = new HashMap<>();
47 params.put(SdcCommon.VSP_ID,"1");
48 params.put(SdcCommon.VERSION, new Version(1,1));
49 VspDetails vspDetails = new VspDetails();
50 vspDetails.setOnboardingMethod("HEAT");
51 Mockito.when(vendorSoftwareProductInfoDao.get(any())).thenReturn(vspDetails);
52 vspOnboardingMethodHealer.heal(params);
53 assertEquals(vspDetails.getOnboardingMethod(),"NetworkPackage");
54 assertEquals(vspDetails.getOnboardingOrigin(),"zip");
58 public void checkHealingWithManualOnboarding() throws Exception{
59 VspOnboardingMethodHealer vspOnboardingMethodHealer = new VspOnboardingMethodHealer(vendorSoftwareProductInfoDao);
60 Map<String,Object> params = new HashMap<>();
61 params.put(SdcCommon.VSP_ID,"1");
62 params.put(SdcCommon.VERSION, new Version(1,1));
63 VspDetails vspDetails = new VspDetails();
64 vspDetails.setOnboardingMethod("Manual");
65 Mockito.when(vendorSoftwareProductInfoDao.get(any())).thenReturn(vspDetails);
66 vspOnboardingMethodHealer.heal(params);
67 assertEquals(vspDetails.getOnboardingMethod(),"Manual");
68 assertEquals(vspDetails.getOnboardingOrigin(),null);