Added message version check to tests. Now 2/4 associate tests fail as expected.

1 files changed, 5 insertions, 1 deletions

diff --git a/src/DotNetOpenAuth.Test/OpenId/AssociationHandshakeTests.cs b/src/DotNetOpenAuth.Test/OpenId/AssociationHandshakeTests.cs
index 24b54cf..eb0bccc 100644
--- a/src/DotNetOpenAuth.Test/OpenId/AssociationHandshakeTests.cs
+++ b/src/DotNetOpenAuth.Test/OpenId/AssociationHandshakeTests.cs
@@ -78,7 +78,8 @@ namespace DotNetOpenAuth.Test.OpenId {
 					op.SecuritySettings = this.ProviderSecuritySettings;
 					op.AutoRespond();
 				});
-			coordinator.IncomingMessageFilter = (message) => {
+			coordinator.IncomingMessageFilter = message => {
+				Assert.AreSame(opDescription.Protocol.Version, message.ProtocolVersion, "The message was for version {0} but was expected to be for {1}.", message.ProtocolVersion, opDescription.Protocol.Version);
 				var associateSuccess = message as AssociateSuccessfulResponse;
 				var associateFailed = message as AssociateUnsuccessfulResponse;
 				if (associateSuccess != null) {
@@ -88,6 +89,9 @@ namespace DotNetOpenAuth.Test.OpenId {
 					associateUnsuccessfulResponse = associateFailed;
 				}
 			};
+			coordinator.OutgoingMessageFilter = message => {
+				Assert.AreSame(opDescription.Protocol.Version, message.ProtocolVersion, "The message was for version {0} but was expected to be for {1}.", message.ProtocolVersion, opDescription.Protocol.Version);
+			};
 			coordinator.Run();
 
 			if (expectSuccess) {