Created
July 29, 2009 22:15
-
-
Save austegard/158430 to your computer and use it in GitHub Desktop.
A correction of the code on http://www.hanselman.com/blog/ASPNETMVCSessionAtMix08TDDAndMvcMockHelpers.aspx
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
using System; | |
using System.Collections.Specialized; | |
using System.Web; | |
using System.Web.Mvc; | |
using System.Web.Routing; | |
using Moq; | |
namespace Mvc.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 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(); | |
ControllerContext context = new ControllerContext(new RequestContext(httpContext, new RouteData()), controller); | |
controller.ControllerContext = context; | |
} | |
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); | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment