public async Task CompleteAsyncOrDispose_CleansUpAfterTransfer() { var clientMock = new DocumentDbWriteClientMock(); var configurationMock = Mocks .Of<IDocumentDbBulkSinkAdapterInstanceConfiguration>(m => m.Collection == "TestCollection" && m.StoredProcName == "test" && m.BatchSize == 5 && m.MaxScriptSize == 1024) .First(); using (var adapter = new DocumentDbBulkSinkAdapter(clientMock, PassThroughTransformation.Instance, configurationMock)) { await adapter.InitializeAsync(); await adapter.CompleteAsync(CancellationToken.None); } CollectionAssert.AreEquivalent(clientMock.CreatedStoredProcedures.ToArray(), clientMock.DeletedStoredProcedures.ToArray(), TestResources.BulkSinkAdapterDidNotCleanup); }
public async Task WriteSampleData_AllDataStored() { const string CollectionName = "TestCollection"; const int NumberOfItems = 10; var clientMock = new DocumentDbWriteClientMock(); var configuration = Mocks .Of<IDocumentDbParallelSinkAdapterInstanceConfiguration>() .Where(m => m.Collections == new[] { CollectionName } && m.NumberOfParallelRequests == 2) .First(); using (var adapter = new DocumentDbParallelSinkAdapter(clientMock, PassThroughTransformation.Instance, configuration)) { await adapter.InitializeAsync(); await WriteDataAsync(adapter, SampleData.GetSimpleDataItems(NumberOfItems)); } CollectionAssert.AreEquivalent(new [] { CollectionName }, clientMock.CreatedCollections.ToArray(), TestResources.CollectionWasNotCreated); Assert.AreEqual(NumberOfItems, clientMock.NumberOfDocumentsCreated, TestResources.InvalidNumberOfDataItemsTransferred); Assert.AreEqual(0, clientMock.CreatedStoredProcedures.Count, TestResources.ParallelSinkAdapterCreatedStoredProcedure); }