public async Task AddAsync_Creates_IfTrue_AndNotFound(string partitionKeyPath, int collectionThroughput)
{
// Arrange
var mockService = new Mock <ICosmosDBService>(MockBehavior.Strict);
CosmosDBContext context = CosmosDBTestUtility.CreateContext(mockService.Object, partitionKeyPath: partitionKeyPath,
throughput: collectionThroughput, createIfNotExists: true);
var collector = new CosmosDBAsyncCollector <Item>(context);
mockService
.SetupSequence(m => m.UpsertDocumentAsync(It.IsAny <Uri>(), It.IsAny <object>()))
.Throws(CosmosDBTestUtility.CreateDocumentClientException(HttpStatusCode.NotFound))
.Returns(Task.FromResult(new Document()));
CosmosDBTestUtility.SetupDatabaseMock(mockService);
CosmosDBTestUtility.SetupCollectionMock(mockService, partitionKeyPath, collectionThroughput);
// Act
await collector.AddAsync(new Item { Text = "hello!" });
// Assert
mockService.VerifyAll();
// Verify that we upsert again after creation.
mockService.Verify(m => m.UpsertDocumentAsync(It.IsAny <Uri>(), It.IsAny <object>()), Times.Exactly(2));
}
public async Task AddAsync_CreatesDocument()
{
// Arrange
var mockDocDBService = new Mock <ICosmosDBService>(MockBehavior.Strict);
mockDocDBService
.Setup(m => m.UpsertDocumentAsync(It.IsAny <Uri>(), It.IsAny <object>()))
.ReturnsAsync(new Document());
var context = CosmosDBTestUtility.CreateContext(mockDocDBService.Object);
var collector = new CosmosDBAsyncCollector <Item>(context);
// Act
await collector.AddAsync(new Item { Text = "hello!" });
// Assert
mockDocDBService.VerifyAll();
}
public async Task AddAsync_DoesNotCreate_IfUpsertSucceeds()
{
// Arrange
var mockDocDBService = new Mock <ICosmosDBService>(MockBehavior.Strict);
var context = CosmosDBTestUtility.CreateContext(mockDocDBService.Object);
context.ResolvedAttribute.CreateIfNotExists = true;
var collector = new CosmosDBAsyncCollector <Item>(context);
mockDocDBService
.Setup(m => m.UpsertDocumentAsync(It.IsAny <Uri>(), It.IsAny <object>()))
.Returns(Task.FromResult(new Document()));
//// Act
await collector.AddAsync(new Item { Text = "hello!" });
// Assert
mockDocDBService.VerifyAll();
}
public async Task AddAsync_ThrowsWithCustomMessage_IfNotFound()
{
// Arrange
var mockDocDBService = new Mock <ICosmosDBService>(MockBehavior.Strict);
mockDocDBService
.Setup(m => m.UpsertDocumentAsync(It.IsAny <Uri>(), It.IsAny <object>()))
.ThrowsAsync(CosmosDBTestUtility.CreateDocumentClientException(HttpStatusCode.NotFound));
var context = CosmosDBTestUtility.CreateContext(mockDocDBService.Object, createIfNotExists: false);
var collector = new CosmosDBAsyncCollector <Item>(context);
// Act
var ex = await Assert.ThrowsAsync <InvalidOperationException>(() => collector.AddAsync(new Item {
Text = "hello!"
}));
// Assert
Assert.Contains(CosmosDBTestUtility.CollectionName, ex.Message);
Assert.Contains(CosmosDBTestUtility.DatabaseName, ex.Message);
mockDocDBService.VerifyAll();
}
