Fixed mock CoordinatingChannel to send direct responses and indirect messages while more truly simulating the ordinary channel experience.

1 files changed, 6 insertions, 3 deletions

diff --git a/src/DotNetOpenAuth.Test/Mocks/CoordinatingChannel.cs b/src/DotNetOpenAuth.Test/Mocks/CoordinatingChannel.cs
index 2c10db9..6fe5248 100644
--- a/src/DotNetOpenAuth.Test/Mocks/CoordinatingChannel.cs
+++ b/src/DotNetOpenAuth.Test/Mocks/CoordinatingChannel.cs
@@ -44,6 +44,11 @@ namespace DotNetOpenAuth.Test.Mocks {
 			this.VerifyMessageAfterReceiving(CloneSerializedParts(message));
 		}
 
+		internal void PostMessage(IProtocolMessage message) {
+			this.incomingMessage = CloneSerializedParts(message);
+			this.incomingMessageSignal.Set();
+		}
+
 		protected internal override HttpRequestInfo GetRequestFromContext() {
 			return new HttpRequestInfo((IDirectedProtocolMessage)this.AwaitIncomingMessage());
 		}
@@ -62,9 +67,7 @@ namespace DotNetOpenAuth.Test.Mocks {
 
 		protected override UserAgentResponse SendDirectMessageResponse(IProtocolMessage response) {
 			this.ProcessMessageFilter(response, true);
-			this.RemoteChannel.incomingMessage = CloneSerializedParts(response);
-			this.RemoteChannel.incomingMessageSignal.Set();
-			return null;
+			return new CoordinatingUserAgentResponse(CloneSerializedParts(response), this.RemoteChannel);
 		}
 
 		protected override UserAgentResponse SendIndirectMessage(IDirectedProtocolMessage message) {