Cookie #9

README.md

@@ -57,6 +57,7 @@ WireMock supports matching of requests to stubs and verification queries using t
 * HTTP Method
 * Query parameters
 * Headers
+* Cookies
 * Request body
 
 

src/WireMock/HttpListenerRequestMapper.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.IO;
 using System.Linq;
 using System.Net;
@@ -23,8 +24,11 @@ public RequestMessage Map(HttpListenerRequest listenerRequest)
             string bodyAsString = body != null ? listenerRequest.ContentEncoding.GetString(body) : null;
             var listenerHeaders = listenerRequest.Headers;
             var headers = listenerHeaders.AllKeys.ToDictionary(k => k, k => listenerHeaders[k]);
+            var cookies = new Dictionary<string, string>();
+            foreach (Cookie cookie in listenerRequest.Cookies)
+                cookies.Add(cookie.Name, cookie.Value);
 
-            return new RequestMessage(url, verb, body, bodyAsString, headers);
+            return new RequestMessage(url, verb, body, bodyAsString, headers, cookies);
         }
 
         /// <summary>

src/WireMock/Matchers/Request/RequestMessageCookieMatcher.cs

@@ -0,0 +1,65 @@
+using System;
+using System.Collections.Generic;
+using JetBrains.Annotations;
+using WireMock.Validation;
+
+namespace WireMock.Matchers.Request
+{
+    /// <summary>
+    /// The request cookie matcher.
+    /// </summary>
+    public class RequestMessageCookieMatcher : IRequestMatcher
+    {
+        private readonly string _name;
+
+        private readonly IMatcher _matcher;
+
+        private readonly Func<IDictionary<string, string>, bool> _cookieFunc;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="RequestMessageCookieMatcher"/> class.
+        /// </summary>
+        /// <param name="name">The name.</param>
+        /// <param name="pattern">The pattern.</param>
+        /// <param name="ignoreCase">The ignoreCase.</param>
+        public RequestMessageCookieMatcher([NotNull] string name, [NotNull] string pattern, bool ignoreCase = true)
+        {
+            Check.NotNull(name, nameof(name));
+            Check.NotNull(pattern, nameof(pattern));
+
+            _name = name;
+            _matcher = new WildcardMatcher(pattern, ignoreCase);
+        }
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="RequestMessageCookieMatcher"/> class.
+        /// </summary>
+        /// <param name="func">
+        /// The func.
+        /// </param>
+        public RequestMessageCookieMatcher([NotNull] Func<IDictionary<string, string>, bool> func)
+        {
+            Check.NotNull(func, nameof(func));
+            _cookieFunc = func;
+        }
+
+        /// <summary>
+        /// Determines whether the specified RequestMessage is match.
+        /// </summary>
+        /// <param name="requestMessage">The RequestMessage.</param>
+        /// <returns>
+        ///   <c>true</c> if the specified RequestMessage is match; otherwise, <c>false</c>.
+        /// </returns>
+        public bool IsMatch(RequestMessage requestMessage)
+        {
+            if (_cookieFunc != null)
+                return _cookieFunc(requestMessage.Cookies);
+
+            if (requestMessage.Cookies == null)
+                return false;
+
+            string headerValue = requestMessage.Cookies[_name];
+            return _matcher.IsMatch(headerValue);
+        }
+    }
+}

src/WireMock/Matchers/Request/RequestMessageHeaderMatcher.cs

@@ -64,6 +64,9 @@ public bool IsMatch(RequestMessage requestMessage)
             if (_headerFunc != null)
                 return _headerFunc(requestMessage.Headers);
 
+            if (requestMessage.Headers == null)
+                return false;
+
             string headerValue = requestMessage.Headers[_name];
             return _matcher.IsMatch(headerValue);
         }

src/WireMock/Matchers/WildcardMatcher.cs

@@ -40,12 +40,12 @@ public bool IsMatch(string input)
         /// <summary>
         /// Copy/paste from http://www.codeproject.com/Tips/57304/Use-wildcard-characters-and-to-compare-strings
         /// </summary>
-        private bool MatchWildcardString(string pattern, string input)
+        private bool MatchWildcardString([NotNull] string pattern, string input)
         {
             if (input != null && _ignoreCase)
                 input = input.ToLower();
 
-            if (pattern != null && _ignoreCase)
+            if (_ignoreCase)
                 pattern = pattern.ToLower();
 
             if (string.CompareOrdinal(pattern, input) == 0)

src/WireMock/RequestBuilders/IHeadersAndCookiesRequestBuilder.cs

@@ -0,0 +1,45 @@
+using System;
+using System.Collections.Generic;
+using JetBrains.Annotations;
+using WireMock.Matchers.Request;
+
+namespace WireMock.RequestBuilders
+{
+    /// <summary>
+    /// The HeadersAndCookieRequestBuilder interface.
+    /// </summary>
+    public interface IHeadersAndCookiesRequestBuilder : IBodyRequestBuilder, IRequestMatcher, IParamsRequestBuilder
+    {
+        /// <summary>
+        /// The with header.
+        /// </summary>
+        /// <param name="name">The name.</param>
+        /// <param name="pattern">The pattern.</param>
+        /// <param name="ignoreCase">ignore Case</param>
+        /// <returns>The <see cref="IHeadersAndCookiesRequestBuilder"/>.</returns>
+        IHeadersAndCookiesRequestBuilder WithHeader(string name, string pattern, bool ignoreCase = true);
+
+        /// <summary>
+        /// The with header.
+        /// </summary>
+        /// <param name="func">The headers func.</param>
+        /// <returns>The <see cref="IHeadersAndCookiesRequestBuilder"/>.</returns>
+        IHeadersAndCookiesRequestBuilder WithHeader([NotNull] params Func<IDictionary<string, string>, bool>[] func);
+
+        /// <summary>
+        /// The with header.
+        /// </summary>
+        /// <param name="name">The name.</param>
+        /// <param name="pattern">The pattern.</param>
+        /// <param name="ignoreCase">ignore Case</param>
+        /// <returns>The <see cref="IHeadersAndCookiesRequestBuilder"/>.</returns>
+        IHeadersAndCookiesRequestBuilder WithCookie(string name, string pattern, bool ignoreCase = true);
+
+        /// <summary>
+        /// The with header.
+        /// </summary>
+        /// <param name="cookieFunc">The func.</param>
+        /// <returns>The <see cref="IHeadersAndCookiesRequestBuilder"/>.</returns>
+        IHeadersAndCookiesRequestBuilder WithCookie([NotNull] params Func<IDictionary<string, string>, bool>[] cookieFunc);
+    }
+}

src/WireMock/RequestBuilders/IVerbRequestBuilder.cs

@@ -5,61 +5,61 @@ namespace WireMock.RequestBuilders
     /// <summary>
     /// The VerbRequestBuilder interface.
     /// </summary>
-    public interface IVerbRequestBuilder : IHeadersRequestBuilder
+    public interface IVerbRequestBuilder : IHeadersAndCookiesRequestBuilder
     {
         /// <summary>
         /// The using get.
         /// </summary>
         /// <returns>
-        /// The <see cref="IHeadersRequestBuilder"/>.
+        /// The <see cref="IHeadersAndCookiesRequestBuilder"/>.
         /// </returns>
-        IHeadersRequestBuilder UsingGet();
+        IHeadersAndCookiesRequestBuilder UsingGet();
 
         /// <summary>
         /// The using post.
         /// </summary>
         /// <returns>
-        /// The <see cref="IHeadersRequestBuilder"/>.
+        /// The <see cref="IHeadersAndCookiesRequestBuilder"/>.
         /// </returns>
-        IHeadersRequestBuilder UsingPost();
+        IHeadersAndCookiesRequestBuilder UsingPost();
 
         /// <summary>
         /// The using delete.
         /// </summary>
         /// <returns>
-        /// The <see cref="IHeadersRequestBuilder"/>.
+        /// The <see cref="IHeadersAndCookiesRequestBuilder"/>.
         /// </returns>
-        IHeadersRequestBuilder UsingDelete();
+        IHeadersAndCookiesRequestBuilder UsingDelete();
 
         /// <summary>
         /// The using put.
         /// </summary>
         /// <returns>
-        /// The <see cref="IHeadersRequestBuilder"/>.
+        /// The <see cref="IHeadersAndCookiesRequestBuilder"/>.
         /// </returns>
-        IHeadersRequestBuilder UsingPut();
+        IHeadersAndCookiesRequestBuilder UsingPut();
 
         /// <summary>
         /// The using head.
         /// </summary>
         /// <returns>
-        /// The <see cref="IHeadersRequestBuilder"/>.
+        /// The <see cref="IHeadersAndCookiesRequestBuilder"/>.
         /// </returns>
-        IHeadersRequestBuilder UsingHead();
+        IHeadersAndCookiesRequestBuilder UsingHead();
 
         /// <summary>
         /// The using any verb.
         /// </summary>
         /// <returns>
-        /// The <see cref="IHeadersRequestBuilder"/>.
+        /// The <see cref="IHeadersAndCookiesRequestBuilder"/>.
         /// </returns>
-        IHeadersRequestBuilder UsingAnyVerb();
+        IHeadersAndCookiesRequestBuilder UsingAnyVerb();
 
         /// <summary>
         /// The using verb.
         /// </summary>
         /// <param name="verbs">The verb.</param>
-        /// <returns>The <see cref="IHeadersRequestBuilder"/>.</returns>
-        IHeadersRequestBuilder UsingVerb([NotNull] params string[] verbs);
+        /// <returns>The <see cref="IHeadersAndCookiesRequestBuilder"/>.</returns>
+        IHeadersAndCookiesRequestBuilder UsingVerb([NotNull] params string[] verbs);
     }
 }