FaultInjection: Adds method to add FaultInjection using CosmosClientBuilder

Microsoft.Azure.Cosmos/FaultInjection/src/FaultInjector.cs

@@ -7,7 +7,7 @@ namespace Microsoft.Azure.Cosmos.FaultInjection
     using System.Collections.Generic;
     using Microsoft.Azure.Documents.FaultInjection;
 
-    public class FaultInjector
+    public class FaultInjector : IFaultInjector
     {
         private readonly ChaosInterceptorFactory chaosInterceptorFactory;
 
@@ -48,5 +48,10 @@ internal IChaosInterceptorFactory GetChaosInterceptorFactory()
         {
             return this.chaosInterceptorFactory;
         }
+
+        IChaosInterceptorFactory IFaultInjector.GetChaosInterceptorFactory()
+        {
+            return this.chaosInterceptorFactory;
+        }
     }
 }

Microsoft.Azure.Cosmos/src/CosmosClientOptions.cs

@@ -12,7 +12,8 @@ namespace Microsoft.Azure.Cosmos
     using System.Net;
     using System.Net.Http;
     using System.Net.Security;
-    using System.Security.Cryptography.X509Certificates;
+    using System.Security.Cryptography.X509Certificates;
+    using Microsoft.Azure.Cosmos.FaultInjection;
     using Microsoft.Azure.Cosmos.Fluent;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Client;
@@ -72,7 +73,8 @@ public class CosmosClientOptions
         private PortReuseMode? portReuseMode;
         private IWebProxy webProxy;
         private Func<HttpClient> httpClientFactory;
-        private string applicationName;
+        private string applicationName;
+        private IFaultInjector faultInjector;
 
         /// <summary>
         /// Creates a new CosmosClientOptions
@@ -898,6 +900,22 @@ internal Func<X509Certificate2, X509Chain, SslPolicyErrors, bool> GetServerCerti
         /// </summary>
         public CosmosClientTelemetryOptions CosmosClientTelemetryOptions { get; set; }
 
+        /// <summary>
+        /// Create a client with Fault Injection capabilities.
+        /// </summary>
+        public IFaultInjector FaultInjector
+        {
+            get => this.faultInjector;
+            set
+            {
+                this.faultInjector = value;
+                if (this.faultInjector != null)
+                {
+                    this.ChaosInterceptorFactory = this.faultInjector.GetChaosInterceptorFactory();
+                }
+            }
+        }
+
         internal IChaosInterceptorFactory ChaosInterceptorFactory { get; set; }
 
         internal void SetSerializerIfNotConfigured(CosmosSerializer serializer)

Microsoft.Azure.Cosmos/src/FaultInjection/IFaultInjector.cs

@@ -0,0 +1,16 @@
+//------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+//------------------------------------------------------------
+namespace Microsoft.Azure.Cosmos.FaultInjection
+{
+    /// <summary>  
+    /// Interface for injecting faults into the Cosmos DB client operations.  
+    /// </summary>  
+    public interface IFaultInjector
+    {
+        /// <summary>  
+        /// Gets the chaosInterceptorFactory for client building  
+        /// </summary>  
+        internal IChaosInterceptorFactory GetChaosInterceptorFactory();
+    }
+}

Microsoft.Azure.Cosmos/src/Fluent/CosmosClientBuilder.cs

@@ -13,6 +13,7 @@ namespace Microsoft.Azure.Cosmos.Fluent
     using global::Azure;
     using global::Azure.Core;
     using Microsoft.Azure.Cosmos.Core.Trace;
+    using Microsoft.Azure.Cosmos.FaultInjection;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Client;
 
@@ -755,6 +756,17 @@ internal CosmosClientBuilder WithFaultInjection(IChaosInterceptorFactory chaosIn
         {
             this.clientOptions.ChaosInterceptorFactory = chaosInterceptorFactory;
             return this;
+        }
+
+        /// <summary>
+        /// Enables SDK to inject fault. Used for testing applications.  
+        /// </summary>
+        /// <param name="faultInjector"></param>
+        /// <returns>>The <see cref="CosmosClientBuilder"/> object</returns>
+        public CosmosClientBuilder WithFaultInjection(IFaultInjector faultInjector)
+        {
+            this.clientOptions.ChaosInterceptorFactory = faultInjector.GetChaosInterceptorFactory();
+            return this;
         }
 
         /// <summary>

Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/Contracts/ContractTests.cs

@@ -37,9 +37,9 @@ public void ClientDllNamespaceTest()
         {
 
 #if INTERNAL
-            int expected = 7;
+            int expected = 8;
 #else
-            int expected = 5;
+            int expected = 6;
 #endif
             ContractTests.NamespaceCountTest(typeof(CosmosClient), expected);
         }

Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/Contracts/DotNetSDKAPI.json

@@ -2802,6 +2802,16 @@
           ],
           "MethodInfo": "Microsoft.Azure.Cosmos.CosmosSerializer Serializer;CanRead:True;CanWrite:True;Microsoft.Azure.Cosmos.CosmosSerializer get_Serializer();IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;Void set_Serializer(Microsoft.Azure.Cosmos.CosmosSerializer);IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
         },
+        "Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector FaultInjector": {
+          "Type": "Property",
+          "Attributes": [],
+          "MethodInfo": "Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector FaultInjector;CanRead:True;CanWrite:True;Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector get_FaultInjector();IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;Void set_FaultInjector(Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector);IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
+        },
+        "Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector get_FaultInjector()": {
+          "Type": "Method",
+          "Attributes": [],
+          "MethodInfo": "Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector get_FaultInjector();IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
+        },
         "System.Collections.Generic.IEnumerable`1[System.Uri] AccountInitializationCustomEndpoints": {
           "Type": "Property",
           "Attributes": [],
@@ -3113,6 +3123,11 @@
           ],
           "MethodInfo": "Void set_EnableTcpConnectionEndpointRediscovery(Boolean);IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
         },
+        "Void set_FaultInjector(Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector)": {
+          "Type": "Method",
+          "Attributes": [],
+          "MethodInfo": "Void set_FaultInjector(Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector);IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
+        },
         "Void set_GatewayModeMaxConnectionLimit(Int32)": {
           "Type": "Method",
           "Attributes": [],
@@ -4537,6 +4552,11 @@
       },
       "NestedTypes": {}
     },
+    "Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector;;IsAbstract:True;IsSealed:False;IsInterface:True;IsEnum:False;IsClass:False;IsValueType:False;IsNested:False;IsGenericType:False;IsSerializable:False": {
+      "Subclasses": {},
+      "Members": {},
+      "NestedTypes": {}
+    },
     "Microsoft.Azure.Cosmos.FeedIterator;System.Object;IsAbstract:True;IsSealed:False;IsInterface:False;IsEnum:False;IsClass:True;IsValueType:False;IsNested:False;IsGenericType:False;IsSerializable:False": {
       "Subclasses": {},
       "Members": {
@@ -4922,6 +4942,11 @@
           "Attributes": [],
           "MethodInfo": "Microsoft.Azure.Cosmos.Fluent.CosmosClientBuilder WithCustomSerializer(Microsoft.Azure.Cosmos.CosmosSerializer);IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
         },
+        "Microsoft.Azure.Cosmos.Fluent.CosmosClientBuilder WithFaultInjection(Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector)": {
+          "Type": "Method",
+          "Attributes": [],
+          "MethodInfo": "Microsoft.Azure.Cosmos.Fluent.CosmosClientBuilder WithFaultInjection(Microsoft.Azure.Cosmos.FaultInjection.IFaultInjector);IsAbstract:False;IsStatic:False;IsVirtual:False;IsGenericMethod:False;IsConstructor:False;IsFinal:False;"
+        },
         "Microsoft.Azure.Cosmos.Fluent.CosmosClientBuilder WithHttpClientFactory(System.Func`1[System.Net.Http.HttpClient])": {
           "Type": "Method",
           "Attributes": [],