Created
September 26, 2011 15:59
-
-
Save graylikeme/1242578 to your computer and use it in GitHub Desktop.
Mock helper
This file contains 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
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 Mock<HttpSessionStateBase>(); | |
var server = new Mock<HttpServerUtilityBase>(); | |
context.Setup(ctx => ctx.Request).Returns(request.Object); | |
context.Setup(ctx => ctx.Response).Returns(response.Object); | |
context.Setup(ctx => ctx.Session).Returns(session.Object); | |
context.Setup(ctx => ctx.Server).Returns(server.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(); | |
var context = new ControllerContext(new RequestContext(httpContext, new RouteData()), controller); | |
controller.ControllerContext = context; | |
} | |
public static void SetFakeControllerContext(this Controller controller, RouteData routeData) | |
{ | |
var httpContext = FakeHttpContext(); | |
var context = new ControllerContext(new RequestContext(httpContext, routeData), controller); | |
controller.ControllerContext = context; | |
} | |
static string GetUrlFileName(string url) | |
{ | |
if (url.Contains("?")) | |
return url.Substring(0, url.IndexOf("?")); | |
return url; | |
} | |
static NameValueCollection GetQueryStringParameters(string url) | |
{ | |
if (url.Contains("?")) | |
{ | |
var parameters = new NameValueCollection(); | |
string[] parts = url.Split("?".ToCharArray()); | |
string[] keys = parts[1].Split("&".ToCharArray()); | |
foreach (var part in keys.Select(key => key.Split("=".ToCharArray()))) | |
{ | |
parameters.Add(part[0], part[1]); | |
} | |
return parameters; | |
} | |
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); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment