public void CloseRespectsTheCancellationToken()
{
var client = new AmqpEventHubClient("my.eventhub.com", "somePath", Mock.Of <TokenCredential>(), new EventHubClientOptions(), Mock.Of <EventHubRetryPolicy>());
var cancellationSource = new CancellationTokenSource();
cancellationSource.Cancel();
Assert.That(async() => await client.CloseAsync(cancellationSource.Token), Throws.InstanceOf <TaskCanceledException>());
}
public async Task GetPartitionPropertiesAsyncValidatesClosed()
{
using var cancellationSource = new CancellationTokenSource();
var client = new AmqpEventHubClient("my.eventhub.com", "somePath", Mock.Of <TokenCredential>(), new EventHubClientOptions(), Mock.Of <EventHubRetryPolicy>());
await client.CloseAsync(cancellationSource.Token);
Assert.That(async() => await client.GetPartitionPropertiesAsync("Fred", cancellationSource.Token), Throws.InstanceOf <EventHubsObjectClosedException>());
}
public async Task CreateConsumerValidatesClosed()
{
using var cancellationSource = new CancellationTokenSource();
var mockRetryPolicy = Mock.Of <EventHubRetryPolicy>();
var client = new AmqpEventHubClient("my.eventhub.com", "somePath", Mock.Of <TokenCredential>(), new EventHubClientOptions(), mockRetryPolicy);
await client.CloseAsync(cancellationSource.Token);
Assert.That(() => client.CreateConsumer("group", "0", EventPosition.Earliest, new EventHubConsumerOptions(), mockRetryPolicy), Throws.InstanceOf <EventHubsObjectClosedException>());
}
public async Task CloseMarksTheClientAsClosed()
{
var client = new AmqpEventHubClient("my.eventhub.com", "somePath", Mock.Of <TokenCredential>(), new EventHubClientOptions(), Mock.Of <EventHubRetryPolicy>());
Assert.That(client.Closed, Is.False, "The client should not be closed on creation");
await client.CloseAsync(CancellationToken.None);
Assert.That(client.Closed, Is.True, "The client should be marked as closed after closing");
}
