public void BatchSizeTest(int provideCount, int tasks) { var sut = new BulkProviderClient("testindex", Mock.Of <IElasticsearchCustomClient>()); for (int i = 0; i < provideCount; i++) { var frameworkProvider = new ProviderDocument(ElasticsearchDocumentTypes.PROVIDER_FRAMEWORK_DOCUMENT); sut.Index <ProviderDocument>(c => c.Document(frameworkProvider)); } sut.GetTasks().Count.Should().Be(tasks); }
public void ShouldCallClient(int provideCount, int callCount) { var mockElasticCustomClient = new Mock <ElasticsearchCustomClient>(Mock.Of <IElasticsearchClientFactory>(), Mock.Of <ILog>()); var sut = new BulkProviderClient("testindex", mockElasticCustomClient.Object); for (int i = 0; i < provideCount; i++) { var frameworkProvider = new ProviderDocument(ElasticsearchDocumentTypes.PROVIDER_FRAMEWORK_DOCUMENT); sut.Index <ProviderDocument>(c => c.Document(frameworkProvider)); } sut.GetTasks(); mockElasticCustomClient.Verify(x => x.BulkAsync(It.IsAny <IBulkRequest>(), It.IsAny <string>()), Times.Exactly(callCount)); }