1 files changed, 46 insertions, 0 deletions
diff --git a/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs b/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
new file mode 100644
index 0000000..d7092b4
--- /dev/null
+++ b/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
@@ -0,0 +1,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
+	}
+}