aai-core/src/main/java/org/openecomp/aai/introspection/generator/CreateExample.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.openecomp.aai
   4  * ================================================================================
   5  * Copyright (C) 2017 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.openecomp.aai.introspection.generator;
  22
  23 import java.util.ArrayList;
  24 import java.util.List;
  25 import java.util.Random;
  26
  27 import org.openecomp.aai.exceptions.AAIException;
  28 import org.openecomp.aai.introspection.Introspector;
  29 import org.openecomp.aai.introspection.IntrospectorWalker;
  30 import org.openecomp.aai.introspection.Loader;
  31 import org.openecomp.aai.introspection.PropertyPredicates;
  32 import org.openecomp.aai.introspection.Wanderer;
  33
  34 public class CreateExample implements Wanderer {
  35
  36         private Random rand = new Random();
  37         private final long range = 100000000L;
  38         private Loader loader = null;
  39         private Introspector result = null;
  40         private String objectName = null;
  41         private List<String> blacklist = null;
  42
  43         /**
  44          * Instantiates a new creates the example.
  45          *
  46          * @param loader the loader
  47          * @param objectName the object name
  48          */
  49         public CreateExample(Loader loader, String objectName) {
  50
  51                 this.loader = loader;
  52                 this.objectName = objectName;
  53                 this.blacklist = new ArrayList<>();
  54
  55         }
  56
  57         /**
  58          * Gets the example object.
  59          *
  60          * @return the example object
  61          * @throws AAIException
  62          */
  63         public Introspector getExampleObject() throws AAIException {
  64                 result = loader.introspectorFromName(objectName);
  65                 blacklist = new ArrayList<>();
  66                 blacklist.add("any");
  67                 blacklist.add("relationship-list");
  68                 if (!result.isContainer()) {
  69                         blacklist.add("resource-version");
  70                 }
  71                 IntrospectorWalker walker = new IntrospectorWalker(this, PropertyPredicates.includeInExamples());
  72
  73                 walker.preventCycles(true);
  74                 walker.setBlacklist(blacklist);
  75                 walker.walk(result);
  76                 //this.getExampleObject(result);
  77
  78                 return result;
  79         }
  80
  81         /**
  82          * Gets the value.
  83          *
  84          * @param property the property
  85          * @param type the type
  86          * @param suffix the suffix
  87          * @return the value
  88          */
  89         private Object getValue(String property, String type, String suffix) {
  90                 long randLong = (long)(rand.nextDouble()*range);
  91                 Integer randInt = rand.nextInt(100000);
  92                 Integer randShrt = rand.nextInt(20000);
  93                 short randShort = randShrt.shortValue();
  94
  95                 Object newObj = null;
  96                 if (type.contains("java.lang.String")) {
  97                         newObj = "example-" + property + "-val-" + randInt + suffix;
  98                 } else if ( type.toLowerCase().equals("long") ||type.contains("java.lang.Long")) {
  99                         newObj = randLong;
 100                 } else if(type.toLowerCase().equals("boolean") || type.contains("java.lang.Boolean")){
 101                         newObj = Boolean.TRUE;
 102                 } else if ( type.toLowerCase().equals("int") || type.contains("java.lang.Integer")){
 103                         newObj = randInt;
 104                 }  else if ( type.toLowerCase().equals("short") || type.contains("java.lang.Short")){
 105                         newObj = randShort;
 106                 }
 107
 108                 return newObj;
 109         }
 110
 111         /**
 112          * {@inheritDoc}
 113          */
 114         @Override
 115         public void processPrimitive(String propName, Introspector obj) {
 116                 String  propType = obj.getType(propName);
 117
 118                 Object val = this.getValue(propName, propType, "");
 119                 obj.setValue(propName, val);
 120         }
 121
 122         /**
 123          * {@inheritDoc}
 124          */
 125         @Override
 126         public void processPrimitiveList(String propName, Introspector obj) {
 127                 int listSize = 2;
 128                 String propType = "";
 129                 List<Object> list = new ArrayList<>();
 130                 for (int i = 0; i < listSize; i++) {
 131                         propType = obj.getGenericType(propName);
 132                         Object val = this.getValue(propName, propType, "-" + (i + 1));
 133                         list.add(val);
 134                 }
 135                 obj.setValue(propName, list);
 136         }
 137
 138         /**
 139          * {@inheritDoc}
 140          */
 141         @Override
 142         public void processComplexObj(Introspector obj) {
 143
 144         }
 145
 146         /**
 147          * {@inheritDoc}
 148          */
 149         @Override
 150         public void modifyComplexList(List<Introspector> list, List<Object> listReference, Introspector parent, Introspector child) {
 151                 // TODO Auto-generated method stub
 152
 153         }
 154
 155         /**
 156          * {@inheritDoc}
 157          */
 158         @Override
 159         public boolean createComplexObjIfNull() {
 160                 return true;
 161         }
 162
 163         /**
 164          * {@inheritDoc}
 165          */
 166         @Override
 167         public int createComplexListSize(Introspector parent, Introspector child) {
 168                 return 1;
 169         }
 170 }