2013-11-26 222 views
6

在我目前正在使用的ASP.Net MVC 4应用程序中,有许多具有仓库属性的模型。我希望所有这些模型都具有验证功能,以确保输入的仓库是有效的仓库。看起来最简单的方法是使用自定义的ValidationAttribute类。然后验证代码将被集中,我可以将属性添加到每个模型中的属性中。ASP.Net MVC 4自定义验证属性依赖注入

我需要调用服务以确保仓库是有效的仓库。我有一个代表此服务的接口,我正在使用Ninject在使用此服务的应用程序中执行依赖注入。这样我就可以使用模拟并轻松地对应用程序进行单元测试。

我希望我的自定义ValidationAttribute类在使用此服务时使用依赖注入。这里是我创建的类:

public class MustBeValidWarehouse : ValidationAttribute 
{ 
    public override bool IsValid(object value) 
    { 
    if (value is string) 
    { 
     string warehouse = value.ToString(); 
     NinjectDependencyResolver depres = new NinjectDependencyResolver(); 
     Type inventServiceType = typeof(IInventService); 
     IInventService inventserv = depres.GetService(inventServiceType) as IInventService; 
     return (inventserv.GetWarehouses().Where(m => m.WarehouseId == warehouse).Count() != 0); 

    } 
    else 
    { 
     return false; 
    } 
    } 
} 


public class NinjectDependencyResolver : IDependencyResolver 
{ 
    private IKernel kernel; 
    public NinjectDependencyResolver() 
    { 
     kernel = new StandardKernel(); 
     AddBindings(); 
    } 

    public object GetService(Type serviceType) 
    { 
     return kernel.TryGet(serviceType); 
    } 

    public IEnumerable<object> GetServices(Type serviceType) 
    { 
     return kernel.GetAll(serviceType); 
    } 

    private void AddBindings() 
    { 
     kernel.Bind<IInventService>().To<InventService>(); 
    } 
} 

依赖注入工作正常,但它不容易测试。在单元测试中,没有办法将模拟的IInventService注入到类中。通常为了解决这个问题,我会让类构造函数接受一个I​​InventService参数,这样我就可以在我的单元测试中传入一个模拟对象。然而,我不认为我可以让这个类构造函数将IInventService类作为参数,因为那样我相信当我在我的类中添加此属性时,我必须传入该参数。

有没有办法让这段代码更具可测性?如果不是,那么有没有更好的方法来处理这个问题?

回答

9

您需要在ASP.NET MVC中使用DependencyResolver类。如果正确连接容器DependencyResolver.Current将使用您的容器来解决依赖关系。

public class MustBeValidWarehouse : ValidationAttribute 
{ 
    public override bool IsValid(object value) 
    { 
     if (value is string) 
     { 
      string warehouse = value.ToString(); 
      IInventService inventserv = DependencyResolver.Current.GetService<IInventService>(); 
      return (inventserv.GetWarehouses().Where(m => m.WarehouseId == warehouse).Count() != 0); 
     } 
     return false; 
    } 
} 

在您的课堂测试您可以DepedencyResolver.Current这样提供一个模拟:

DependencyResolver.SetResolver(resolverMock); 
+0

伟大的工作,谢谢。 –