1 files changed, 38 insertions, 0 deletions
diff --git a/src/DotNetOpenAuth.Test/CookieContainerExtensions.cs b/src/DotNetOpenAuth.Test/CookieContainerExtensions.cs
new file mode 100644
index 0000000..5a09d13
--- /dev/null
+++ b/src/DotNetOpenAuth.Test/CookieContainerExtensions.cs
@@ -0,0 +1,38 @@
+//-----------------------------------------------------------------------
+// <copyright file="CookieContainerExtensions.cs" company="Andrew Arnott">
+//     Copyright (c) Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace DotNetOpenAuth.Test {
+	using System;
+	using System.Collections.Generic;
+	using System.Net;
+	using System.Net.Http;
+
+	using Validation;
+
+	internal static class CookieContainerExtensions {
+		internal static void SetCookies(this CookieContainer container, HttpResponseMessage response, Uri requestUri = null) {
+			Requires.NotNull(container, "container");
+			Requires.NotNull(response, "response");
+
+			IEnumerable<string> cookieHeaders;
+			if (response.Headers.TryGetValues("Set-Cookie", out cookieHeaders)) {
+				foreach (string cookie in cookieHeaders) {
+					container.SetCookies(requestUri ?? response.RequestMessage.RequestUri, cookie);
+				}
+			}
+		}
+
+		internal static void ApplyCookies(this CookieContainer container, HttpRequestMessage request) {
+			Requires.NotNull(container, "container");
+			Requires.NotNull(request, "request");
+
+			string cookieHeader = container.GetCookieHeader(request.RequestUri);
+			if (!string.IsNullOrEmpty(cookieHeader)) {
+				request.Headers.TryAddWithoutValidation("Cookie", cookieHeader);
+			}
+		}
+	}
+}
+\ No newline at end of file