1 files changed, 17 insertions, 1 deletions
diff --git a/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs b/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
index af971a0..e655238 100644
--- a/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
+++ b/src/DotNetOpenAuth.Test/Mocks/TestWebRequestHandler.cs
@@ -12,7 +12,7 @@ namespace DotNetOpenAuth.Test.Mocks {
 	using DotNetOpenAuth.Messaging;
 	using DotNetOpenAuth.OAuth.ChannelElements;
 
-	internal class TestWebRequestHandler : IDirectWebRequestHandler {
+	internal class TestWebRequestHandler : IDirectSslWebRequestHandler {
 		private StringBuilder postEntity;
 
 		/// <summary>
@@ -72,5 +72,21 @@ namespace DotNetOpenAuth.Test.Mocks {
 		}
 
 		#endregion
+
+		#region IDirectSslWebRequestHandler Members
+
+		public TextWriter GetRequestStream(HttpWebRequest request, bool requireSsl) {
+			ErrorUtilities.VerifyProtocol(!requireSsl || request.RequestUri.Scheme == Uri.UriSchemeHttps, "disallowed request");
+			return this.GetRequestStream(request);
+		}
+
+		public DirectWebResponse GetResponse(HttpWebRequest request, bool requireSsl) {
+			ErrorUtilities.VerifyProtocol(!requireSsl || request.RequestUri.Scheme == Uri.UriSchemeHttps, "disallowed request");
+			var result = this.GetResponse(request);
+			ErrorUtilities.VerifyProtocol(!requireSsl || result.FinalUri.Scheme == Uri.UriSchemeHttps, "disallowed request");
+			return result;
+		}
+
+		#endregion
 	}
 }