コード例 #1
0
        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);
        }
コード例 #2
0
        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));
        }