ms/neng/src/test/java/org/onap/ccsdk/apps/ms/neng/core/gen/NameGeneratorDependencyOnUnnamedTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : CCSDK.apps
   4  * ================================================================================
   5  * Copyright (C) 2018 AT&T Intellectual Property. 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
  21 package org.onap.ccsdk.apps.ms.neng.core.gen;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.mockito.Matchers.anyInt;
  25 import static org.mockito.Matchers.anyObject;
  26 import static org.mockito.Mockito.doAnswer;
  27 import static org.mockito.Mockito.mock;
  28 import static org.mockito.Mockito.when;
  29
  30 import java.util.ArrayList;
  31 import java.util.HashMap;
  32 import java.util.List;
  33 import java.util.Map;
  34 import org.junit.Before;
  35 import org.junit.Test;
  36 import org.junit.runner.RunWith;
  37 import org.mockito.Mock;
  38 import org.mockito.Mockito;
  39 import org.mockito.invocation.InvocationOnMock;
  40 import org.mockito.runners.MockitoJUnitRunner;
  41 import org.mockito.stubbing.Answer;
  42 import org.onap.ccsdk.apps.ms.neng.core.persistence.NamePersister;
  43 import org.onap.ccsdk.apps.ms.neng.core.policy.FilePolicyReader;
  44 import org.onap.ccsdk.apps.ms.neng.core.policy.PolicyFinder;
  45 import org.onap.ccsdk.apps.ms.neng.core.policy.PolicyParameters;
  46 import org.onap.ccsdk.apps.ms.neng.core.seq.SequenceGenerator;
  47 import org.onap.ccsdk.apps.ms.neng.core.validator.AaiNameValidator;
  48 import org.onap.ccsdk.apps.ms.neng.core.validator.DbNameValidator;
  49 import org.onap.ccsdk.apps.ms.neng.persistence.entity.GeneratedName;
  50
  51 @RunWith(MockitoJUnitRunner.class)
  52 public class NameGeneratorDependencyOnUnnamedTest {
  53     @Mock
  54     private PolicyParameters policyParams = mock(PolicyParameters.class);
  55     @Mock
  56     private PolicyFinder policyFinder = mock(PolicyFinder.class);
  57     @Mock
  58     private SequenceGenerator sequenceGenerator = mock(SequenceGenerator.class);
  59     @Mock
  60     private DbNameValidator dbValidator = mock(DbNameValidator.class);
  61     @Mock
  62     private AaiNameValidator aaiValidator = mock(AaiNameValidator.class);
  63     @Mock
  64     private NamePersister namePresister = mock(NamePersister.class);
  65     private Map<String, Map<String, String>> earlierNames = new HashMap<>();
  66     private Map<String, Map<String, ?>> policyCache = new HashMap<>();
  67
  68     /**
  69      * Setup policy params.
  70      */
  71     @Before
  72     public void setupPolicyParams() throws Exception {
  73         Mockito.lenient().when(policyParams.mapFunction("substr")).thenReturn("substring");
  74         Mockito.lenient().when(policyParams.mapFunction("to_lower_case")).thenReturn("toLowerCase");
  75         Mockito.lenient().when(policyParams.mapFunction("to_upper_case")).thenReturn("toUpperCase");
  76     }
  77
  78     protected Map<String, String> makeVmRequest(String policy) {
  79         Map<String, String> requestElement = new HashMap<>();
  80         requestElement.put("external-key", "923456");
  81         requestElement.put("policy-instance-name", policy);
  82         requestElement.put("naming-type", "VM");
  83         requestElement.put("resource-name", "vm-name");
  84         requestElement.put("complex", "abcdeasdf");
  85         return requestElement;
  86     }
  87
  88     @Test
  89     public void generate() throws Exception {
  90         String policyName = "SDNC_Policy.Config_MS_VNF_VM_NamingPolicy";
  91         Map<String, String> requestElement2 = makeVmRequest(policyName);
  92         List<Map<String, String>> allElements = new ArrayList<>();
  93         allElements.add(requestElement2);
  94
  95         Map<String, Object> policy = new FilePolicyReader("vnf_and_vm_policy.json").getPolicy();
  96         Mockito.lenient().when(policyFinder.findPolicy(policyName)).thenReturn(policy);
  97         Mockito.lenient().when(aaiValidator.validate(anyObject(), anyObject())).thenReturn(true);
  98         Mockito.lenient().when(dbValidator.validate(anyObject(), anyObject())).thenReturn(true);
  99         Mockito.lenient().when(sequenceGenerator.generate(anyObject(), anyObject(), anyObject(), anyObject(), anyInt())).thenReturn(1L);
 100
 101         final List<Object> savedNames = new ArrayList<>();
 102         doAnswer(new Answer<Void>() {
 103             @Override
 104             public Void answer(InvocationOnMock invocation) throws Throwable {
 105                 savedNames.add(invocation.getArguments()[0]);
 106                 return null;
 107             }
 108         }).when(namePresister).persist(anyObject());
 109
 110         NameGenerator gen2 = new NameGenerator(policyFinder, policyParams, sequenceGenerator, dbValidator, aaiValidator,
 111                         namePresister, requestElement2, allElements, earlierNames, policyCache, new ArrayList<>());
 112
 113         Map<String, String> resp2 = gen2.generate();
 114         assertEquals("vm-name", resp2.get("resource-name"));
 115         assertEquals("923456", resp2.get("external-key"));
 116         assertEquals("abcde001ve1mts001", resp2.get("resource-value"));
 117
 118         assertEquals(2, savedNames.size());
 119
 120         assertEquals("abcde001ve1", ((GeneratedName) savedNames.get(0)).getName());
 121         assertEquals(null, ((GeneratedName) savedNames.get(0)).getExternalId());
 122         assertEquals("VNF", ((GeneratedName) savedNames.get(0)).getElementType());
 123         assertEquals("abcde", ((GeneratedName) savedNames.get(0)).getPrefix());
 124         assertEquals(1, ((GeneratedName) savedNames.get(0)).getSequenceNumber().longValue());
 125         assertEquals("001", ((GeneratedName) savedNames.get(0)).getSequenceNumberEnc());
 126         assertEquals("ve1", ((GeneratedName) savedNames.get(0)).getSuffix());
 127
 128         assertEquals("abcde001ve1mts001", ((GeneratedName) savedNames.get(1)).getName());
 129         assertEquals("923456", ((GeneratedName) savedNames.get(1)).getExternalId());
 130         assertEquals("VM", ((GeneratedName) savedNames.get(1)).getElementType());
 131         assertEquals("abcde001ve1mts", ((GeneratedName) savedNames.get(1)).getPrefix());
 132         assertEquals(1, ((GeneratedName) savedNames.get(1)).getSequenceNumber().longValue());
 133         assertEquals("001", ((GeneratedName) savedNames.get(1)).getSequenceNumberEnc());
 134         assertEquals(null, ((GeneratedName) savedNames.get(1)).getSuffix());
 135     }
 136 }
 137