Improved precision of calculating which message type to instantiate.

1 files changed, 14 insertions, 1 deletions

diff --git a/src/DotNetOpenAuth.Test/OAuthWrap/OAuthWrapChannelTests.cs b/src/DotNetOpenAuth.Test/OAuthWrap/OAuthWrapChannelTests.cs
index 1f76e8f..433035c 100644
--- a/src/DotNetOpenAuth.Test/OAuthWrap/OAuthWrapChannelTests.cs
+++ b/src/DotNetOpenAuth.Test/OAuthWrap/OAuthWrapChannelTests.cs
@@ -10,17 +10,22 @@ namespace DotNetOpenAuth.Test.OAuthWrap {
 	using System.Linq;
 	using System.Text;
 	using DotNetOpenAuth.Messaging;
+	using DotNetOpenAuth.OAuthWrap;
 	using DotNetOpenAuth.OAuthWrap.ChannelElements;
+	using DotNetOpenAuth.OAuthWrap.Messages;
 	using NUnit.Framework;
 
 	[TestFixture]
 	public class OAuthWrapChannelTests : OAuthWrapTestBase {
 		private OAuthWrapChannel channel;
+		private IMessageFactory messageFactory;
+		private MessageReceivingEndpoint recipient = new MessageReceivingEndpoint("http://who", HttpDeliveryMethods.PostRequest);
 
 		public override void SetUp() {
 			base.SetUp();
 
 			this.channel = new OAuthWrapChannel();
+			this.messageFactory = OAuthWrapChannel_Accessor.AttachShadow(this.channel).MessageFactory;
 		}
 
 		/// <summary>
@@ -28,7 +33,15 @@ namespace DotNetOpenAuth.Test.OAuthWrap {
 		/// </summary>
 		[TestCase]
 		public void MessageFactory() {
-			// TODO: code here
+			var fields = new Dictionary<string, string> {
+				{ Protocol.wrap_refresh_token, "abc" },
+			};
+			IDirectedProtocolMessage request = messageFactory.GetNewRequestMessage(recipient, fields);
+			Assert.IsInstanceOf(typeof(RefreshAccessTokenRequest), request);
+
+			fields.Clear();
+			fields[Protocol.wrap_access_token] = "abc";
+			Assert.IsInstanceOf(typeof(RefreshAccessTokenSuccessResponse), messageFactory.GetNewResponseMessage(request, fields));
 		}
 	}
 }