Merge pull request #50 from TestCentric/issue-49

Fluent syntax for Intializing ExtensionManager
TestCentric · Oct 17, 2024 · 6930a15 · 6930a15
2 parents b382e37 + 2f4e864
commit 6930a15
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/src/testcentric.extensibility.api/IExtensionManager.cs b/src/testcentric.extensibility.api/IExtensionManager.cs
@@ -55,7 +55,8 @@ public interface IExtensionManager
         /// Find all ExtensionPoints in a list of assemblies and add them to the ExtensionPoints property.
         /// </summary>
         /// <param name="assemblies">The assemblies to be examined for ExtensionPoints</param>
-        void FindExtensionPoints(params Assembly[] assemblies);
+        /// <returns>The current instance of IExtensionManager</returns>
+        IExtensionManager FindExtensionPoints(params Assembly[] assemblies);
 
         /// <summary>
         /// Gets an IExtensionPoint given its path
@@ -79,7 +80,8 @@ public interface IExtensionManager
         /// <param name="startDir">
         /// Path to the directory containing the initial .addins files used to locate extensions
         /// </param>
-        void FindExtensions(string startDir);
+        /// <returns>The current instance of IExtensionManager</returns>
+        IExtensionManager FindExtensions(string startDir);
 
         /// <summary>
         /// Enumerates all extension objects matching a given type

diff --git a/src/testcentric.extensibility/ExtensionManager.cs b/src/testcentric.extensibility/ExtensionManager.cs
@@ -107,7 +107,7 @@ public IEnumerable<IExtensionPoint> ExtensionPoints
         }
 
         /// <inheritdoc/>
-        public void FindExtensionPoints(params Assembly[] assemblies)
+        public IExtensionManager FindExtensionPoints(params Assembly[] assemblies)
         {
             foreach (var assembly in assemblies)
             {
@@ -138,6 +138,8 @@ public void FindExtensionPoints(params Assembly[] assemblies)
                 }
             }
 
+            return this;
+
             void AddExtensionPoint(string path, Type type, AssemblyName assemblyName, string description = null)
             {
                 if (_pathIndex.ContainsKey(path))
@@ -178,7 +180,7 @@ public IEnumerable<IExtensionNode> Extensions
         }
 
         /// <inheritdoc/>
-        public void FindExtensions(string startDir)
+        public IExtensionManager FindExtensions(string startDir)
         {
             // Find potential extension assemblies
             log.Info("Initializing Extensions...");
@@ -188,6 +190,8 @@ public void FindExtensions(string startDir)
             // Check each assembly to see if it contains extensions
             foreach (var candidate in _extensionAssemblies)
                 FindExtensionsInAssembly(candidate);
+
+            return this;
         }
 
         /// <inheritdoc/>