85 lines
2.7 KiB
C#
85 lines
2.7 KiB
C#
// 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 NzbDrone.Test.Common.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
|
|
}
|
|
} |