Added registration of bindings from external assemblies (#69)

Contributor: @VakarisL
solidtoken · Sep 3, 2021 · 9891f2d · 9891f2d
1 parent 8677491
commit 9891f2d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/SpecFlow.DependencyInjection/ServiceCollectionFinder.cs b/SpecFlow.DependencyInjection/ServiceCollectionFinder.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using System.Reflection;
 using Microsoft.Extensions.DependencyInjection;
@@ -46,7 +47,7 @@ protected virtual Func<IServiceCollection> FindCreateScenarioServiceCollection()
                                 var serviceCollection = GetServiceCollection(methodInfo);
                                 if (scenarioDependenciesAttribute.AutoRegisterBindings)
                                 {
-                                    AddBindingAttributes(assembly, serviceCollection);
+                                    AddBindingAttributes(assemblies, serviceCollection);
                                 }
                                 return serviceCollection;
                             };
@@ -62,11 +63,14 @@ private static IServiceCollection GetServiceCollection(MethodBase methodInfo)
             return (IServiceCollection)methodInfo.Invoke(null, null);
         }
 
-        private static void AddBindingAttributes(Assembly assembly, IServiceCollection serviceCollection)
+        private static void AddBindingAttributes(IEnumerable<Assembly> bindingAssemblies, IServiceCollection serviceCollection)
         {
-            foreach (var type in assembly.GetTypes().Where(t => Attribute.IsDefined(t, typeof(BindingAttribute))))
+            foreach(var assembly in bindingAssemblies)
             {
-                serviceCollection.AddSingleton(type);
+                foreach (var type in assembly.GetTypes().Where(t => Attribute.IsDefined(t, typeof(BindingAttribute))))
+                {
+                    serviceCollection.AddSingleton(type);
+                }
             }
         }
     }