Adding AzureComponentFactoryWrapper (#2971)

test/Microsoft.Azure.WebJobs.Host.TestCommon/AzureComponentFactoryWrapper.cs

@@ -0,0 +1,33 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the MIT License. See License.txt in the project root for license information.
+
+using System;
+using Azure.Core;
+using Microsoft.Extensions.Azure;
+using Microsoft.Extensions.Configuration;
+
+namespace Microsoft.Azure.WebJobs.Host.TestCommon
+{
+    public class AzureComponentFactoryWrapper : AzureComponentFactory
+    {
+        private readonly AzureComponentFactory _factory;
+        private readonly TokenCredential _tokenCredential;
+
+        public AzureComponentFactoryWrapper(AzureComponentFactory factory, TokenCredential tokenCredential)
+        {
+            _factory = factory;
+            _tokenCredential = tokenCredential;
+        }
+
+        public override TokenCredential CreateTokenCredential(IConfiguration configuration)
+        {
+            return _tokenCredential != null ? _tokenCredential : _factory.CreateTokenCredential(configuration);
+        }
+
+        public override object CreateClientOptions(Type optionsType, object serviceVersion, IConfiguration configuration)
+            => _factory.CreateClientOptions(optionsType, serviceVersion, configuration);
+
+        public override object CreateClient(Type clientType, IConfiguration configuration, TokenCredential credential, object clientOptions)
+            => _factory.CreateClient(clientType, configuration, credential, clientOptions);
+    }
+}