corrected code to use new get template function

Signed-off-by: Holly Hunt <holly.hunt1101@hotmail.com>
galasa-dev · Apr 9, 2024 · d8e5c1b · d8e5c1b
1 parent 85eb140
commit d8e5c1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/openapi2beans/pkg/generator/package2java.go b/openapi2beans/pkg/generator/package2java.go
@@ -15,20 +15,19 @@ import (
 
 func convertJavaPackageToJavaFiles(javaPackage *JavaPackage, fs files.FileSystem, storeFilePath string) {
 	log.Print("convertJavaPackageToJavaFiles: Creating classes")
-	classTemplate, err := embedded.GetJavaClassTemplate()
+	classTemplate, err := embedded.GetJavaTemplate(embedded.GET_JAVA_TEMPLATE_CLASS_OPTION)
 	if err == nil {
 		for _, javaClass := range javaPackage.Classes {
 			createJavaClassFile(javaClass, fs, classTemplate, storeFilePath)
 		}
 		log.Print("convertJavaPackageToJavaFiles: Creating enums")
-		enumTemplate, err := embedded.GetJavaEnumTemplate()
+		enumTemplate, err := embedded.GetJavaTemplate(embedded.GET_JAVA_TEMPLATE_ENUM_OPTION)
 		if err == nil {
 			for _, javaEnum := range javaPackage.Enums {
 				createJavaEnumFile(javaEnum, fs, enumTemplate, storeFilePath)
 			}
 		}
 	}
-
 }
 
 // TODO: Make errors output from these functions fatal

diff --git a/openapi2beans/pkg/generator/package2java_test.go b/openapi2beans/pkg/generator/package2java_test.go
@@ -19,13 +19,13 @@ const (
 )
 
 func getEmbeddedClassTemplate(t *testing.T) *mustache.Template {
-	classTemplate, err := embedded.GetJavaClassTemplate()
+	classTemplate, err := embedded.GetJavaTemplate(embedded.GET_JAVA_TEMPLATE_CLASS_OPTION)
 	assert.Nil(t, err)
 	return classTemplate
 }
 
 func getEmbeddedEnumTemplate(t *testing.T) *mustache.Template {
-	enumTemplate, err := embedded.GetJavaEnumTemplate()
+	enumTemplate, err := embedded.GetJavaTemplate(embedded.GET_JAVA_TEMPLATE_ENUM_OPTION)
 	assert.Nil(t, err)
 	return enumTemplate
 }