private void AssertAuthenticationSucceeds( MongoClientSettings settings, bool async, bool speculativeAuthenticatationShouldSucceedIfPossible = true) { // If we don't use a DisposableClient, the second run of AuthenticationSucceedsWithMongoDB_X509_mechanism // will fail because the backing Cluster's connections will be associated with a dropped user using (var client = DriverTestConfiguration.CreateDisposableClient(settings)) { // The first command executed with the MongoClient triggers either the sync or async variation of the // MongoClient's IAuthenticator if (async) { _ = client.ListDatabaseNamesAsync().GetAwaiter().GetResult().ToList(); } else { _ = client.ListDatabaseNames().ToList(); } if (Feature.SpeculativeAuthentication.IsSupported(CoreTestConfiguration.ServerVersion) && speculativeAuthenticatationShouldSucceedIfPossible) { var cancellationToken = CancellationToken.None; var serverSelector = new ReadPreferenceServerSelector(settings.ReadPreference); var server = client.Cluster.SelectServer(serverSelector, cancellationToken); var channel = server.GetChannel(cancellationToken); var isMasterResult = channel.ConnectionDescription.IsMasterResult; isMasterResult.SpeculativeAuthenticate.Should().NotBeNull(); } } }
public void Should_call_custom_server_selector() { var eventCapturer = new EventCapturer() .Capture <ClusterSelectingServerEvent>() .Capture <ClusterSelectedServerEvent>(); var customServerSelector = new CustomServerSelector(); using (var client = DriverTestConfiguration.CreateDisposableClient( clientSettings => clientSettings.ClusterConfigurator = c => { c.ConfigureCluster(s => s.With(postServerSelector: customServerSelector)); c.Subscribe(eventCapturer); }, logger: null)) { var collection = client .GetDatabase(DriverTestConfiguration.DatabaseNamespace.DatabaseName) .GetCollection <BsonDocument>(DriverTestConfiguration.CollectionNamespace.CollectionName) .WithReadPreference(ReadPreference.Nearest); customServerSelector.CustomSelectorWasCalled = false; eventCapturer.Clear(); collection.CountDocuments(new BsonDocument()); customServerSelector.CustomSelectorWasCalled.Should().BeTrue(); eventCapturer.Next().Should().BeOfType <ClusterSelectingServerEvent>(); eventCapturer.Next().Should().BeOfType <ClusterSelectedServerEvent>(); } }
public void KillCursor_should_actually_work() { RequireServer.Check().Supports(Feature.KillCursorsCommand); var eventCapturer = new EventCapturer().Capture <CommandSucceededEvent>(x => x.CommandName.Equals("killCursors")); using (var client = DriverTestConfiguration.CreateDisposableClient(eventCapturer)) { IAsyncCursor <BsonDocument> cursor; var database = client.GetDatabase("test"); var collection = database.GetCollection <BsonDocument>(GetType().Name); var documents = new List <BsonDocument>(); for (int i = 0; i < 1000; i++) { documents.Add(new BsonDocument("x", i)); } collection.InsertMany(documents); cursor = collection.FindSync("{}"); cursor.MoveNext(); var cursorId = ((AsyncCursor <BsonDocument>)cursor)._cursorId(); cursorId.Should().NotBe(0); cursor.Dispose(); var desiredResult = BsonDocument.Parse($"{{ \"cursorsKilled\" : [{cursorId}], \"cursorsNotFound\" : [], " + $"\"cursorsAlive\" : [], \"cursorsUnknown\" : [], \"ok\" : 1.0 }}"); var result = ((CommandSucceededEvent)eventCapturer.Events[0]).Reply; result.IsSameOrEqualTo(desiredResult); } }
private DisposableMongoClient GetClient(Action <ClusterBuilder> clusterConfigurator) { return(DriverTestConfiguration.CreateDisposableClient((MongoClientSettings clientSettings) => { clientSettings.ClusterConfigurator = clusterConfigurator; clientSettings.RetryWrites = true; })); }
// private methods private DisposableMongoClient CreateDisposableClient(EventCapturer eventCapturer, ReadPreference readPreference) { return(DriverTestConfiguration.CreateDisposableClient((MongoClientSettings settings) => { settings.ClusterConfigurator = c => c.Subscribe(eventCapturer); settings.ReadPreference = readPreference; })); }
private DisposableMongoClient CreateDisposableClient(EventCapturer eventCapturer, bool useMultipleShardRouters) { return(DriverTestConfiguration.CreateDisposableClient((MongoClientSettings settings) => { settings.ClusterConfigurator = c => c.Subscribe(eventCapturer); }, useMultipleShardRouters)); }
private DisposableMongoClient GetClient(Action <ClusterBuilder> clusterConfigurator, bool retryWrites = true) { return(DriverTestConfiguration.CreateDisposableClient((MongoClientSettings clientSettings) => { clientSettings.ClusterConfigurator = clusterConfigurator; clientSettings.RetryWrites = retryWrites; }, logger: CreateLogger <DisposableMongoClient>())); }
private DisposableMongoClient CreateDisposableClient(EventCapturer capturedEvents) { return(DriverTestConfiguration.CreateDisposableClient( settings => { settings.RetryWrites = false; settings.ClusterConfigurator = c => { c.Subscribe(capturedEvents); }; })); }
private DisposableMongoClient CreateDisposableClient(EventCapturer capturedEvents) { return(DriverTestConfiguration.CreateDisposableClient( settings => { settings.HeartbeatInterval = TimeSpan.FromMilliseconds(5); // the default value for spec tests settings.RetryWrites = false; settings.ClusterConfigurator = c => { c.Subscribe(capturedEvents); }; })); }
private DisposableMongoClient CreateDisposableClient(EventCapturer eventCapturer, bool useMultipleShardRouters) { // Increase localThresholdMS and wait until all nodes are discovered to avoid false positives. var client = DriverTestConfiguration.CreateDisposableClient((MongoClientSettings settings) => { settings.ClusterConfigurator = c => c.Subscribe(eventCapturer); settings.LocalThreshold = TimeSpan.FromMilliseconds(1000); }, useMultipleShardRouters); var timeOut = TimeSpan.FromSeconds(60); SpinWait.SpinUntil(() => client.Cluster.Description.Type != ClusterType.Unknown, timeOut).Should().BeTrue(); return(client); }
private void AssertAuthenticationFails(MongoClientSettings settings, bool async) { using (var client = DriverTestConfiguration.CreateDisposableClient(settings)) { Exception exception; if (async) { exception = Record.Exception(() => client.ListDatabaseNamesAsync().GetAwaiter().GetResult().ToList()); } else { exception = Record.Exception(() => client.ListDatabaseNames().ToList()); } exception.Should().BeOfType <MongoAuthenticationException>(); } }
public void Connection_pool_should_not_be_cleared_when_replSetStepDown_and_GetMore([Values(false, true)] bool async) { RequireServer.Check().Supports(Feature.KeepConnectionPoolWhenReplSetStepDown).ClusterType(ClusterType.ReplicaSet); var eventCapturer = new EventCapturer().Capture <ConnectionPoolClearedEvent>(); using (var client = CreateDisposableClient(eventCapturer)) { var database = client.GetDatabase(_databaseName, new MongoDatabaseSettings { WriteConcern = WriteConcern.WMajority }); database.DropCollection(_databaseName); var collection = database.GetCollection <BsonDocument>(_collectionName, new MongoCollectionSettings { WriteConcern = WriteConcern.WMajority }); var adminDatabase = client.GetDatabase("admin").WithWriteConcern(WriteConcern.W1); collection.InsertMany( new[] { new BsonDocument("x", 1), new BsonDocument("x", 2), new BsonDocument("x", 3), new BsonDocument("x", 4), new BsonDocument("x", 5), }); eventCapturer.Clear(); var cursor = collection.FindSync(FilterDefinition <BsonDocument> .Empty, new FindOptions <BsonDocument> { BatchSize = 2 }); cursor.MoveNext(); foreach (var secondary in client.Cluster.Description.Servers.Where(c => c.Type == ServerType.ReplicaSetSecondary)) { RunOnSecondary(client, secondary.EndPoint, BsonDocument.Parse("{ replSetFreeze : 0 }")); } var replSetStepDownCommand = BsonDocument.Parse("{ replSetStepDown : 30, force : true }"); BsonDocument replSetStepDownResult; if (async) { replSetStepDownResult = adminDatabase.RunCommandAsync <BsonDocument>(replSetStepDownCommand).GetAwaiter().GetResult(); } else { replSetStepDownResult = adminDatabase.RunCommand <BsonDocument>(replSetStepDownCommand); } replSetStepDownResult.Should().NotBeNull(); replSetStepDownResult.GetValue("ok", false).ToBoolean().Should().BeTrue(); cursor.MoveNext(); eventCapturer.Events.Should().BeEmpty(); // it also means that no new PoolClearedEvent } void RunOnSecondary(IMongoClient primaryClient, EndPoint secondaryEndpoint, BsonDocument command) { var secondarySettings = primaryClient.Settings.Clone(); secondarySettings.ClusterConfigurator = null; #pragma warning disable CS0618 // Type or member is obsolete secondarySettings.ConnectionMode = ConnectionMode.Direct; #pragma warning restore CS0618 // Type or member is obsolete var secondaryDnsEndpoint = (DnsEndPoint)secondaryEndpoint; secondarySettings.Server = new MongoServerAddress(secondaryDnsEndpoint.Host, secondaryDnsEndpoint.Port); using (var secondaryClient = DriverTestConfiguration.CreateDisposableClient(secondarySettings)) { var adminDatabase = secondaryClient.GetDatabase(DatabaseNamespace.Admin.DatabaseName); adminDatabase.RunCommand <BsonDocument>(command); } } }
private DisposableMongoClient GetClient(EventCapturer capturer) { return(DriverTestConfiguration.CreateDisposableClient(capturer)); }