mirror of
https://github.com/Radarr/Radarr.git
synced 2024-09-11 12:02:35 +02:00
Use DryIoc for Automoqer, drop Unity dependency
[common]
This commit is contained in:
parent
f2a7d0d520
commit
e3468daba0
@ -3,17 +3,11 @@
|
|||||||
using System.Diagnostics;
|
using System.Diagnostics;
|
||||||
using System.IO;
|
using System.IO;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using System.Linq.Expressions;
|
|
||||||
using System.Reflection;
|
using System.Reflection;
|
||||||
using System.Runtime.CompilerServices;
|
using DryIoc;
|
||||||
using Moq;
|
using Moq;
|
||||||
using Moq.Language.Flow;
|
|
||||||
using NzbDrone.Common.Composition;
|
using NzbDrone.Common.Composition;
|
||||||
using NzbDrone.Common.EnvironmentInfo;
|
using NzbDrone.Common.EnvironmentInfo;
|
||||||
using NzbDrone.Test.Common.AutoMoq.Unity;
|
|
||||||
using Unity;
|
|
||||||
|
|
||||||
[assembly: InternalsVisibleTo("AutoMoq.Tests")]
|
|
||||||
|
|
||||||
namespace NzbDrone.Test.Common.AutoMoq
|
namespace NzbDrone.Test.Common.AutoMoq
|
||||||
{
|
{
|
||||||
@ -21,32 +15,18 @@ namespace NzbDrone.Test.Common.AutoMoq
|
|||||||
public class AutoMoqer
|
public class AutoMoqer
|
||||||
{
|
{
|
||||||
public readonly MockBehavior DefaultBehavior = MockBehavior.Default;
|
public readonly MockBehavior DefaultBehavior = MockBehavior.Default;
|
||||||
public Type ResolveType;
|
private IContainer _container;
|
||||||
private IUnityContainer _container;
|
|
||||||
private IDictionary<Type, object> _registeredMocks;
|
private IDictionary<Type, object> _registeredMocks;
|
||||||
|
|
||||||
public AutoMoqer()
|
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>()
|
public virtual T Resolve<T>()
|
||||||
{
|
{
|
||||||
ResolveType = typeof(T);
|
|
||||||
var result = _container.Resolve<T>();
|
var result = _container.Resolve<T>();
|
||||||
SetConstant(result);
|
SetConstant(result);
|
||||||
ResolveType = null;
|
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -59,7 +39,6 @@ public virtual Mock<T> GetMock<T>()
|
|||||||
public virtual Mock<T> GetMock<T>(MockBehavior behavior)
|
public virtual Mock<T> GetMock<T>(MockBehavior behavior)
|
||||||
where T : class
|
where T : class
|
||||||
{
|
{
|
||||||
ResolveType = null;
|
|
||||||
var type = GetTheMockType<T>();
|
var type = GetTheMockType<T>();
|
||||||
if (GetMockHasNotBeenCalledForThisType(type))
|
if (GetMockHasNotBeenCalledForThisType(type))
|
||||||
{
|
{
|
||||||
@ -78,90 +57,83 @@ public virtual Mock<T> GetMock<T>(MockBehavior behavior)
|
|||||||
|
|
||||||
public virtual void SetMock(Type type, Mock mock)
|
public virtual void SetMock(Type type, Mock mock)
|
||||||
{
|
{
|
||||||
if (_registeredMocks.ContainsKey(type) == false)
|
if (GetMockHasNotBeenCalledForThisType(type))
|
||||||
{
|
{
|
||||||
_registeredMocks.Add(type, mock);
|
_registeredMocks.Add(type, mock);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (mock != null)
|
if (mock != null)
|
||||||
{
|
{
|
||||||
_container.RegisterInstance(type, mock.Object);
|
_container.RegisterInstance(type, mock.Object, ifAlreadyRegistered: IfAlreadyRegistered.Replace);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual void SetConstant<T>(T instance)
|
public virtual void SetConstant<T>(T instance)
|
||||||
{
|
{
|
||||||
_container.RegisterInstance(instance);
|
_container.RegisterInstance(instance, ifAlreadyRegistered: IfAlreadyRegistered.Replace);
|
||||||
SetMock(instance.GetType(), null);
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (serviceType == typeof(object))
|
||||||
|
{
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (serviceType.IsGenericType && serviceType.IsOpenGeneric())
|
||||||
|
{
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the Mock object for the abstract class or interface
|
||||||
|
if (serviceType.IsInterface || serviceType.IsAbstract)
|
||||||
|
{
|
||||||
|
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);
|
||||||
|
|
||||||
|
return new[] { new DynamicRegistration(mockFactory, IfAlreadyRegistered.Keep) };
|
||||||
|
}
|
||||||
|
|
||||||
|
// concrete types
|
||||||
|
var concreteTypeFactory = serviceType.ToFactory(Reuse.Singleton, FactoryMethod.ConstructorWithResolvableArgumentsIncludingNonPublic);
|
||||||
|
|
||||||
|
return new[] { new DynamicRegistration(concreteTypeFactory) };
|
||||||
|
},
|
||||||
|
DynamicRegistrationFlags.Service | DynamicRegistrationFlags.AsFallback));
|
||||||
|
|
||||||
|
c.Register(typeof(Mock<>), Reuse.Singleton, FactoryMethod.DefaultConstructor());
|
||||||
|
|
||||||
|
return c;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ISetup<T, TResult> Setup<T, TResult>(Expression<Func<T, TResult>> expression)
|
private void SetupAutoMoqer(IContainer container)
|
||||||
where T : class
|
|
||||||
{
|
|
||||||
return GetMock<T>().Setup(expression);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Verify<T>(Expression<Action<T>> expression)
|
|
||||||
where T : class
|
|
||||||
{
|
|
||||||
GetMock<T>().Verify(expression);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Verify<T>(Expression<Action<T>> expression, string failMessage)
|
|
||||||
where T : class
|
|
||||||
{
|
|
||||||
GetMock<T>().Verify(expression, failMessage);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Verify<T>(Expression<Action<T>> expression, Times times)
|
|
||||||
where T : class
|
|
||||||
{
|
|
||||||
GetMock<T>().Verify(expression, times);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Verify<T>(Expression<Action<T>> expression, Times times, string failMessage)
|
|
||||||
where T : class
|
|
||||||
{
|
|
||||||
GetMock<T>().Verify(expression, times, failMessage);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void VerifyAllMocks()
|
|
||||||
{
|
|
||||||
foreach (var registeredMock in _registeredMocks)
|
|
||||||
{
|
|
||||||
var mock = registeredMock.Value as Mock;
|
|
||||||
if (mock != null)
|
|
||||||
{
|
|
||||||
mock.VerifyAll();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void SetupAutoMoqer(IUnityContainer container)
|
|
||||||
{
|
{
|
||||||
_container = container;
|
_container = container;
|
||||||
container.RegisterInstance(this);
|
container.RegisterInstance(this);
|
||||||
|
|
||||||
_registeredMocks = new Dictionary<Type, object>();
|
_registeredMocks = new Dictionary<Type, object>();
|
||||||
|
|
||||||
RegisterPlatformLibrary(container);
|
LoadPlatformLibrary();
|
||||||
AddTheAutoMockingContainerExtensionToTheContainer(container);
|
|
||||||
|
|
||||||
AssemblyLoader.RegisterNativeResolver(new[] { "System.Data.SQLite", "Radarr.Core" });
|
AssemblyLoader.RegisterNativeResolver(new[] { "System.Data.SQLite", "Radarr.Core" });
|
||||||
}
|
}
|
||||||
|
|
||||||
private static void AddTheAutoMockingContainerExtensionToTheContainer(IUnityContainer container)
|
|
||||||
{
|
|
||||||
container.AddNewExtension<AutoMockingContainerExtension>();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
private Mock<T> TheRegisteredMockForThisType<T>(Type type)
|
private Mock<T> TheRegisteredMockForThisType<T>(Type type)
|
||||||
where T : class
|
where T : class
|
||||||
{
|
{
|
||||||
@ -178,7 +150,7 @@ private void CreateANewMockAndRegisterIt<T>(Type type, MockBehavior behavior)
|
|||||||
|
|
||||||
private bool GetMockHasNotBeenCalledForThisType(Type type)
|
private bool GetMockHasNotBeenCalledForThisType(Type type)
|
||||||
{
|
{
|
||||||
return _registeredMocks.ContainsKey(type) == false;
|
return !_registeredMocks.ContainsKey(type);
|
||||||
}
|
}
|
||||||
|
|
||||||
private static Type GetTheMockType<T>()
|
private static Type GetTheMockType<T>()
|
||||||
@ -187,7 +159,7 @@ private static Type GetTheMockType<T>()
|
|||||||
return typeof(T);
|
return typeof(T);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void RegisterPlatformLibrary(IUnityContainer container)
|
private void LoadPlatformLibrary()
|
||||||
{
|
{
|
||||||
var assemblyName = "Radarr.Windows";
|
var assemblyName = "Radarr.Windows";
|
||||||
|
|
||||||
|
@ -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[]) };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -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);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -9,7 +9,6 @@
|
|||||||
<PackageReference Include="NLog" Version="4.7.14" />
|
<PackageReference Include="NLog" Version="4.7.14" />
|
||||||
<PackageReference Include="NUnit" Version="3.13.2" />
|
<PackageReference Include="NUnit" Version="3.13.2" />
|
||||||
<PackageReference Include="RestSharp" Version="106.15.0" />
|
<PackageReference Include="RestSharp" Version="106.15.0" />
|
||||||
<PackageReference Include="Unity" Version="5.11.10" />
|
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ProjectReference Include="..\NzbDrone.Common\Radarr.Common.csproj" />
|
<ProjectReference Include="..\NzbDrone.Common\Radarr.Common.csproj" />
|
||||||
|
Loading…
Reference in New Issue
Block a user