Fix failing tests.

github · grokys · Nov 8, 2017 · Oct 20, 2017 · Oct 23, 2017 · Oct 23, 2017
commit 9db05a73139d354bb8246739c03c92cf3ce04cba
diff --git a/src/UnitTests/GitHub.App/Models/RepositoryHostsTests.cs b/src/UnitTests/GitHub.App/Models/RepositoryHostsTests.cs
@@ -153,7 +153,7 @@ static IConnectionManager CreateConnectionManager(params HostAddress[] hostAddre
 
             result.Connections.Returns(connections);
 
-            result.LogIn(null, null, null).ReturnsForAnyArgs(x =>
+            Func<NSubstitute.Core.CallInfo, Task<IConnection>> login = async x =>
             {
                 var hostAddress = x.Arg<HostAddress>();
 
@@ -167,9 +167,17 @@ static IConnectionManager CreateConnectionManager(params HostAddress[] hostAddre
                     x.ArgAt<string>(1),
                     null,
                     null);
+
+                if (result.ConnectionCreated != null)
+                {
+                    await result.ConnectionCreated(connection);
+                }
+
                 connections.Add(connection);
                 return connection;
-            });
+            };
+
+            result.LogIn(null, null, null).ReturnsForAnyArgs(login);
 
             result.LogOut(null).ReturnsForAnyArgs(x =>
             {