public async Task ReleaseCompletes()
{
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
DocumentServiceLeaseCore lease = new DocumentServiceLeaseCore()
{
LeaseId = Guid.NewGuid().ToString(),
LeaseToken = "0",
Owner = Guid.NewGuid().ToString(),
FeedRange = new FeedRangePartitionKeyRange("0")
};
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
Func <Func <DocumentServiceLease, DocumentServiceLease>, bool> validateUpdater = (Func <DocumentServiceLease, DocumentServiceLease> updater) =>
{
DocumentServiceLease afterUpdateLease = updater(lease);
return(afterUpdateLease.Owner == null);
};
mockUpdater.Setup(c => c.UpdateLeaseAsync(
It.IsAny <DocumentServiceLease>(),
It.IsAny <string>(),
It.IsAny <PartitionKey>(),
It.Is <Func <DocumentServiceLease, DocumentServiceLease> >(f => validateUpdater(f))))
.ReturnsAsync(lease);
ResponseMessage leaseResponse = new ResponseMessage(System.Net.HttpStatusCode.OK)
{
Content = new CosmosJsonDotNetSerializer().ToStream(lease)
};
Mock <ContainerInternal> mockedContainer = new Mock <ContainerInternal>();
mockedContainer.Setup(c => c.ReadItemStreamAsync(
It.Is <string>(id => id == lease.LeaseId),
It.IsAny <PartitionKey>(),
It.IsAny <ItemRequestOptions>(),
It.IsAny <CancellationToken>())).ReturnsAsync(leaseResponse);
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
Mock.Of <ContainerInternal>(),
mockedContainer.Object,
mockUpdater.Object,
options,
Mock.Of <RequestOptionsFactory>());
await documentServiceLeaseManagerCosmos.ReleaseAsync(lease);
}
public async Task PopulateMissingRange()
{
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
DocumentServiceLeaseCore lease = new DocumentServiceLeaseCore()
{
LeaseToken = "0",
Owner = Guid.NewGuid().ToString()
};
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
Func <Func <DocumentServiceLease, DocumentServiceLease>, bool> validateUpdater = (Func <DocumentServiceLease, DocumentServiceLease> updater) =>
{
// Simulate dirty read from db
DocumentServiceLease afterUpdateLease = updater(lease);
return(afterUpdateLease.FeedRange != null);
};
mockUpdater.Setup(c => c.UpdateLeaseAsync(
It.IsAny <DocumentServiceLease>(),
It.IsAny <string>(),
It.IsAny <PartitionKey>(),
It.Is <Func <DocumentServiceLease, DocumentServiceLease> >(f => validateUpdater(f))))
.ReturnsAsync(lease);
Mock <ContainerInternal> containerMock = new Mock <ContainerInternal>();
Mock <CosmosClientContext> mockContext = new Mock <CosmosClientContext>();
containerMock.Setup(c => c.ClientContext).Returns(mockContext.Object);
containerMock.Setup(c => c.LinkUri).Returns("http://localhot");
containerMock.Setup(c => c.GetCachedRIDAsync(It.IsAny <bool>(), It.IsAny <CancellationToken>())).ReturnsAsync("test");
MockDocumentClient mockDocumentClient = new MockDocumentClient();
mockContext.Setup(c => c.DocumentClient).Returns(mockDocumentClient);
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
containerMock.Object,
Mock.Of <ContainerInternal>(),
mockUpdater.Object,
options,
Mock.Of <RequestOptionsFactory>());
DocumentServiceLease afterAcquire = await documentServiceLeaseManagerCosmos.AcquireAsync(lease);
Assert.IsNotNull(afterAcquire.FeedRange);
}
public async Task ReleaseWhen404WithSomeSubstatusDoesThrow()
{
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
DocumentServiceLeaseCore lease = new DocumentServiceLeaseCore()
{
LeaseId = Guid.NewGuid().ToString(),
LeaseToken = "0",
Owner = Guid.NewGuid().ToString(),
FeedRange = new FeedRangePartitionKeyRange("0")
};
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
ResponseMessage leaseResponse = new ResponseMessage(System.Net.HttpStatusCode.NotFound);
leaseResponse.Headers.SubStatusCode = Documents.SubStatusCodes.ReadSessionNotAvailable;
Mock <ContainerInternal> mockedContainer = new Mock <ContainerInternal>();
mockedContainer.Setup(c => c.ReadItemStreamAsync(
It.Is <string>(id => id == lease.LeaseId),
It.IsAny <PartitionKey>(),
It.IsAny <ItemRequestOptions>(),
It.IsAny <CancellationToken>())).ReturnsAsync(leaseResponse);
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
Mock.Of <ContainerInternal>(),
mockedContainer.Object,
mockUpdater.Object,
options,
Mock.Of <RequestOptionsFactory>());
CosmosException cosmosException = await Assert.ThrowsExceptionAsync <CosmosException>(() => documentServiceLeaseManagerCosmos.ReleaseAsync(lease));
Assert.AreEqual(System.Net.HttpStatusCode.NotFound, cosmosException.StatusCode);
Assert.AreEqual((int)Documents.SubStatusCodes.ReadSessionNotAvailable, cosmosException.SubStatusCode);
mockUpdater.Verify(c => c.UpdateLeaseAsync(
It.IsAny <DocumentServiceLease>(),
It.IsAny <string>(),
It.IsAny <PartitionKey>(),
It.IsAny <Func <DocumentServiceLease, DocumentServiceLease> >()), Times.Never);
}
public async Task CreatesPartitionKeyBasedLease(int factoryType)
{
RequestOptionsFactory requestOptionsFactory = GetRequestOptionsFactory(factoryType);
string continuation = Guid.NewGuid().ToString();
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
Documents.PartitionKeyRange partitionKeyRange = new Documents.PartitionKeyRange()
{
Id = "0",
MinInclusive = "",
MaxExclusive = "FF"
};
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
Mock <ContainerInternal> mockedContainer = new Mock <ContainerInternal>();
mockedContainer.Setup(c => c.CreateItemStreamAsync(
It.IsAny <Stream>(),
It.IsAny <PartitionKey>(),
It.IsAny <ItemRequestOptions>(),
It.IsAny <CancellationToken>())).ReturnsAsync((Stream stream, PartitionKey partitionKey, ItemRequestOptions options, CancellationToken token) => new ResponseMessage(System.Net.HttpStatusCode.OK)
{
Content = stream
});
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
Mock.Of <ContainerInternal>(),
mockedContainer.Object,
mockUpdater.Object,
options,
requestOptionsFactory);
DocumentServiceLease afterAcquire = await documentServiceLeaseManagerCosmos.CreateLeaseIfNotExistAsync(partitionKeyRange, continuation);
DocumentServiceLeaseCore pkRangeBasedLease = (DocumentServiceLeaseCore)afterAcquire;
Assert.IsNotNull(pkRangeBasedLease);
Assert.AreEqual(continuation, afterAcquire.ContinuationToken);
Assert.AreEqual(partitionKeyRange.Id, pkRangeBasedLease.CurrentLeaseToken);
ValidateRequestOptionsFactory(requestOptionsFactory, pkRangeBasedLease);
}
public async Task ReleaseWhenNotExistDoesNotThrow()
{
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
DocumentServiceLeaseCore lease = new DocumentServiceLeaseCore()
{
LeaseId = Guid.NewGuid().ToString(),
LeaseToken = "0",
Owner = Guid.NewGuid().ToString(),
FeedRange = new FeedRangePartitionKeyRange("0")
};
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
ResponseMessage leaseResponse = new ResponseMessage(System.Net.HttpStatusCode.NotFound);
Mock <ContainerInternal> mockedContainer = new Mock <ContainerInternal>();
mockedContainer.Setup(c => c.ReadItemStreamAsync(
It.Is <string>(id => id == lease.LeaseId),
It.IsAny <PartitionKey>(),
It.IsAny <ItemRequestOptions>(),
It.IsAny <CancellationToken>())).ReturnsAsync(leaseResponse);
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
Mock.Of <ContainerInternal>(),
mockedContainer.Object,
mockUpdater.Object,
options,
Mock.Of <RequestOptionsFactory>());
await documentServiceLeaseManagerCosmos.ReleaseAsync(lease);
mockUpdater.Verify(c => c.UpdateLeaseAsync(
It.IsAny <DocumentServiceLease>(),
It.IsAny <string>(),
It.IsAny <PartitionKey>(),
It.IsAny <Func <DocumentServiceLease, DocumentServiceLease> >()), Times.Never);
}
public async Task CreatesEPKBasedLease(int factoryType)
{
RequestOptionsFactory requestOptionsFactory = GetRequestOptionsFactory(factoryType);
string continuation = Guid.NewGuid().ToString();
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
FeedRangeEpk feedRangeEpk = new FeedRangeEpk(new Documents.Routing.Range <string>("AA", "BB", true, false));
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
Mock <ContainerInternal> mockedContainer = new Mock <ContainerInternal>();
mockedContainer.Setup(c => c.CreateItemStreamAsync(
It.IsAny <Stream>(),
It.IsAny <PartitionKey>(),
It.IsAny <ItemRequestOptions>(),
It.IsAny <CancellationToken>())).ReturnsAsync((Stream stream, PartitionKey partitionKey, ItemRequestOptions options, CancellationToken token) => new ResponseMessage(System.Net.HttpStatusCode.OK)
{
Content = stream
});
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
Mock.Of <ContainerInternal>(),
mockedContainer.Object,
mockUpdater.Object,
options,
requestOptionsFactory);
DocumentServiceLease afterAcquire = await documentServiceLeaseManagerCosmos.CreateLeaseIfNotExistAsync(feedRangeEpk, continuation);
DocumentServiceLeaseCoreEpk epkBasedLease = (DocumentServiceLeaseCoreEpk)afterAcquire;
Assert.IsNotNull(epkBasedLease);
Assert.AreEqual(continuation, afterAcquire.ContinuationToken);
Assert.AreEqual(feedRangeEpk.Range.Min, ((FeedRangeEpk)epkBasedLease.FeedRange).Range.Min);
Assert.AreEqual(feedRangeEpk.Range.Max, ((FeedRangeEpk)epkBasedLease.FeedRange).Range.Max);
ValidateRequestOptionsFactory(requestOptionsFactory, epkBasedLease);
}
public async Task AcquireCompletes()
{
DocumentServiceLeaseStoreManagerOptions options = new DocumentServiceLeaseStoreManagerOptions
{
HostName = Guid.NewGuid().ToString()
};
DocumentServiceLeaseCore lease = new DocumentServiceLeaseCore()
{
LeaseToken = "0",
Owner = Guid.NewGuid().ToString(),
FeedRange = new FeedRangePartitionKeyRange("0")
};
Mock <DocumentServiceLeaseUpdater> mockUpdater = new Mock <DocumentServiceLeaseUpdater>();
Func <Func <DocumentServiceLease, DocumentServiceLease>, bool> validateUpdater = (Func <DocumentServiceLease, DocumentServiceLease> updater) =>
{
DocumentServiceLease afterUpdateLease = updater(lease);
return(options.HostName == afterUpdateLease.Owner);
};
mockUpdater.Setup(c => c.UpdateLeaseAsync(
It.IsAny <DocumentServiceLease>(),
It.IsAny <string>(),
It.IsAny <PartitionKey>(),
It.Is <Func <DocumentServiceLease, DocumentServiceLease> >(f => validateUpdater(f))))
.ReturnsAsync(lease);
DocumentServiceLeaseManagerCosmos documentServiceLeaseManagerCosmos = new DocumentServiceLeaseManagerCosmos(
Mock.Of <ContainerInternal>(),
Mock.Of <ContainerInternal>(),
mockUpdater.Object,
options,
Mock.Of <RequestOptionsFactory>());
DocumentServiceLease afterAcquire = await documentServiceLeaseManagerCosmos.AcquireAsync(lease);
Assert.AreEqual(options.HostName, afterAcquire.Owner);
}
