CSHARP-3027: Improve pinning test

vincentkam · vincentkam · commit 06700ca83faf · 2020-04-01T17:30:22.000-04:00
diff --git a/tests/MongoDB.Driver.Tests/PinnedShardRouterTests.cs b/tests/MongoDB.Driver.Tests/PinnedShardRouterTests.cs
@@ -176,7 +176,8 @@ private DisposableMongoClient CreateDisposableClient(EventCapturer eventCapturer
                 }, 
                 useMultipleShardRouters);
             var timeOut = TimeSpan.FromSeconds(60);
-            SpinWait.SpinUntil(() => client.Cluster.Description.Type != ClusterType.Unknown, timeOut).Should().BeTrue();
+            bool AllServersConnected() => client.Cluster.Description.Servers.All(s => s.State == ServerState.Connected);
+            SpinWait.SpinUntil(AllServersConnected, timeOut).Should().BeTrue();
             return client;
         }
 

Original file line number	Diff line number	Diff line change
`@@ -176,7 +176,8 @@ private DisposableMongoClient CreateDisposableClient(EventCapturer eventCapturer`
`176`	`176`	`},`
`177`	`177`	`useMultipleShardRouters);`
`178`	`178`	`var timeOut = TimeSpan.FromSeconds(60);`
`179`		`- SpinWait.SpinUntil(() => client.Cluster.Description.Type != ClusterType.Unknown, timeOut).Should().BeTrue();`
	`179`	`+ bool AllServersConnected() => client.Cluster.Description.Servers.All(s => s.State == ServerState.Connected);`
	`180`	`+ SpinWait.SpinUntil(AllServersConnected, timeOut).Should().BeTrue();`
`180`	`181`	`return client;`
`181`	`182`	`}`
`182`	`183`