Use DryIoc for Automoqer, drop Unity dependency

(cherry picked from commit e3468daba04b52fbf41ce3004934a26b0220ec4f)

src/NzbDrone.Test.Common/AutoMoq/AutoMoqer.cs

@@ -3,17 +3,11 @@ using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO;
 using System.Linq;
-using System.Linq.Expressions;
 using System.Reflection;
-using System.Runtime.CompilerServices;
+using DryIoc;
 using Moq;
-using Moq.Language.Flow;
 using NzbDrone.Common.Composition;
 using NzbDrone.Common.EnvironmentInfo;
-using NzbDrone.Test.Common.AutoMoq.Unity;
-using Unity;
-
-[assembly: InternalsVisibleTo("AutoMoq.Tests")]
 
 namespace NzbDrone.Test.Common.AutoMoq
 {
@@ -21,32 +15,18 @@ namespace NzbDrone.Test.Common.AutoMoq
     public class AutoMoqer
     {
         public readonly MockBehavior DefaultBehavior = MockBehavior.Default;
-        public Type ResolveType;
+        private IContainer _container;
-        private IUnityContainer _container;
         private IDictionary<Type, object> _registeredMocks;
 
         public AutoMoqer()
         {
-            SetupAutoMoqer(new UnityContainer());
+            SetupAutoMoqer(CreateTestContainer(new Container()));
-        }
-
-        public AutoMoqer(MockBehavior defaultBehavior)
-        {
-            DefaultBehavior = defaultBehavior;
-            SetupAutoMoqer(new UnityContainer());
-        }
-
-        public AutoMoqer(IUnityContainer container)
-        {
-            SetupAutoMoqer(container);
         }
 
         public virtual T Resolve<T>()
         {
-            ResolveType = typeof(T);
             var result = _container.Resolve<T>();
             SetConstant(result);
-            ResolveType = null;
             return result;
         }
 
@@ -59,7 +39,6 @@ namespace NzbDrone.Test.Common.AutoMoq
         public virtual Mock<T> GetMock<T>(MockBehavior behavior)
             where T : class
         {
-            ResolveType = null;
             var type = GetTheMockType<T>();
             if (GetMockHasNotBeenCalledForThisType(type))
             {
@@ -78,87 +57,81 @@ namespace NzbDrone.Test.Common.AutoMoq
 
         public virtual void SetMock(Type type, Mock mock)
         {
-            if (_registeredMocks.ContainsKey(type) == false)
+            if (GetMockHasNotBeenCalledForThisType(type))
             {
                 _registeredMocks.Add(type, mock);
             }
 
             if (mock != null)
             {
-                _container.RegisterInstance(type, mock.Object);
+                _container.RegisterInstance(type, mock.Object, ifAlreadyRegistered: IfAlreadyRegistered.Replace);
             }
         }
 
         public virtual void SetConstant<T>(T instance)
         {
-            _container.RegisterInstance(instance);
+            _container.RegisterInstance(instance, ifAlreadyRegistered: IfAlreadyRegistered.Replace);
             SetMock(instance.GetType(), null);
         }
 
-        public ISetup<T> Setup<T>(Expression<Action<T>> expression)
+        private IContainer CreateTestContainer(IContainer container)
-            where T : class
         {
-            return GetMock<T>().Setup(expression);
+            var c = container.CreateChild(IfAlreadyRegistered.Replace,
-        }
+                container.Rules
+                    .WithDynamicRegistration((serviceType, serviceKey) =>
+                    {
+                        // ignore services with non-default key
+                        if (serviceKey != null)
+                        {
+                            return null;
+                        }
 
-        public ISetup<T, TResult> Setup<T, TResult>(Expression<Func<T, TResult>> expression)
+                        if (serviceType == typeof(object))
-            where T : class
+                        {
-        {
+                            return null;
-            return GetMock<T>().Setup(expression);
+                        }
-        }
 
-        public void Verify<T>(Expression<Action<T>> expression)
+                        if (serviceType.IsGenericType && serviceType.IsOpenGeneric())
-            where T : class
+                        {
-        {
+                            return null;
-            GetMock<T>().Verify(expression);
+                        }
-        }
 
-        public void Verify<T>(Expression<Action<T>> expression, string failMessage)
+                        // get the Mock object for the abstract class or interface
-            where T : class
+                        if (serviceType.IsInterface || serviceType.IsAbstract)
-        {
+                        {
-            GetMock<T>().Verify(expression, failMessage);
+                            var mockType = typeof(Mock<>).MakeGenericType(serviceType);
-        }
+                            var mockFactory = new DelegateFactory(r =>
+                            {
+                                var mock = (Mock)r.Resolve(mockType);
+                                SetMock(serviceType, mock);
+                                return mock.Object;
+                            }, Reuse.Singleton);
 
-        public void Verify<T>(Expression<Action<T>> expression, Times times)
+                            return new[] { new DynamicRegistration(mockFactory, IfAlreadyRegistered.Keep) };
-            where T : class
+                        }
-        {
-            GetMock<T>().Verify(expression, times);
-        }
 
-        public void Verify<T>(Expression<Action<T>> expression, Times times, string failMessage)
+                        // concrete types
-            where T : class
+                        var concreteTypeFactory = serviceType.ToFactory(Reuse.Singleton, FactoryMethod.ConstructorWithResolvableArgumentsIncludingNonPublic);
-        {
-            GetMock<T>().Verify(expression, times, failMessage);
-        }
 
-        public void VerifyAllMocks()
+                        return new[] { new DynamicRegistration(concreteTypeFactory) };
-        {
+                    },
-            foreach (var registeredMock in _registeredMocks)
+                    DynamicRegistrationFlags.Service | DynamicRegistrationFlags.AsFallback));
-            {
+
-                var mock = registeredMock.Value as Mock;
+            c.Register(typeof(Mock<>), Reuse.Singleton, FactoryMethod.DefaultConstructor());
-                if (mock != null)
+
-                {
+            return c;
-                    mock.VerifyAll();
-                }
-            }
         }
 
-        private void SetupAutoMoqer(IUnityContainer container)
+        private void SetupAutoMoqer(IContainer container)
         {
             _container = container;
             container.RegisterInstance(this);
 
-            RegisterPlatformLibrary(container);
-
             _registeredMocks = new Dictionary<Type, object>();
-            AddTheAutoMockingContainerExtensionToTheContainer(container);
-            AssemblyLoader.RegisterSQLiteResolver();
-        }
 
-        private static void AddTheAutoMockingContainerExtensionToTheContainer(IUnityContainer container)
+            LoadPlatformLibrary();
-        {
+
-            container.AddNewExtension<AutoMockingContainerExtension>();
+            AssemblyLoader.RegisterSQLiteResolver();
-            return;
         }
 
         private Mock<T> TheRegisteredMockForThisType<T>(Type type)
@@ -177,7 +150,7 @@ namespace NzbDrone.Test.Common.AutoMoq
 
         private bool GetMockHasNotBeenCalledForThisType(Type type)
         {
-            return _registeredMocks.ContainsKey(type) == false;
+            return !_registeredMocks.ContainsKey(type);
         }
 
         private static Type GetTheMockType<T>()
@@ -186,7 +159,7 @@ namespace NzbDrone.Test.Common.AutoMoq
             return typeof(T);
         }
 
-        private void RegisterPlatformLibrary(IUnityContainer container)
+        private void LoadPlatformLibrary()
         {
             var assemblyName = "Prowlarr.Windows";
 

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

@@ -1,80 +0,0 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Reflection;
-using Moq;
-using Unity;
-using Unity.Builder;
-using Unity.Strategies;
-
-namespace NzbDrone.Test.Common.AutoMoq.Unity
-{
-    public 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(ref BuilderContext 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);
-            }
-        }
-
-        private bool AMockObjectShouldBeCreatedForThisType(Type type)
-        {
-            var mocker = _container.Resolve<AutoMoqer>();
-            return TypeIsNotRegistered(type) && (mocker.ResolveType == null || mocker.ResolveType != type);
-        }
-
-        private static Type GetTheTypeFromTheBuilderContext(BuilderContext context)
-        {
-            // return (context.OriginalBuildKey).Type;
-            return context.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[]) };
-        }
-    }
-}

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

@@ -1,35 +0,0 @@
-using System;
-using System.Collections.Generic;
-using Unity.Builder;
-using Unity.Extension;
-
-namespace NzbDrone.Test.Common.AutoMoq.Unity
-{
-    public class AutoMockingContainerExtension : UnityContainerExtension
-    {
-        private readonly IList<Type> _registeredTypes = new List<Type>();
-
-        protected override void Initialize()
-        {
-            SetEventsOnContainerToTrackAllRegisteredTypes();
-            SetBuildingStrategyForBuildingUnregisteredTypes();
-        }
-
-        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);
-        }
-    }
-}

src/NzbDrone.Test.Common/Prowlarr.Test.Common.csproj

@@ -10,7 +10,6 @@
     <PackageReference Include="NUnit" Version="3.13.3" />
     <PackageReference Include="RestSharp" Version="106.15.0" />
     <PackageReference Include="RestSharp.Serializers.SystemTextJson" Version="106.15.0" />
-    <PackageReference Include="Unity" Version="5.11.10" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\NzbDrone.Common\Prowlarr.Common.csproj" />