public async Task Should_execute_operations_with_same_partition_key_together() { var fakeContainer = new FakeContainer(); var fakeCosmosClient = new FakeCosmosClient(fakeContainer); var containerHolderHolderResolver = new ContainerHolderResolver(new FakeProvider(fakeCosmosClient), new ContainerInformation("fakeContainer", new PartitionKeyPath("/deep/down")), "fakeDatabase"); var storageSession = new StorageSession(containerHolderHolderResolver, new ContextBag(), true); var firstOperation = new FakeOperation { PartitionKey = new PartitionKey("PartitionKey1") }; storageSession.AddOperation(firstOperation); var secondOperation = new FakeOperation { PartitionKey = new PartitionKey("PartitionKey1") }; storageSession.AddOperation(secondOperation); await((ICompletableSynchronizedStorageSession)storageSession).CompleteAsync(); Assert.That(firstOperation.WasApplied, Is.True); Assert.That(secondOperation.WasApplied, Is.True); Assert.That(firstOperation.AppliedBatch, Is.EqualTo(secondOperation.AppliedBatch), "Operations with the same partition key must be in the same batch"); }
public void Should_execute_and_dispose_release_operations_with_different_partition_key_distinct_when_not_completed() { var fakeContainer = new FakeContainer(); var fakeCosmosClient = new FakeCosmosClient(fakeContainer); var containerHolderHolderResolver = new ContainerHolderResolver(new FakeProvider(fakeCosmosClient), new ContainerInformation("fakeContainer", new PartitionKeyPath("/deep/down")), "fakeDatabase"); var storageSession = new StorageSession(containerHolderHolderResolver, new ContextBag(), true); var firstReleaseOperation = new ReleaseLockOperation { PartitionKey = new PartitionKey("PartitionKey1") }; storageSession.AddOperation(firstReleaseOperation); var secondReleaseOperation = new ReleaseLockOperation { PartitionKey = new PartitionKey("PartitionKey2") }; storageSession.AddOperation(secondReleaseOperation); storageSession.Dispose(); Assert.That(firstReleaseOperation.WasApplied, Is.True); Assert.That(secondReleaseOperation.WasApplied, Is.True); Assert.That(firstReleaseOperation.WasDisposed, Is.True); Assert.That(secondReleaseOperation.WasDisposed, Is.True); Assert.That(firstReleaseOperation.AppliedBatch, Is.Not.EqualTo(secondReleaseOperation.AppliedBatch), "Release operations with the different partition keys must be in different batches."); }
public async Task Should_not_dispose_release_operations_when_operations_not_successful() { var fakeContainer = new FakeContainer(); var fakeCosmosClient = new FakeCosmosClient(fakeContainer); var containerHolderHolderResolver = new ContainerHolderResolver(new FakeProvider(fakeCosmosClient), new ContainerInformation("fakeContainer", new PartitionKeyPath("/deep/down")), "fakeDatabase"); var storageSession = new StorageSession(containerHolderHolderResolver, new ContextBag(), true); var operation = new ThrowOnApplyOperation { PartitionKey = new PartitionKey("PartitionKey1") }; storageSession.AddOperation(operation); var releaseOperation = new ReleaseLockOperation { PartitionKey = new PartitionKey("PartitionKey1") }; storageSession.AddOperation(releaseOperation); try { await((ICompletableSynchronizedStorageSession)storageSession).CompleteAsync(); } catch { // ignored } Assert.That(releaseOperation.WasApplied, Is.False); Assert.That(releaseOperation.WasDisposed, Is.False); }
public void Should_execute_and_dispose_release_operations_as_best_effort() { var fakeContainer = new FakeContainer { TransactionalBatchFactory = () => new ThrowsOnExecuteAsyncTransactionalBatch() }; var fakeCosmosClient = new FakeCosmosClient(fakeContainer); var containerHolderHolderResolver = new ContainerHolderResolver(new FakeProvider(fakeCosmosClient), new ContainerInformation("fakeContainer", new PartitionKeyPath("/deep/down")), "fakeDatabase"); var storageSession = new StorageSession(containerHolderHolderResolver, new ContextBag(), true); var firstReleaseOperation = new ReleaseLockOperation { PartitionKey = new PartitionKey("PartitionKey1") }; storageSession.AddOperation(firstReleaseOperation); var secondReleaseOperation = new ReleaseLockOperation { PartitionKey = new PartitionKey("PartitionKey2") }; storageSession.AddOperation(secondReleaseOperation); Assert.DoesNotThrow(() => storageSession.Dispose()); Assert.That(firstReleaseOperation.WasApplied, Is.True); Assert.That(secondReleaseOperation.WasApplied, Is.True); Assert.That(firstReleaseOperation.WasDisposed, Is.True); Assert.That(secondReleaseOperation.WasDisposed, Is.True); }