lots of different things ;)

NzbDrone.Test.Common/AutoMoq/Unity/AutoMockingBuilderStrategy.cs

@@ -0,0 +1,84 @@
+// ReSharper disable RedundantUsingDirective
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using Microsoft.Practices.ObjectBuilder2;
+using Microsoft.Practices.Unity;
+using Moq;
+
+namespace AutoMoq.Unity
+{
+    internal class AutoMockingBuilderStrategy : BuilderStrategy
+    {
+        private readonly IUnityContainer _container;
+        private readonly MockRepository _mockFactory;
+        private readonly IEnumerable<Type> _registeredTypes;
+
+        public AutoMockingBuilderStrategy(IEnumerable<Type> registeredTypes, IUnityContainer container)
+        {
+            var autoMoqer = container.Resolve<AutoMoqer>();
+            _mockFactory = new MockRepository(autoMoqer.DefaultBehavior);
+            _registeredTypes = registeredTypes;
+            _container = container;
+        }
+
+        public override void PreBuildUp(IBuilderContext context)
+        {
+            var autoMoqer = _container.Resolve<AutoMoqer>();
+
+            var type = GetTheTypeFromTheBuilderContext(context);
+            if (AMockObjectShouldBeCreatedForThisType(type))
+            {
+                var mock = CreateAMockObject(type);
+                context.Existing = mock.Object;
+                autoMoqer.SetMock(type, mock);
+            }
+        }
+
+        #region private methods
+
+        private bool AMockObjectShouldBeCreatedForThisType(Type type)
+        {
+            var mocker = _container.Resolve<AutoMoqer>();
+            return TypeIsNotRegistered(type) && (mocker.ResolveType == null || mocker.ResolveType != type);
+            //return TypeIsNotRegistered(type) && type.IsInterface;
+        }
+
+        private static Type GetTheTypeFromTheBuilderContext(IBuilderContext context)
+        {
+            return (context.OriginalBuildKey).Type;
+        }
+
+        private bool TypeIsNotRegistered(Type type)
+        {
+            return _registeredTypes.Any(x => x.Equals(type)) == false;
+        }
+
+        private Mock CreateAMockObject(Type type)
+        {
+            var createMethod = GenerateAnInterfaceMockCreationMethod(type);
+
+            return InvokeTheMockCreationMethod(createMethod);
+        }
+
+        private Mock InvokeTheMockCreationMethod(MethodInfo createMethod)
+        {
+            return (Mock)createMethod.Invoke(_mockFactory, new object[] { new List<object>().ToArray() });
+        }
+
+        private MethodInfo GenerateAnInterfaceMockCreationMethod(Type type)
+        {
+            var createMethodWithNoParameters = _mockFactory.GetType().GetMethod("Create", EmptyArgumentList());
+
+            return createMethodWithNoParameters.MakeGenericMethod(new[] { type });
+        }
+
+        private static Type[] EmptyArgumentList()
+        {
+            return new[] { typeof(object[]) };
+        }
+
+        #endregion
+    }
+}

NzbDrone.Test.Common/AutoMoq/Unity/AutoMockingContainerExtension.cs

@@ -0,0 +1,40 @@
+// ReSharper disable RedundantUsingDirective
+using System;
+using System.Collections.Generic;
+using Microsoft.Practices.Unity;
+using Microsoft.Practices.Unity.ObjectBuilder;
+
+namespace AutoMoq.Unity
+{
+    internal class AutoMockingContainerExtension : UnityContainerExtension
+    {
+        private readonly IList<Type> registeredTypes = new List<Type>();
+
+        protected override void Initialize()
+        {
+            SetEventsOnContainerToTrackAllRegisteredTypes();
+            SetBuildingStrategyForBuildingUnregisteredTypes();
+        }
+
+        #region private methods
+
+        private void SetEventsOnContainerToTrackAllRegisteredTypes()
+        {
+            Context.Registering += ((sender, e) => RegisterType(e.TypeFrom));
+            Context.RegisteringInstance += ((sender, e) => RegisterType(e.RegisteredType));
+        }
+
+        private void RegisterType(Type typeToRegister)
+        {
+            registeredTypes.Add(typeToRegister);
+        }
+
+        private void SetBuildingStrategyForBuildingUnregisteredTypes()
+        {
+            var strategy = new AutoMockingBuilderStrategy(registeredTypes, Container);
+            Context.Strategies.Add(strategy, UnityBuildStage.PreCreation);
+        }
+
+        #endregion
+    }
+}