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