summaryrefslogtreecommitdiffstats
path: root/src/DotNetOpenAuth.Test/CookieContainerExtensions.cs
blob: 5a09d13fcf1df0cb77cca7a8303cda7e3f38b2e9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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);
			}
		}
	}
}