summaryrefslogtreecommitdiffstats
path: root/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
blob: 7117bc1bb23091a6f89cfc610518547ccf8c7ee7 (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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
//-----------------------------------------------------------------------
// <copyright file="TestWebRequestHandler.cs" company="Andrew Arnott">
//     Copyright (c) Andrew Arnott. All rights reserved.
// </copyright>
//-----------------------------------------------------------------------

namespace DotNetOpenAuth.Test.Mocks {
	using System;
	using System.IO;
	using System.Net;
	using System.Text;
	using DotNetOpenAuth.Messaging;
	using DotNetOpenAuth.OAuth.ChannelElements;

	internal class TestWebRequestHandler : IWebRequestHandler {
		private StringBuilder postEntity;

		/// <summary>
		/// Gets or sets the callback used to provide the mock response for the mock request.
		/// </summary>
		internal Func<HttpWebRequest, Response> Callback { get; set; }

		/// <summary>
		/// Gets the stream that was written out as if on an HTTP request.
		/// </summary>
		internal Stream RequestEntityStream {
			get {
				if (this.postEntity == null) {
					return null;
				}
				return new MemoryStream(Encoding.UTF8.GetBytes(this.postEntity.ToString()));
			}
		}

		/// <summary>
		/// Gets the stream that was written out as if on an HTTP request as an ordinary string.
		/// </summary>
		internal string RequestEntityAsString {
			get {
				return this.postEntity != null ? this.postEntity.ToString() : null;
			}
		}

		#region IWebRequestHandler Members

		/// <summary>
		/// Prepares an <see cref="HttpWebRequest"/> that contains an POST entity for sending the entity.
		/// </summary>
		/// <param name="request">The <see cref="HttpWebRequest"/> that should contain the entity.</param>
		/// <returns>
		/// The writer the caller should write out the entity data to.
		/// </returns>
		public TextWriter GetRequestStream(HttpWebRequest request) {
			this.postEntity = new StringBuilder();
			return new StringWriter(this.postEntity);
		}

		/// <summary>
		/// Processes an <see cref="HttpWebRequest"/> and converts the
		/// <see cref="HttpWebResponse"/> to a <see cref="Response"/> instance.
		/// </summary>
		/// <param name="request">The <see cref="HttpWebRequest"/> to handle.</param>
		/// <returns>
		/// An instance of <see cref="Response"/> describing the response.
		/// </returns>
		public Response GetResponse(HttpWebRequest request) {
			if (this.Callback == null) {
				throw new InvalidOperationException("Set the Callback property first.");
			}

			return this.Callback(request);
		}

		#endregion
	}
}