1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
// 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;
}
}
}
|
