我这么玩Web Api(二):数据验证,全局数据验证与单元测试

时间:2021-12-18 04:24:23

目录

一、模型状态 - ModelState
二、数据注解 - Data Annotations
三、自定义数据注解
四、全局数据验证
五、单元测试
 

一、模型状态 - ModelState

  我理解的ModelState是微软在ASP.NET MVC中提出的一种新机制,它主要实现以下几个功能:

  1. 保存客户端传过来的数据,如果验证不通过,把数据返回到客户端,这样可以保存用户输入,不需要重新输入。

  2. 验证数据,以及保存数据对应的错误信息。

  3. 微软的一种DRY(Don't Repeat Yourself)设计,通过ModelState可以做服务端验证,同时可以配合jquery validation生成前端数据验证。

  但是在Web API里面,ModelState的主要功能就只剩下第2点了。

  需要注意的是,ModelState一般只做输入验证,一些其他的业务验证还有要在特定的地方进行处理。

二、数据注解 - Data Annotations

  数据注解可以理解为验证数据的逻辑或方法,微软本身有提供一批数据注解,当然我们也可以自定义数据注解,以下是微软提供的常见的数据注解:

  1. Required - 非空验证。

  当一个输入是null时会引发一个验证错误。

  当属性类型是string的时候,如果设置了AllowEmptyStrings = false(默认为false),那么输入空字符串或者空格,也会引发一个验证错误。

    [Required]
public string Name { get; set; } [Required(AllowEmptyStrings = true)]
public string Exchange { get; set; }

  2. StringLength - 长度验证。

  当输入大于指定最大长度,或者小于最大指定长度时,会引发一个验证错误。 

    [StringLength()]
public string Symbol { get; set; } [StringLength(, MinimumLength = )]
public string Name { get; set; }

  3. RegularExpression - 正则表达式验证。

  当输入内容不满足指定的正则表达式时,会引发一个验证错误。

  注:在.NET Framework 4.6.1添加了一个MatchTimeoutInMilliseconds属性,用来设定正则表达时验证时长。如超时,则抛出RegexMatchTimeoutException异常。

    [RegularExpression("your expression")]
public string Symbol { get; set; }

  4. Range - 值范围验证

  当输入的值小于最小值或者大于最大值时,会引发一个验证错误,这里要求验证字段的类型需要实现IComparable接口。

    [Range(, )]
public double OpenPrice { get; set; } [Range(typeof(double), "", "")]
public double ClosePrice { get; set; }

  5. Compare - 对比验证

  确保对象两个属性拥有相同的值。如果两个值不同,会引发一个验证错误。

    public string Name { get; set; }

    [Compare("Name")]
public string ConfirmName { get; set; }

  

  6. Remote - 远程调用验证

  Remote可以利用服务端回调函数执行客户端的验证逻辑。

  注:该数据注解是ASP.NET MVC特有的注解,在Web Api中无此注解。

    [Remote("CheckName", "Account"]
public string UserName{ get; set; } public class AccountController: Controller
{
public JsonResult CheckName(string name)
{
return Json(true);
}
}

三、自定义数据注解

  如果觉得微软提供的数据注解不够用,也可以自己写数据注解,只需要继承ValidationAttribute,并复写IsValid方法。

  下面是一个来自《ASP.NET MVC 5高级编程》的一个例子MaxWordsAttribute,用于限制属性的单词个数。

    public class MaxWordsAttribute : ValidationAttribute
{
private readonly int _maxWords; public MaxWordsAttribute(int maxWords)
{
_maxWords = maxWords;
} protected override ValidationResult IsValid(object value, ValidationContext validationContext)
{
if (value != null)
{
var valueAsString = value.ToString(); if (valueAsString.Split(' ').Length > _maxWords)
{
return new ValidationResult("Too many words!");
}
} return ValidationResult.Success;
}
}

  

    [Required]
[MaxWords()]
public string Name { get; set; }
    [HttpPost]
public IHttpActionResult Create(Stock stock)
{
if (!ModelState.IsValid)
{
return BadRequest(ModelState);
} return CreatedAtRoute("Get", new { symbol = stock.Symbol }, stock);
}

  Swashbuckle Help Page测试效果如下:

  我这么玩Web Api(二):数据验证,全局数据验证与单元测试

  如何使用Help Page可参考我上一篇文章《我这么玩Web Api(一):帮助页面或用户手册(Microsoft and Swashbuckle Help Page)》

四、全局数据验证

  我们在使用数据验证的时候,往往会出现许多重复的代码,如下图:

  我这么玩Web Api(二):数据验证,全局数据验证与单元测试

  有没有办法减少这些重复的代码呢?我从“Model Validation in ASP.NET Web API”这篇文章中找到了方法。

  首先,我们需要写一个GlobalActionFilterAttribute。

    public class GlobalActionFilterAttribute: ActionFilterAttribute
{
public override void OnActionExecuting(HttpActionContext actionContext)
{
if (actionContext.ModelState.IsValid == false)
{
actionContext.Response = actionContext.Request.CreateErrorResponse(HttpStatusCode.BadRequest, actionContext.ModelState);
}
}
}

  然后,在WebApiConfig里注册一下这个Attribute。

    public static void Register(HttpConfiguration config)
{
config.MapHttpAttributeRoutes(); config.Routes.MapHttpRoute("DefaultApi", "api/{controller}/{id}", new { id = RouteParameter.Optional } ); //register the custom action filter
config.Filters.Add(new GlobalActionFilterAttribute());
}

  那么,我们把Controller中的数据验证注释掉,依旧会得到相同的效果。

  

  如果想只对Post请求进行验证,可以在GlobalActionFilterAttribute加对请求方式的判断:

    public class GlobalActionFilterAttribute : ActionFilterAttribute
{
public override void OnActionExecuting(HttpActionContext actionContext)
{
//If you only want to validate the post request.
if (actionContext.Request.Method != HttpMethod.Post)
{
return;
} if (actionContext.ModelState.IsValid == false)
{
actionContext.Response = actionContext.Request.CreateErrorResponse(HttpStatusCode.BadRequest, actionContext.ModelState);
}
}
}

  如果某些Controller或Action需要绕过数据验证,那么可以这么实现:

  1. 定义一个BypassModelStateValidationAttribute

    [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, Inherited = false)]
public sealed class BypassModelStateValidationAttribute : Attribute
{ }

  

  2. 在不需要验证的Controller或者Action上加这个Attribute

    [HttpPut]
[BypassModelStateValidation]
public IHttpActionResult Update(Stock stock)
{
//if (!ModelState.IsValid)
//{
// return BadRequest(ModelState);
//} return StatusCode(HttpStatusCode.NoContent);
}

  3. 在GlobalActionFilterAttribute加对BypassModelStateValidationAttribute的判断:

    public class GlobalActionFilterAttribute : ActionFilterAttribute
{
public override void OnActionExecuting(HttpActionContext actionContext)
{
//If you only want to validate the post request.
if (actionContext.Request.Method != HttpMethod.Post)
{
return;
} var passby = actionContext.ActionDescriptor.GetCustomAttributes<BypassModelStateValidationAttribute>().Any() ||
actionContext.ControllerContext.ControllerDescriptor.GetCustomAttributes<BypassModelStateValidationAttribute>().Any(); if (passby)
{
return;
} if (actionContext.ModelState.IsValid == false)
{
actionContext.Response = actionContext.Request.CreateErrorResponse(HttpStatusCode.BadRequest, actionContext.ModelState);
}
}
}

五、单元测试

  我使用BDD的风格编写单元测试,关于BDD的详细信息,可查看我之前的文章《行为驱动开发(BDD)实践示例》

  对于全局数据验证,我设计了3个测试用例。

  1. 非Post请求不做验证 - HttpMethodNotMatched

  feature描述:

  我这么玩Web Api(二):数据验证,全局数据验证与单元测试

  测试代码:

    [Binding]
[Scope(Scenario = @"HttpMethodNotMatched")]
public class HttpMethodNotMatchedTest : GlobalActionFilterAttributeTests
{
[Given(@"非Post方式的请求")]
public void Given()
{
HttpActionContext.Request.Method = HttpMethod.Get;
} [When(@"执行OnActionExecuting方法")]
public void When()
{
GlobalActionFilterAttribute.OnActionExecuting(HttpActionContext);
} [Then(@"Response为空")]
public void Then()
{
Assert.IsNull(HttpActionContext.Response);
}
}

  2. 设置了跳过验证 - BypassModelStateValidation

  feature描述:

  我这么玩Web Api(二):数据验证,全局数据验证与单元测试

  测试代码:

    [Binding]
[Scope(Scenario = @"BypassModelStateValidation")]
public class BypassModelStateValidationTest : GlobalActionFilterAttributeTests
{
[Given(@"BypassModelStateValidationAttribute")]
public void Given()
{
HttpActionContext.Request.Method = HttpMethod.Post; HttpActionContext.ActionDescriptor = ActionDescriptorMock.Object;
ActionDescriptorMock.Setup(m => m.GetCustomAttributes<BypassModelStateValidationAttribute>()).Returns(new Collection<BypassModelStateValidationAttribute>(new[] { new BypassModelStateValidationAttribute() })); HttpActionContext.ControllerContext.ControllerDescriptor = ControllerDescriptorMock.Object;
ControllerDescriptorMock.Setup(m => m.GetCustomAttributes<BypassModelStateValidationAttribute>()).Returns(new Collection<BypassModelStateValidationAttribute>());
} [When(@"执行OnActionExecuting方法")]
public void When()
{
GlobalActionFilterAttribute.OnActionExecuting(HttpActionContext);
} [Then(@"Response为空")]
public void Then()
{
Assert.IsNull(HttpActionContext.Response);
}
}

  3. 验证不通过 - ModelStateInvalid

  feature描述:

  我这么玩Web Api(二):数据验证,全局数据验证与单元测试

  

  测试代码:

    [Binding]
[Scope(Scenario = @"ModelStateInvalid")]
public class ModelStateInvalidTest : GlobalActionFilterAttributeTests
{
[Given(@"ModelState错误信息")]
public void Given()
{
HttpActionContext.Request.Method = HttpMethod.Post; HttpActionContext.ActionDescriptor = ActionDescriptorMock.Object;
ActionDescriptorMock.Setup(m => m.GetCustomAttributes<BypassModelStateValidationAttribute>()).Returns(new Collection<BypassModelStateValidationAttribute>()); HttpActionContext.ControllerContext.ControllerDescriptor = ControllerDescriptorMock.Object;
ControllerDescriptorMock.Setup(m => m.GetCustomAttributes<BypassModelStateValidationAttribute>()).Returns(new Collection<BypassModelStateValidationAttribute>()); HttpActionContext.ModelState.AddModelError("stock.Name", "The Name field is required.");
} [When(@"执行OnActionExecuting方法")]
public void When()
{
GlobalActionFilterAttribute.OnActionExecuting(HttpActionContext);
} [Then(@"返回Bad Request")]
public void Then()
{
Assert.AreEqual(HttpStatusCode.BadRequest, HttpActionContext.Response.StatusCode);
}
}

  单元测试结果:

  我这么玩Web Api(二):数据验证,全局数据验证与单元测试

  说明:

  GlobalActionFilterAttributeTests是单元测试的父类,公共的部分可以抽取到这里。其中ContextUtil是微软源码中的测试辅助类。

    public class GlobalActionFilterAttributeTests
{
protected readonly Mock<HttpActionDescriptor> ActionDescriptorMock = new Mock<HttpActionDescriptor>();
protected readonly Mock<HttpControllerDescriptor> ControllerDescriptorMock = new Mock<HttpControllerDescriptor>();
protected HttpActionContext HttpActionContext;
protected GlobalActionFilterAttribute GlobalActionFilterAttribute; public GlobalActionFilterAttributeTests()
{
HttpActionContext = ContextUtil.CreateActionContext();
GlobalActionFilterAttribute = new GlobalActionFilterAttribute();
}
}

源码下载

https://github.com/ErikXu/WebApi.Trial