Copy entry_defintion_type to TOSCA.meta

[sdc.git] / openecomp-be / lib / openecomp-sdc-translator-lib / openecomp-sdc-translator-core / src / test / java / org / openecomp / sdc / translator / services / heattotosca / impl / functiontranslation / FunctionTranslationGetFileImplTest.java

diff --git a/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/test/java/org/openecomp/sdc/translator/services/heattotosca/impl/functiontranslation/FunctionTranslationGetFileImplTest.java b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/test/java/org/openecomp/sdc/translator/services/heattotosca/impl/functiontranslation/FunctionTranslationGetFileImplTest.java
index b79671d..847a88c 100644 (file)
--- a/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/test/java/org/openecomp/sdc/translator/services/heattotosca/impl/functiontranslation/FunctionTranslationGetFileImplTest.java
+++ b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/test/java/org/openecomp/sdc/translator/services/heattotosca/impl/functiontranslation/FunctionTranslationGetFileImplTest.java
@@ -1,43 +1,39 @@
-/*-
- * ============LICENSE_START=======================================================
- * SDC
- * ================================================================================
- * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
- * ================================================================================
+/*
+ * Copyright © 2016-2018 European Support Limited
+ *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *      http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
- * ============LICENSE_END=========================================================
  */
 
 package org.openecomp.sdc.translator.services.heattotosca.impl.functiontranslation;
 
 import org.junit.Assert;
 import org.junit.Test;
+import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
 import org.openecomp.core.utilities.file.FileUtils;
 import org.openecomp.sdc.heat.datatypes.model.HeatOrchestrationTemplate;
-import org.openecomp.sdc.tosca.datatypes.model.NodeTemplate;
+import org.openecomp.sdc.tosca.csar.AsdPackageHelper;
+import org.openecomp.sdc.tosca.csar.ManifestUtils;
 import org.openecomp.sdc.tosca.services.ToscaConstants;
 import org.openecomp.sdc.tosca.services.ToscaFileOutputService;
 import org.openecomp.sdc.tosca.services.impl.ToscaFileOutputServiceCsarImpl;
-import org.openecomp.sdc.translator.services.heattotosca.FunctionTranslationFactory;
 import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
+import org.openecomp.sdc.translator.services.heattotosca.FunctionTranslationFactory;
 
 import java.util.HashMap;
 import java.util.List;
 
-/**
- * @author SHIRIA
- * @since December 18, 2016.
- */
+import static org.openecomp.sdc.translator.services.heattotosca.impl.functiontranslation.FunctionTranslator.getFunctionTranslateTo;
+
 public class FunctionTranslationGetFileImplTest {
   @Test
   public void testGetFileWithExtensionFunction() {
@@ -71,23 +67,23 @@ public class FunctionTranslationGetFileImplTest {
                                               HeatOrchestrationTemplate heatOrchestrationTemplate,
                                               NodeTemplate nodeTemplate,
                                               TranslationContext context) {
-    Assert.assertEquals(true, FunctionTranslationFactory.getInstance(functionName).isPresent());
+    Assert.assertTrue(FunctionTranslationFactory.getInstance(functionName).isPresent());
     if(FunctionTranslationFactory.getInstance(functionName).isPresent()) {
+      FunctionTranslator functionTranslator = new FunctionTranslator(getFunctionTranslateTo(null, null, heatFileName,
+              heatOrchestrationTemplate, context), null, function, nodeTemplate);
       Object result = FunctionTranslationFactory.getInstance(functionName).get()
-          .translateFunction(null, null, null, functionName, function, heatFileName,
-              heatOrchestrationTemplate,
-              nodeTemplate, context);
+          .translateFunction(functionTranslator);
       Assert.assertNotNull(((HashMap) result).get("get_artifact"));
       List artifactParameters = (List) ((HashMap) result).get("get_artifact");
       Assert.assertNotNull(artifactParameters);
-      Assert.assertEquals(artifactParameters.size(), 2);
-      Assert.assertEquals(artifactParameters.get(0), ToscaConstants.MODELABLE_ENTITY_NAME_SELF);
-      Assert.assertEquals(artifactParameters.get(1), ((String) function).split("\\.")[0]);
+      Assert.assertEquals(2, artifactParameters.size());
+      Assert.assertEquals(ToscaConstants.MODELABLE_ENTITY_NAME_SELF, artifactParameters.get(0));
+      Assert.assertEquals(((String) function).split("\\.")[0], artifactParameters.get(1));
 
       Assert.assertNotNull(nodeTemplate.getArtifacts());
       Assert.assertNotNull(
           nodeTemplate.getArtifacts().get(FileUtils.getFileWithoutExtention((String) function)));
-      ToscaFileOutputService toscaFileOutputService = new ToscaFileOutputServiceCsarImpl();
+      ToscaFileOutputService toscaFileOutputService = new ToscaFileOutputServiceCsarImpl(new AsdPackageHelper(new ManifestUtils()));
       Assert.assertEquals(
           nodeTemplate.getArtifacts().get(FileUtils.getFileWithoutExtention((String) function))
               .getFile(), "../" + toscaFileOutputService.getArtifactsFolderName() + "/" + function);