Merge pull request #11 from MobileTeleSystems/feature/base_template

Fix call base template
MobileTeleSystems · Oct 23, 2024 · d682f32 · d682f32
2 parents e863d83 + 48135e9
commit d682f32
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/src/ApiCodeGenerator.OpenApi/DefaultTemplateFactory.cs b/src/ApiCodeGenerator.OpenApi/DefaultTemplateFactory.cs
@@ -58,12 +58,8 @@ public ITemplate CreateTemplate(string language, string template, object model)
         {
             if (model is Fluid.TemplateContext templateContext)
             {
-                var currentTemplate = (string)templateContext.AmbientValues["__template"];
+                var currentTemplate = ((string)templateContext.AmbientValues["__template"]).TrimEnd('!');
                 template = template.Substring(0, template.Length - BASE_TMPL_SUFFIX.Length);
-                if (!template.EndsWith("!"))
-                {
-                    template += "!";
-                }
 
                 providers = providers.SkipWhile(p => currentTemplate != p.GetFullName(template, language)).Skip(1);
             }

diff --git a/test/ApiCodeGenerator.OpenApi.Tests/DefaultTemplateFactoryTests.cs b/test/ApiCodeGenerator.OpenApi.Tests/DefaultTemplateFactoryTests.cs
@@ -51,7 +51,7 @@ public TestTemplateProvider(string templateText)
 
         public string? GetFullName(string name, string language)
         {
-            return name == "Class" || name == "Class!"
+            return name == "Class"
                 ? $"{_providerKey}.{name}"
                 : null;
         }