Fixes two more tests. 2 left

1 files changed, 7 insertions, 3 deletions

diff --git a/src/DotNetOpenAuth.AspNet.Test/OAuthClientTest.cs b/src/DotNetOpenAuth.AspNet.Test/OAuthClientTest.cs
index 5016de2..25dd50b 100644
--- a/src/DotNetOpenAuth.AspNet.Test/OAuthClientTest.cs
+++ b/src/DotNetOpenAuth.AspNet.Test/OAuthClientTest.cs
@@ -6,6 +6,7 @@
 
 namespace DotNetOpenAuth.AspNet.Test {
 	using System;
+	using System.Collections.Specialized;
 	using System.Web;
 	using DotNetOpenAuth.AspNet;
 	using DotNetOpenAuth.AspNet.Clients;
@@ -34,13 +35,14 @@ namespace DotNetOpenAuth.AspNet.Test {
 		[TestCase]
 		public async Task RequestAuthenticationInvokeMethodOnWebWorker() {
 			// Arrange
+			var returnUri = new Uri("http://live.com/my/path.cshtml?q=one");
 			var webWorker = new Mock<IOAuthWebWorker>(MockBehavior.Strict);
 			webWorker
-				.Setup(w => w.RequestAuthenticationAsync(It.Is<Uri>(u => u.ToString().Equals("http://live.com/my/path.cshtml?q=one")), It.IsAny<CancellationToken>()))
+				.Setup(w => w.RequestAuthenticationAsync(returnUri, It.IsAny<CancellationToken>()))
+				.Returns(Task.FromResult(new Uri("http://someauth/uri")))
 				.Verifiable();
 
 			var client = new MockOAuthClient(webWorker.Object);
-			var returnUri = new Uri("http://live.com/my/path.cshtml?q=one");
 			var context = new Mock<HttpContextBase>();
 
 			// Act
@@ -94,7 +96,9 @@ namespace DotNetOpenAuth.AspNet.Test {
 			var request = new AuthorizedTokenRequest(endpoint, new Version("1.0"));
 
 			var webWorker = new Mock<IOAuthWebWorker>(MockBehavior.Strict);
-			webWorker.Setup(w => w.ProcessUserAuthorizationAsync(It.IsAny<HttpContextBase>(), CancellationToken.None)).Returns(Task.FromResult<AccessTokenResponse>(null)).Verifiable();
+			webWorker
+				.Setup(w => w.ProcessUserAuthorizationAsync(It.IsAny<HttpContextBase>(), CancellationToken.None))
+				.Returns(Task.FromResult(new AccessTokenResponse("ok", "secret", new NameValueCollection()))).Verifiable();
 
 			var client = new MockOAuthClient(webWorker.Object);
 			var context = new Mock<HttpContextBase>();