Fixing tests adding test

Author: StanleyGoldman

test/GitHub.App.UnitTests/Services/OAuthCallbackListenerTests.cs

@@ -5,6 +5,7 @@
 using GitHub.Extensions;
 using GitHub.Services;
 using NSubstitute;
+using NSubstitute.ExceptionExtensions;
 using Rothko;
 using NUnit.Framework;
 
@@ -32,6 +33,8 @@ public async Task ListenStopsHttpListenerAsync()
 
             await target.Listen("id1", CancellationToken.None);
 
+            target.RedirectLastContext(new Uri("http://github.com"));
+
             httpListener.Received(1).Stop();
         }
 
@@ -70,9 +73,39 @@ public async Task SuccessfulResponseClosesResponseAsync()
 
             await target.Listen("id1", CancellationToken.None);
 
+            target.RedirectLastContext(new Uri("http://github.com"));
+
             context.Response.Received(1).Close();
         }
 
+        [Test]
+        public async Task ContextFailureStopsListener()
+        {
+            var httpListener = Substitute.For<IHttpListener>();
+            httpListener
+                .When(x => x.Start())
+                .Do(_ => httpListener.IsListening.Returns(true));
+
+            var context = Substitute.For<IHttpListenerContext>();
+            context.Request.Url.Returns(new Uri($"https://localhost:42549?code=1234&state={"id1"}"));
+            httpListener.GetContext().Returns(context);
+            httpListener.GetContextAsync().Returns(context);
+
+            httpListener.GetContextAsync().Throws(new Exception());
+            var target = new OAuthCallbackListener(httpListener);
+
+            try
+            {
+                await target.Listen("id1", CancellationToken.None);
+            }
+            catch (Exception e)
+            {
+                
+            }
+
+            httpListener.Received(1).Stop();
+        }
+
         static IHttpListener CreateHttpListener(string id)
         {
             var result = Substitute.For<IHttpListener>();