summaryrefslogtreecommitdiffstats
path: root/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
blob: d7092b41fbfd0f040800af9be2a51eca1eb96ad9 (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
//-----------------------------------------------------------------------
// <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;

		internal Func<HttpWebRequest, Response> Callback { get; set; }

		internal Stream RequestEntityStream {
			get {
				if (this.postEntity == null) {
					return null;
				}
				return new MemoryStream(Encoding.UTF8.GetBytes(this.postEntity.ToString()));
			}
		}

		#region IWebRequestHandler Members

		public TextWriter GetRequestStream(HttpWebRequest request) {
			this.postEntity = new StringBuilder();
			return new StringWriter(this.postEntity);
		}

		public Response GetResponse(HttpWebRequest request) {
			if (this.Callback == null) {
				throw new InvalidOperationException("Set the Callback property first.");
			}

			return this.Callback(request);
		}

		#endregion
	}
}