feat: Implement ElementWrapper class to enhance IHasFolder functionality for IElement instances

Author: dandrejvv

Modules/Intent.Modules.UnitTesting/Templates/ElementWrapper.cs

@@ -0,0 +1,47 @@
+using Intent.Metadata.Models;
+using Intent.Modules.Common.Types.Api;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Intent.Modules.UnitTesting.Templates;
+
+/// <summary>
+/// Wrapper class to provide IHasFolder functionality for IElement instances.
+/// This mimics the behavior of the old strongly-typed model classes.
+/// </summary>
+internal class ElementWrapper : IHasFolder
+{
+    private readonly IElement _element;
+    private readonly FolderModel _folder;
+
+    public ElementWrapper(IElement element)
+    {
+        _element = element ?? throw new ArgumentNullException(nameof(element));
+        
+        // Initialize folder if parent is a folder
+        _folder = _element.ParentElement?.SpecializationTypeId == FolderModel.SpecializationTypeId 
+            ? new FolderModel(_element.ParentElement) 
+            : null;
+    }
+
+    public string Id => _element.Id;
+    
+    public string Name => _element.Name;
+    
+    public FolderModel Folder => _folder;
+    
+    public IEnumerable<string> GetParentFolderNames()
+    {
+        var folders = new List<string>();
+        var currentFolder = _folder;
+        
+        while (currentFolder != null)
+        {
+            folders.Insert(0, currentFolder.Name);
+            currentFolder = currentFolder.Folder;
+        }
+        
+        return folders;
+    }
+}

Modules/Intent.Modules.UnitTesting/Templates/TestHelpers.cs

@@ -28,25 +28,23 @@ public static string GetTestElementNormalizedPath<TModel>(this CSharpTemplateBas
 
     public static string GetOperationNormalizedPath<TModel>(this CSharpTemplateBase<TModel> template)
     {
-        var model = template.Model as IElementWrapper;
-        if (model?.InternalElement == null || model.InternalElement.SpecializationTypeId != SpecializationTypeIds.Service)
+        var model = template.Model as IElement;
+        if (model == null || model.SpecializationTypeId != SpecializationTypeIds.Service)
         {
             return template.GetFolderPath();
         }
 
-        var parentService = model as IHasFolder;
-        if (parentService == null)
-        {
-            return template.GetFolderPath();
-        }
+        // Wrap the element to get IHasFolder functionality
+        var wrapper = new ElementWrapper(model);
 
         // these have to be done as two seperate join calls, otherwise the incorrect overloaded Join method is called
-        if(parentService.GetParentFolderNames().Count > 0)
+        var parentFolders = wrapper.GetParentFolderNames().ToList();
+        if(parentFolders.Count > 0)
         {
-            return string.Join("/", parentService.GetParentFolderNames());
+            return string.Join("/", parentFolders);
         }
 
-        return string.Join("/", model.InternalElement.Name.Replace("Service", string.Empty));
+        return string.Join("/", model.Name.Replace("Service", string.Empty));
     }
 
     public static void PopulateTestConstructor(ICSharpTemplate template, CSharpConstructor ctor, ITemplate handlerTemplate,