Created
October 25, 2013 06:41
-
-
Save ctolkien/7150322 to your computer and use it in GitHub Desktop.
var foo = new SomeController();
foo.SetFakeControllerContext();
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
using System; | |
using System.Collections.Generic; | |
using System.Collections.Specialized; | |
using System.Linq; | |
using System.Security.Principal; | |
using System.Web; | |
using System.Web.Mvc; | |
using System.Web.Routing; | |
using Moq; | |
namespace Tab.Web.Tests | |
{ | |
public static class MvcMockHelpers | |
{ | |
public static HttpContextBase FakeHttpContext() | |
{ | |
var context = new Mock<HttpContextBase>(); | |
var request = new Mock<HttpRequestBase>(); | |
var response = new Mock<HttpResponseBase>(); | |
var session = new MockHttpSession(); | |
var server = new Mock<HttpServerUtilityBase>(); | |
var user = new Mock<IPrincipal>(); | |
//var url = new Mock<UrlHelper>(); | |
server.Setup(x => x.UrlDecode(It.IsAny<string>())).Returns("user1@fake.dom"); | |
var cookieCollection = new HttpCookieCollection(); | |
response.Setup(x => x.Cookies).Returns(cookieCollection); | |
request.Setup(x => x.Cookies).Returns(cookieCollection); | |
user.Setup(x => x.Identity.Name).Returns("1"); | |
context.Setup(ctx => ctx.Request).Returns(request.Object); | |
context.Setup(ctx => ctx.Response).Returns(response.Object); | |
context.Setup(ctx => ctx.Session).Returns(session); | |
context.Setup(ctx => ctx.Server).Returns(server.Object); | |
context.Setup(ctx => ctx.User).Returns(user.Object); | |
context.Setup(ctx => ctx.Request).Returns(request.Object); | |
context.Setup(ctx => ctx.Response).Returns(response.Object); | |
context.Setup(ctx => ctx.Session).Returns(session); | |
context.Setup(ctx => ctx.Server).Returns(server.Object); | |
request.SetupGet(x => x.ApplicationPath).Returns("/"); | |
request.SetupGet(x => x.Url).Returns(new Uri("http://localhost/a", UriKind.Absolute)); | |
request.SetupGet(x => x.ServerVariables).Returns(new NameValueCollection()); | |
response.Setup(x => x.ApplyAppPathModifier(It.IsAny<string>())).Returns<string>(x => x); | |
context.SetupGet(x => x.Request).Returns(request.Object); | |
context.SetupGet(x => x.Response).Returns(response.Object); | |
return context.Object; | |
} | |
public static HttpContextBase FakeHttpContext(string url) | |
{ | |
HttpContextBase context = FakeHttpContext(); | |
context.Request.SetupRequestUrl(url); | |
return context; | |
} | |
public static void SetFakeControllerContext(this Controller controller) | |
{ | |
var httpContext = FakeHttpContext(); | |
ControllerContext context = new ControllerContext(new RequestContext(httpContext, new RouteData()), controller); | |
RequestContext requestcontext = new RequestContext(httpContext, new RouteData()); | |
controller.ControllerContext = context; | |
controller.Url = new UrlHelper(requestcontext); | |
} | |
static string GetUrlFileName(string url) | |
{ | |
if (url.Contains("?")) | |
return url.Substring(0, url.IndexOf("?")); | |
else | |
return url; | |
} | |
static NameValueCollection GetQueryStringParameters(string url) | |
{ | |
if (url.Contains("?")) | |
{ | |
NameValueCollection parameters = new NameValueCollection(); | |
string[] parts = url.Split("?".ToCharArray()); | |
string[] keys = parts[1].Split("&".ToCharArray()); | |
foreach (string key in keys) | |
{ | |
string[] part = key.Split("=".ToCharArray()); | |
parameters.Add(part[0], part[1]); | |
} | |
return parameters; | |
} | |
else | |
{ | |
return null; | |
} | |
} | |
public static void SetHttpMethodResult(this HttpRequestBase request, string httpMethod) | |
{ | |
Mock.Get(request) | |
.Setup(req => req.HttpMethod) | |
.Returns(httpMethod); | |
} | |
public static void SetupRequestUrl(this HttpRequestBase request, string url) | |
{ | |
if (url == null) | |
throw new ArgumentNullException("url"); | |
if (!url.StartsWith("~/")) | |
throw new ArgumentException("Sorry, we expect a virtual url starting with \"~/\"."); | |
var mock = Mock.Get(request); | |
mock.Setup(req => req.QueryString) | |
.Returns(GetQueryStringParameters(url)); | |
mock.Setup(req => req.AppRelativeCurrentExecutionFilePath) | |
.Returns(GetUrlFileName(url)); | |
mock.Setup(req => req.PathInfo) | |
.Returns(string.Empty); | |
} | |
} | |
public class MockHttpSession : HttpSessionStateBase | |
{ | |
readonly Dictionary<string, object> sessionStorage = new Dictionary<string, object>(); | |
public override object this[string name] | |
{ | |
get { return sessionStorage[name]; } | |
set { sessionStorage[name] = value; } | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment