Make MockHandler method protected (#46)

Author: twsouthwick

AutofacContrib.NSubstitute/AutoSubstituteOptionsExtensions.cs

@@ -54,7 +54,7 @@ private NonPublicConstructorFinder()
 
         private class AutoPropertyInjectorMockHandler : MockHandler
         {
-            public override void OnMockCreated(object instance, Type type, IComponentContext context, ISubstitutionContext substitutionContext)
+            protected internal override void OnMockCreated(object instance, Type type, IComponentContext context, ISubstitutionContext substitutionContext)
             {
                 var router = substitutionContext.GetCallRouterFor(instance);
 

AutofacContrib.NSubstitute/MockHandler.cs

@@ -20,6 +20,6 @@ protected MockHandler()
         /// <param name="type">The type the mock was created for.</param>
         /// <param name="context">The current component context.</param>
         /// <param name="substitutionContext">The current substitution context.</param>
-        public abstract void OnMockCreated(object instance, Type type, IComponentContext context, ISubstitutionContext substitutionContext);
+        protected internal abstract void OnMockCreated(object instance, Type type, IComponentContext context, ISubstitutionContext substitutionContext);
     }
 }

AutofacContrib.NSubstitute/MockHandler{T}.cs

@@ -0,0 +1,35 @@
+using Autofac;
+using NSubstitute.Core;
+using System;
+
+namespace AutofacContrib.NSubstitute
+{
+    /// <summary>
+    /// A handler that is used to modify any of the auto-generated NSubstitute mocks when the type is <typeparamref name="T"/>.
+    /// </summary>
+    public abstract class MockHandler<T> : MockHandler
+        where T : class
+    {
+        /// <inheritdoc />
+        protected internal sealed override void OnMockCreated(object instance, Type type, IComponentContext context, ISubstitutionContext substitutionContext)
+        {
+            if (typeof(T) == type && instance is T t)
+            {
+                OnMockCreated(t, context, substitutionContext);
+            }
+        }
+
+        /// <summary>
+        /// Gets the type associated with this handler.
+        /// </summary>
+        protected Type Type => typeof(T);
+
+        /// <summary>
+        /// Provides a way to manage mocks after creation but before returned from the container registry.
+        /// </summary>
+        /// <param name="instance">The mock instance.</param>
+        /// <param name="context">The current component context.</param>
+        /// <param name="substitutionContext">The current substitution context.</param>
+        protected abstract void OnMockCreated(T instance, IComponentContext context, ISubstitutionContext substitutionContext);
+    }
+}