Adding OnCreateClient (#551)

Co-authored-by: Deepanshu Vaid <[email protected]>

Author: deep1712

src/VirtualClient/VirtualClient.TestFramework/InMemorySshClientFactory.cs

@@ -3,6 +3,7 @@
 
 namespace VirtualClient
 {
+    using System;
     using System.Collections.Generic;
     using Renci.SshNet;
 
@@ -24,10 +25,15 @@ public InMemorySshClientFactory()
         /// </summary>
         public IEnumerable<ISshClientProxy> SshClients { get; }
 
+        /// <summary>
+        /// Delegate defines logic to execute when the 'CreateClient' method is called.
+        /// </summary>
+        public Func<ConnectionInfo, ISshClientProxy> OnCreateClient { get; set; }
+
         /// <inheritdoc />
         public ISshClientProxy CreateClient(ConnectionInfo connection)
         {
-            ISshClientProxy sshClient = new InMemorySshClient(connection);
+            ISshClientProxy sshClient = this.OnCreateClient?.Invoke(connection) ?? new InMemorySshClient(connection);
             (this.SshClients as List<ISshClientProxy>).Add(sshClient);
 
             return sshClient;