From b6fef600b2fa2a0c79c7f78210922f16670b4355 Mon Sep 17 00:00:00 2001
From: Magda Zaremba <magdalena.zaremba@codilime.com>
Date: Fri, 20 Apr 2018 15:31:56 +0200
Subject: [PATCH] Customize service types

Restrict to only firewall and analyzer

Change-Id: Ia5de41c856a1ab5aad9afb5a3aaa7cfa561e35ef
Closes-Bug: #1765715
---
 .../juniper/contrail/vro/workflows/custom/ServiceTemplate.kt   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/o11nplugin-contrail-workflows/src/main/kotlin/net/juniper/contrail/vro/workflows/custom/ServiceTemplate.kt b/o11nplugin-contrail-workflows/src/main/kotlin/net/juniper/contrail/vro/workflows/custom/ServiceTemplate.kt
index d9073a27..2197e2bd 100644
--- a/o11nplugin-contrail-workflows/src/main/kotlin/net/juniper/contrail/vro/workflows/custom/ServiceTemplate.kt
+++ b/o11nplugin-contrail-workflows/src/main/kotlin/net/juniper/contrail/vro/workflows/custom/ServiceTemplate.kt
@@ -24,6 +24,7 @@ import net.juniper.contrail.vro.workflows.util.relationDescription
 
 val supportedVersion : Long = 2
 val supportedVersions = listOf(supportedVersion)
+val serviceTypes = listOf("firewall", "analyzer")
 
 internal fun createServiceTemplate(schema: Schema) : WorkflowDefinition {
     val workflowName = "Create service template"
@@ -53,7 +54,7 @@ internal fun createServiceTemplate(schema: Schema) : WorkflowDefinition {
         parameter("serviceType", string) {
             description = propertyDescription<ServiceTemplateType>(schema)
             mandatory = true
-            extractPredefinedAnswers<ServiceTemplateType>(schema)
+            predefinedAnswers = serviceTypes
         }
         parameter("serviceVirtualizationType", string) {
             description = propertyDescription<ServiceTemplateType>(schema)