more testing stuffs

RocketSurgeonsGuild · Sep 9, 2018 · d1bcef4 · d1bcef4
1 parent 02a8c9b
commit d1bcef4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/Azure.Functions/RocketSurgeryWebJobsBuilderExtensions.cs b/src/Azure.Functions/RocketSurgeryWebJobsBuilderExtensions.cs
@@ -24,7 +24,7 @@ namespace Rocket.Surgery.Azure.Functions
 {
     public static class RocketSurgeryWebJobsBuilderExtensions
     {
-        internal static IContainer BuildContainer(ILogger logger, ServiceCollection services, Assembly assembly, object startupInstance)
+        internal static IContainer BuildContainer(ILogger logger, ServiceCollection services, Assembly assembly, object startupInstance, IAssemblyCandidateFinder assemblyCandidateFinder, IAssemblyProvider assemblyProvider)
         {
             var environmentNames = new[]
             {
@@ -48,8 +48,8 @@ internal static IContainer BuildContainer(ILogger logger, ServiceCollection serv
             };
 
             var context = DependencyContext.Load(assembly);
-            var assemblyCandidateFinder = new DependencyContextAssemblyCandidateFinder(context, logger);
-            var assemblyProvider = new DependencyContextAssemblyProvider(context, logger);
+            assemblyCandidateFinder = assemblyCandidateFinder ?? new DependencyContextAssemblyCandidateFinder(context, logger);
+            assemblyProvider = assemblyProvider ?? new DependencyContextAssemblyProvider(context, logger);
             var scanner = new AggregateConventionScanner(assemblyCandidateFinder);
 
             if (startupInstance is IConvention convention)
@@ -91,7 +91,7 @@ internal static IContainer BuildContainer(ILogger logger, ServiceCollection serv
             return diBuilder.Build();
         }
 
-        public static IWebJobsBuilder AddRocketSurgery(this IWebJobsBuilder builder, Assembly  assembly, object startupInstance)
+        public static IWebJobsBuilder AddRocketSurgery(this IWebJobsBuilder builder, Assembly assembly, object startupInstance, IAssemblyCandidateFinder assemblyCandidateFinder = null, IAssemblyProvider assemblyProvider = null)
         {
             //builder.AddExtension<ServiceConfiguration>();
             var logger = new ServiceCollection()
@@ -108,7 +108,7 @@ public static IWebJobsBuilder AddRocketSurgery(this IWebJobsBuilder builder, Ass
                 services.Add(s);
             }
 
-            var container = BuildContainer(logger, services, assembly, startupInstance);
+            var container = BuildContainer(logger, services, assembly, startupInstance, assemblyCandidateFinder, assemblyProvider);
 
             var injectBindingProvider = new ServiceBindingProvider(container, logger);
             builder.Services.TryAddEnumerable(ServiceDescriptor.Singleton<IBindingProvider>(injectBindingProvider));

diff --git a/src/Azure.Functions/RocketSurgeryWebJobsStartup.cs b/src/Azure.Functions/RocketSurgeryWebJobsStartup.cs
@@ -19,7 +19,7 @@ public abstract class RocketSurgeryWebJobsStartup : IWebJobsStartup
     {
         private readonly Assembly _assembly;
 
-        public RocketSurgeryWebJobsStartup(Assembly  assembly)
+        public RocketSurgeryWebJobsStartup(Assembly assembly)
         {
             _assembly = assembly;
         }