openecomp-be/lib/openecomp-sdc-vendor-software-product-lib/openecomp-sdc-vendor-software-product-core/src/test/java/org/openecomp/sdc/vendorsoftwareproduct/dao/ServiceModelDaoFactoryTest.java

   1 package org.openecomp.sdc.vendorsoftwareproduct.dao;
   2
   3 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
   4 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
   5
   6 import org.openecomp.sdc.versioning.dao.types.Version;
   7 import org.openecomp.core.model.dao.ServiceModelDaoFactory;
   8 import org.openecomp.core.model.types.ServiceArtifact;
   9 import org.openecomp.core.utilities.CommonMethods;
  10 import org.openecomp.core.utilities.file.FileContentHandler;
  11 import org.testng.Assert;
  12 import org.testng.annotations.Test;
  13
  14 import java.util.HashMap;
  15 import java.util.Map;
  16
  17 public class ServiceModelDaoFactoryTest {
  18
  19   private static final String vspId = CommonMethods.nextUuId();
  20   private static final Version version = Version.valueOf("1.0");
  21   private static final String baseServiceTemplateName = "baseYaml.yaml";
  22   private static String artifact001;
  23
  24
  25 //  @Test
  26   public void storeServiceModelTest() {
  27
  28
  29     ToscaServiceModel model = getToscaServiceModel();
  30     ServiceModelDaoFactory.getInstance().createInterface().storeServiceModel(vspId, version, model);
  31   }
  32
  33
  34 //  @Test(dependsOnMethods = "storeServiceModelTest")
  35   public void getServiceModelTest() {
  36     Object model =
  37         ServiceModelDaoFactory.getInstance().createInterface().getServiceModel(vspId, version);
  38     Assert.assertNotNull(model);
  39     Assert.assertTrue(model instanceof ToscaServiceModel);
  40     if (model instanceof ToscaServiceModel) {
  41
  42       artifact001 =
  43           (String) ((ToscaServiceModel) model).getArtifactFiles().getFileList().toArray()[0];
  44     }
  45   }
  46
  47 //  @Test(dependsOnMethods = "getServiceModelTest")
  48   public void getServiceModelInfoTest() {
  49     Object info = ServiceModelDaoFactory.getInstance().createInterface()
  50         .getServiceModelInfo(vspId, version, artifact001);
  51     Assert.assertNotNull(info);
  52     Assert.assertTrue(info instanceof ServiceArtifact);
  53     if (info instanceof ServiceArtifact) {
  54       Assert.assertEquals(((ServiceArtifact) info).getName(), artifact001);
  55     }
  56   }
  57
  58   private ToscaServiceModel getToscaServiceModel() {
  59
  60     Map<String, ServiceTemplate> serviceTemplates = getServiceTemplates(baseServiceTemplateName);
  61     FileContentHandler artifacts = getArtifacts();
  62     return new ToscaServiceModel(artifacts, serviceTemplates, baseServiceTemplateName);
  63   }
  64
  65
  66   private Map<String, ServiceTemplate> getServiceTemplates(String base) {
  67
  68     Map<String, ServiceTemplate> serviceTemplates = new HashMap<>();
  69
  70     serviceTemplates.put(base, getServiceTemplate());
  71     serviceTemplates.put("SERV1", getServiceTemplate());
  72     serviceTemplates.put("SERV2", getServiceTemplate());
  73     serviceTemplates.put("SERV3", getServiceTemplate());
  74     serviceTemplates.put("SERV4", getServiceTemplate());
  75
  76     return serviceTemplates;
  77   }
  78
  79   public FileContentHandler getArtifacts() {
  80     Map<String, byte[]> artifacts = new HashMap<>();
  81     artifacts.put("art1", "this is art1".getBytes());
  82     artifacts.put("art2", ("this is art2 desc:" + CommonMethods.nextUuId()).getBytes());
  83     artifacts.put("art2", ("this is art3 desc:" + CommonMethods.nextUuId()).getBytes());
  84     artifacts.put("art2", ("this is art4 desc:" + CommonMethods.nextUuId()).getBytes());
  85
  86     FileContentHandler fileContentHandler = new FileContentHandler();
  87     fileContentHandler.putAll(artifacts);
  88     return fileContentHandler;
  89   }
  90
  91   public ServiceTemplate getServiceTemplate() {
  92     ServiceTemplate serviceTemplate = new ServiceTemplate();
  93     serviceTemplate.setTosca_definitions_version("version 1.0");
  94     serviceTemplate.setDescription(CommonMethods.nextUuId());
  95     return serviceTemplate;
  96   }
  97 }