using System; using System.Collections.Generic; using System.Linq; using JetBrains.Annotations; using Newtonsoft.Json; using Newtonsoft.Json.Linq; using WireMock.Admin.Mappings; using WireMock.Admin.Requests; using WireMock.Logging; using WireMock.Matchers; using WireMock.Matchers.Request; using WireMock.RequestBuilders; using WireMock.ResponseBuilders; namespace WireMock.Server { /// /// The fluent mock server. /// public partial class FluentMockServer { private const string AdminMappings = "/__admin/mappings"; private const string AdminRequests = "/__admin/requests"; private readonly RegexMatcher _adminMappingsGuidPathMatcher = new RegexMatcher(@"^\/__admin\/mappings\/(\{{0,1}([0-9a-fA-F]){8}-([0-9a-fA-F]){4}-([0-9a-fA-F]){4}-([0-9a-fA-F]){4}-([0-9a-fA-F]){12}\}{0,1})$"); private readonly RegexMatcher _adminRequestsGuidPathMatcher = new RegexMatcher(@"^\/__admin\/requests\/(\{{0,1}([0-9a-fA-F]){8}-([0-9a-fA-F]){4}-([0-9a-fA-F]){4}-([0-9a-fA-F]){4}-([0-9a-fA-F]){12}\}{0,1})$"); private readonly JsonSerializerSettings _settings = new JsonSerializerSettings { Formatting = Formatting.None, NullValueHandling = NullValueHandling.Ignore }; private void InitAdmin() { // __admin/mappings Given(Request.Create().WithPath(AdminMappings).UsingGet()).RespondWith(new DynamicResponseProvider(MappingsGet)); Given(Request.Create().WithPath(AdminMappings).UsingPost()).RespondWith(new DynamicResponseProvider(MappingsPost)); Given(Request.Create().WithPath(AdminMappings).UsingDelete()).RespondWith(new DynamicResponseProvider(MappingsDelete)); // __admin/mappings/{guid} Given(Request.Create().WithPath(_adminMappingsGuidPathMatcher).UsingGet()).RespondWith(new DynamicResponseProvider(MappingGet)); Given(Request.Create().WithPath(_adminMappingsGuidPathMatcher).UsingPut().WithHeader("Content-Type", "application/json")).RespondWith(new DynamicResponseProvider(MappingPut)); Given(Request.Create().WithPath(_adminMappingsGuidPathMatcher).UsingDelete()).RespondWith(new DynamicResponseProvider(MappingDelete)); // __admin/requests Given(Request.Create().WithPath(AdminRequests).UsingGet()).RespondWith(new DynamicResponseProvider(RequestsGet)); Given(Request.Create().WithPath(AdminRequests).UsingDelete()).RespondWith(new DynamicResponseProvider(RequestsDelete)); // __admin/request/{guid} Given(Request.Create().WithPath(_adminRequestsGuidPathMatcher).UsingGet()).RespondWith(new DynamicResponseProvider(RequestGet)); Given(Request.Create().WithPath(_adminRequestsGuidPathMatcher).UsingDelete()).RespondWith(new DynamicResponseProvider(RequestDelete)); } #region Mapping/{guid} private ResponseMessage MappingGet(RequestMessage requestMessage) { Guid guid = Guid.Parse(requestMessage.Path.Substring(AdminMappings.Length + 1)); var mapping = Mappings.FirstOrDefault(m => !(m.Provider is DynamicResponseProvider) && m.Guid == guid); if (mapping == null) return new ResponseMessage { StatusCode = 404, Body = "Mapping not found" }; var model = ToMappingModel(mapping); return ToJson(model); } private ResponseMessage MappingPut(RequestMessage requestMessage) { Guid guid = Guid.Parse(requestMessage.Path.TrimStart(AdminMappings.ToCharArray())); var mappingModel = JsonConvert.DeserializeObject(requestMessage.Body); if (mappingModel.Request == null) return new ResponseMessage { StatusCode = 400, Body = "Request missing" }; if (mappingModel.Response == null) return new ResponseMessage { StatusCode = 400, Body = "Response missing" }; var requestBuilder = InitRequestBuilder(mappingModel); var responseBuilder = InitResponseBuilder(mappingModel); Given(requestBuilder) .WithGuid(guid) .RespondWith(responseBuilder); return new ResponseMessage { Body = "Mapping updated" }; } private ResponseMessage MappingDelete(RequestMessage requestMessage) { Guid guid = Guid.Parse(requestMessage.Path.Substring(AdminMappings.Length + 1)); if (DeleteMapping(guid)) return new ResponseMessage { Body = "Mapping removed" }; return new ResponseMessage { Body = "Mapping not found" }; } #endregion Mapping/{guid} #region Mappings private ResponseMessage MappingsGet(RequestMessage requestMessage) { var result = new List(); foreach (var mapping in Mappings.Where(m => !(m.Provider is DynamicResponseProvider))) { var model = ToMappingModel(mapping); result.Add(model); } return ToJson(result); } private ResponseMessage MappingsPost(RequestMessage requestMessage) { var mappingModel = JsonConvert.DeserializeObject(requestMessage.Body); if (mappingModel.Request == null) return new ResponseMessage { StatusCode = 400, Body = "Request missing" }; if (mappingModel.Response == null) return new ResponseMessage { StatusCode = 400, Body = "Response missing" }; var requestBuilder = InitRequestBuilder(mappingModel); var responseBuilder = InitResponseBuilder(mappingModel); IRespondWithAProvider respondProvider = Given(requestBuilder); if (mappingModel.Guid != null && mappingModel.Guid != Guid.Empty) respondProvider = respondProvider.WithGuid(mappingModel.Guid.Value); if (mappingModel.Priority != null) respondProvider = respondProvider.AtPriority(mappingModel.Priority.Value); respondProvider.RespondWith(responseBuilder); return new ResponseMessage { Body = "Mapping added" }; } private ResponseMessage MappingsDelete(RequestMessage requestMessage) { ResetMappings(); return new ResponseMessage { Body = "Mappings deleted" }; } #endregion Mappings #region Request/{guid} private ResponseMessage RequestGet(RequestMessage requestMessage) { Guid guid = Guid.Parse(requestMessage.Path.Substring(AdminRequests.Length + 1)); var entry = LogEntries.FirstOrDefault(r => !r.RequestMessage.Path.StartsWith("/__admin/") && r.Guid == guid); if (entry == null) return new ResponseMessage { StatusCode = 404, Body = "Request not found" }; var model = ToLogEntryModel(entry); return ToJson(model); } private ResponseMessage RequestDelete(RequestMessage requestMessage) { Guid guid = Guid.Parse(requestMessage.Path.Substring(AdminRequests.Length + 1)); if (DeleteLogEntry(guid)) return new ResponseMessage { Body = "Request removed" }; return new ResponseMessage { Body = "Request not found" }; } #endregion Request/{guid} #region Requests private ResponseMessage RequestsGet(RequestMessage requestMessage) { var result = LogEntries .Where(r => !r.RequestMessage.Path.StartsWith("/__admin/")) .Select(ToLogEntryModel); return ToJson(result); } private LogEntryModel ToLogEntryModel(LogEntry logEntry) { return new LogEntryModel { Guid = logEntry.Guid, Request = new LogRequestModel { DateTime = logEntry.RequestMessage.DateTime, Url = logEntry.RequestMessage.Path, AbsoleteUrl = logEntry.RequestMessage.Url, Query = logEntry.RequestMessage.Query, Method = logEntry.RequestMessage.Method, Body = logEntry.RequestMessage.Body, Headers = logEntry.RequestMessage.Headers, Cookies = logEntry.RequestMessage.Cookies }, Response = new LogResponseModel { StatusCode = logEntry.ResponseMessage.StatusCode, Body = logEntry.ResponseMessage.Body, BodyOriginal = logEntry.ResponseMessage.BodyOriginal, Headers = logEntry.ResponseMessage.Headers } }; } private ResponseMessage RequestsDelete(RequestMessage requestMessage) { ResetLogEntries(); return new ResponseMessage { Body = "Requests deleted" }; } #endregion Requests private IRequestBuilder InitRequestBuilder(MappingModel mappingModel) { IRequestBuilder requestBuilder = Request.Create(); string path = mappingModel.Request.Path as string; if (path != null) requestBuilder = requestBuilder.WithPath(path); else { JToken pathToken = (JToken)mappingModel.Request.Path; PathModel pathModel = pathToken.ToObject(); if (pathModel?.Matchers != null) requestBuilder = requestBuilder.WithPath(pathModel.Matchers.Select(Map).ToArray()); } string url = mappingModel.Request.Url as string; if (url != null) requestBuilder = requestBuilder.WithUrl(url); else { JToken urlToken = (JToken)mappingModel.Request.Url; UrlModel urlModel = urlToken.ToObject(); if (urlModel?.Matchers != null) requestBuilder = requestBuilder.WithUrl(urlModel.Matchers.Select(Map).ToArray()); } if (mappingModel.Request.Methods != null) requestBuilder = requestBuilder.UsingVerb(mappingModel.Request.Methods); else requestBuilder = requestBuilder.UsingAnyVerb(); if (mappingModel.Request.Headers != null) { foreach (var headerModel in mappingModel.Request.Headers.Where(h => h.Matchers != null)) { requestBuilder = requestBuilder.WithHeader(headerModel.Name, headerModel.Matchers.Select(Map).ToArray()); } } if (mappingModel.Request.Cookies != null) { foreach (var cookieModel in mappingModel.Request.Cookies.Where(c => c.Matchers != null)) { requestBuilder = requestBuilder.WithCookie(cookieModel.Name, cookieModel.Matchers.Select(Map).ToArray()); } } if (mappingModel.Request.Params != null) { foreach (var paramModel in mappingModel.Request.Params.Where(p => p.Values != null)) { requestBuilder = requestBuilder.WithParam(paramModel.Name, paramModel.Values.ToArray()); } } if (mappingModel.Request.Body?.Matcher != null) { var bodyMatcher = Map(mappingModel.Request.Body.Matcher); requestBuilder = requestBuilder.WithBody(bodyMatcher); } return requestBuilder; } private IResponseBuilder InitResponseBuilder(MappingModel mappingModel) { IResponseBuilder responseBuilder = Response.Create(); if (mappingModel.Response.StatusCode.HasValue) responseBuilder = responseBuilder.WithStatusCode(mappingModel.Response.StatusCode.Value); if (mappingModel.Response.Headers != null) responseBuilder = responseBuilder.WithHeaders(mappingModel.Response.Headers); if (mappingModel.Response.Body != null) responseBuilder = responseBuilder.WithBody(mappingModel.Response.Body); else if (mappingModel.Response.BodyAsJson != null) responseBuilder = responseBuilder.WithBodyAsJson(mappingModel.Response.BodyAsJson); else if (mappingModel.Response.BodyAsBase64 != null) responseBuilder = responseBuilder.WithBodyAsBase64(mappingModel.Response.BodyAsBase64); if (mappingModel.Response.UseTransformer) responseBuilder = responseBuilder.WithTransformer(); if (mappingModel.Response.Delay != null) responseBuilder = responseBuilder.WithDelay(mappingModel.Response.Delay.Value); return responseBuilder; } private MappingModel ToMappingModel(Mapping mapping) { var request = (Request)mapping.RequestMatcher; var response = (Response)mapping.Provider; var pathMatchers = request.GetRequestMessageMatchers(); var urlMatchers = request.GetRequestMessageMatchers(); var headerMatchers = request.GetRequestMessageMatchers(); var cookieMatchers = request.GetRequestMessageMatchers(); var paramsMatchers = request.GetRequestMessageMatchers(); var bodyMatcher = request.GetRequestMessageMatcher(); var methodMatcher = request.GetRequestMessageMatcher(); return new MappingModel { Guid = mapping.Guid, Priority = mapping.Priority, Request = new RequestModel { Path = pathMatchers != null ? new PathModel { Matchers = Map(pathMatchers.Where(m => m.Matchers != null).SelectMany(m => m.Matchers)), Funcs = Map(pathMatchers.Where(m => m.Funcs != null).SelectMany(m => m.Funcs)) } : null, Url = urlMatchers != null ? new UrlModel { Matchers = Map(urlMatchers.Where(m => m.Matchers != null).SelectMany(m => m.Matchers)), Funcs = Map(urlMatchers.Where(m => m.Funcs != null).SelectMany(m => m.Funcs)) } : null, Methods = methodMatcher != null ? methodMatcher.Methods : new[] { "any" }, Headers = headerMatchers?.Select(hm => new HeaderModel { Name = hm.Name, Matchers = Map(hm.Matchers), Funcs = Map(hm.Funcs) }).ToList(), Cookies = cookieMatchers?.Select(cm => new CookieModel { Name = cm.Name, Matchers = Map(cm.Matchers), Funcs = Map(cm.Funcs) }).ToList(), Params = paramsMatchers?.Select(pm => new ParamModel { Name = pm.Key, Values = pm.Values?.ToList(), Funcs = Map(pm.Funcs) }).ToList(), Body = new BodyModel { Matcher = bodyMatcher != null ? Map(bodyMatcher.Matcher) : null, Func = bodyMatcher != null ? Map(bodyMatcher.Func) : null, DataFunc = bodyMatcher != null ? Map(bodyMatcher.DataFunc) : null } }, Response = new ResponseModel { StatusCode = response.ResponseMessage.StatusCode, Headers = response.ResponseMessage.Headers, Body = response.ResponseMessage.Body, UseTransformer = response.UseTransformer, Delay = response.Delay?.Milliseconds } }; } private MatcherModel[] Map([CanBeNull] IEnumerable matchers) { return matchers?.Select(Map).Where(x => x != null).ToArray(); } private MatcherModel Map([CanBeNull] IMatcher matcher) { if (matcher == null) return null; return new MatcherModel { Name = matcher.GetType().Name, Pattern = matcher.GetPattern() }; } private string[] Map([CanBeNull] IEnumerable> funcs) { return funcs?.Select(Map).Where(x => x != null).ToArray(); } private string Map([CanBeNull] Func func) { return func?.ToString(); } private IMatcher Map([CanBeNull] MatcherModel matcher) { if (matcher == null) return null; switch (matcher.Name) { case "RegexMatcher": return new RegexMatcher(matcher.Pattern); case "JsonPathMatcher": return new JsonPathMatcher(matcher.Pattern); case "XPathMatcher": return new XPathMatcher(matcher.Pattern); default: return new WildcardMatcher(matcher.Pattern, matcher.IgnoreCase == true); } } private ResponseMessage ToJson(T result) { return new ResponseMessage { Body = JsonConvert.SerializeObject(result, _settings), StatusCode = 200, Headers = new Dictionary { { "Content-Type", "application/json" } } }; } } }