b732cb03b6a012af557c8a323f7e76612946530b
[sdc.git] /
1 package org.openecomp.sdc.healing.healers;
2
3 import org.junit.Before;
4 import org.junit.Test;
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;
12
13 import java.util.HashMap;
14 import java.util.Map;
15
16 import static org.junit.Assert.assertEquals;
17 import static org.mockito.Matchers.any;
18
19 public class VspOnboardingMethodHealerTest{
20
21     @Mock
22     private VendorSoftwareProductInfoDao vendorSoftwareProductInfoDao;
23
24     @Before
25     public void setUp() throws Exception {
26         MockitoAnnotations.initMocks(VspOnboardingMethodHealerTest.this);
27     }
28
29     @Test
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");
41     }
42
43     @Test
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");
55     }
56
57     @Test
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);
69     }
70 }