Auto create storage container (Azure#17607)

Author: pakrym

sdk/eventhub/Microsoft.Azure.WebJobs.Extensions.EventHubs/src/BlobsCheckpointStore.cs

@@ -2,6 +2,8 @@
 // Licensed under the MIT License. See License.txt in the project root for license information.
 
 using System;
+using System.Threading;
+using System.Threading.Tasks;
 using Azure.Storage.Blobs;
 using Microsoft.Extensions.Logging;
 
@@ -11,6 +13,7 @@ internal partial class BlobsCheckpointStore
     {
         private readonly string _functionId;
         private readonly ILogger _logger;
+        private BlobContainerClient _client;
 
         /// <summary>
         /// The mocking constructor.
@@ -35,6 +38,7 @@ public BlobsCheckpointStore(BlobContainerClient blobContainerClient,
         {
             _functionId = functionId;
             _logger = logger;
+            _client = blobContainerClient;
         }
 
         /// <summary>
@@ -68,5 +72,15 @@ partial void ListCheckpointsError(string fullyQualifiedNamespace, string eventHu
                 "Function '{functionId}': An exception occurred when listing checkpoints for FullyQualifiedNamespace: '{fullyQualifiedNamespace}'; EventHubName: '{eventHubName}'; ConsumerGroup: '{consumerGroup}'.",
                 _functionId, fullyQualifiedNamespace, eventHubName, consumerGroup);
         }
+
+        /// <summary>
+        /// Attempts to create a storage container if one doesn't exists.
+        /// </summary>
+        /// <param name="cancellationToken">A <see cref="CancellationToken" /> instance to signal the request to cancel the operation.</param>
+        /// <returns>A task to be resolved on when the operation has completed.</returns>
+        public async Task CreateIfNotExistsAsync(CancellationToken cancellationToken)
+        {
+            await _client.CreateIfNotExistsAsync(cancellationToken: cancellationToken).ConfigureAwait(false);
+        }
     }
 }

sdk/eventhub/Microsoft.Azure.WebJobs.Extensions.EventHubs/src/Listeners/EventHubListener.cs

@@ -71,6 +71,7 @@ void IDisposable.Dispose()
 
         public async Task StartAsync(CancellationToken cancellationToken)
         {
+            await _checkpointStore.CreateIfNotExistsAsync(cancellationToken).ConfigureAwait(false);
             await _eventProcessorHost.StartProcessingAsync(this, _checkpointStore, cancellationToken).ConfigureAwait(false);
         }
 

sdk/eventhub/Microsoft.Azure.WebJobs.Extensions.EventHubs/tests/WebJobsEventHubTestBase.cs

@@ -25,9 +25,6 @@ public class WebJobsEventHubTestBase
         /// <summary>The active Event Hub resource scope for the test fixture.</summary>
         protected EventHubScope _eventHubScope;
 
-        /// <summary>The active Blob storage resource scope for the test fixture.</summary>
-        protected StorageScope _storageScope;
-
         /// <summary>
         ///   Performs the tasks needed to initialize the test fixture.  This
         ///   method runs once for the entire fixture, prior to running any tests.
@@ -37,7 +34,6 @@ public class WebJobsEventHubTestBase
         public async Task FixtureSetUp()
         {
             _eventHubScope = await EventHubScope.CreateAsync(2);
-            _storageScope = await StorageScope.CreateAsync();
             _testId = Guid.NewGuid().ToString();
         }
 
@@ -49,11 +45,7 @@ public async Task FixtureSetUp()
         [TearDown]
         public async Task FixtureTearDown()
         {
-            await Task.WhenAll
-            (
-                _eventHubScope.DisposeAsync().AsTask(),
-                _storageScope.DisposeAsync().AsTask()
-            );
+            await _eventHubScope.DisposeAsync();
         }
 
         protected void ConfigureTestEventHub(IHostBuilder builder)
@@ -85,7 +77,7 @@ protected void ConfigureTestEventHub(IHostBuilder builder)
                 {
                     services.Configure<EventHubOptions>(options =>
                     {
-                        options.CheckpointContainer = _storageScope.ContainerName;
+                        options.CheckpointContainer = Guid.NewGuid().ToString("D").Substring(0, 13);
                         // Speedup shutdown
                         options.EventProcessorOptions.MaximumWaitTime = TimeSpan.FromSeconds(5);
                     });