69 lines
2.2 KiB
C#
69 lines
2.2 KiB
C#
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
|
|
|
|
using System.Collections.Specialized;
|
|
using System.Web.Mvc;
|
|
using Moq;
|
|
using Xunit;
|
|
using Assert = Microsoft.TestCommon.AssertEx;
|
|
|
|
namespace Microsoft.Web.Mvc.Test
|
|
{
|
|
public class AjaxOnlyAttributeTest
|
|
{
|
|
[Fact]
|
|
public void IsValidForRequestReturnsFalseIfHeaderNotPresent()
|
|
{
|
|
// Arrange
|
|
AjaxOnlyAttribute attr = new AjaxOnlyAttribute();
|
|
ControllerContext controllerContext = GetControllerContext(containsHeader: false);
|
|
|
|
// Act
|
|
bool isValid = attr.IsValidForRequest(controllerContext, null);
|
|
|
|
// Assert
|
|
Assert.False(isValid);
|
|
}
|
|
|
|
[Fact]
|
|
public void IsValidForRequestReturnsTrueIfHeaderIsPresent()
|
|
{
|
|
// Arrange
|
|
AjaxOnlyAttribute attr = new AjaxOnlyAttribute();
|
|
ControllerContext controllerContext = GetControllerContext(containsHeader: true);
|
|
|
|
// Act
|
|
bool isValid = attr.IsValidForRequest(controllerContext, null);
|
|
|
|
// Assert
|
|
Assert.True(isValid);
|
|
}
|
|
|
|
[Fact]
|
|
public void IsValidForRequestThrowsIfControllerContextIsNull()
|
|
{
|
|
// Arrange
|
|
AjaxOnlyAttribute attr = new AjaxOnlyAttribute();
|
|
|
|
// Act & assert
|
|
Assert.ThrowsArgumentNull(
|
|
delegate { attr.IsValidForRequest(null, null); }, "controllerContext");
|
|
}
|
|
|
|
private static ControllerContext GetControllerContext(bool containsHeader)
|
|
{
|
|
Mock<ControllerContext> mockContext = new Mock<ControllerContext> { DefaultValue = DefaultValue.Mock };
|
|
|
|
NameValueCollection nvc = new NameValueCollection();
|
|
if (containsHeader)
|
|
{
|
|
nvc["X-Requested-With"] = "XMLHttpRequest";
|
|
}
|
|
|
|
mockContext.Setup(o => o.HttpContext.Request.Headers).Returns(nvc);
|
|
mockContext.Setup(o => o.HttpContext.Request["X-Requested-With"]).Returns("XMLHttpRequest"); // always assume the request contains this, e.g. as a form value
|
|
|
|
return mockContext.Object;
|
|
}
|
|
}
|
|
}
|