Adding Unit Test and Comments

microsoft · Dec 24, 2024 · ac20c16 · ac20c16
1 parent b0597f4
commit ac20c16
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/src/Agent.Sdk/Util/VssUtil.cs b/src/Agent.Sdk/Util/VssUtil.cs
@@ -164,6 +164,7 @@ public static bool IsCustomServerCertificateValidationSupported(ITraceWriter tra
             return true;
         }
 
+        // The function is to check if the custom server certificate validation is supported on the current platform.
         private static bool CheckSupportOfCustomServerCertificateValidation(ITraceWriter trace)
         {
             using (var handler = new HttpClientHandler())
@@ -175,10 +176,11 @@ private static bool CheckSupportOfCustomServerCertificateValidation(ITraceWriter
                     try
                     {
                         client.GetAsync(_testUri).GetAwaiter().GetResult();
+                        trace.Verbose("Custom Server Validation Callback Successful, SSL diagnostic data collection is enabled.");
                     }
                     catch (Exception e)
                     {
-                        trace.Verbose($"SSL diagnostic data collection is disabled, due to issue:\n{e.Message}");
+                        trace.Verbose($"Custom Server Validation Callback Unsuccessful, SSL diagnostic data collection is disabled, due to issue:\n{e.Message}");
                         return false;
                     }
                     return true;

diff --git a/src/Test/L0/Util/VssUtilL0.cs b/src/Test/L0/Util/VssUtilL0.cs
@@ -59,5 +59,38 @@ public void VerifyOverwriteVssConnectionSetting()
                 }
             }
         }
+
+        [Fact]
+        [Trait("Level", "L0")]
+        [Trait("Category", "Common")]
+        public void VerifyVSSConnectionUsingLegacyHandler()
+        {
+            Regex _serverSideAgentPlatformMatchingRegex = new Regex("vstsagentcore-(.+)(?=/)", RegexOptions.Compiled | RegexOptions.IgnoreCase);
+
+            using (TestHostContext hc = new TestHostContext(this))
+            {
+                Tracing trace = hc.GetTrace();
+                // Act.
+                try
+                {
+                    Environment.SetEnvironmentVariable("AZP_AGENT_USE_LEGACY_HTTP", "true");
+
+                    var exception = Record.Exception(() =>
+                    {
+                        var connection = VssUtil.CreateConnection(
+                            new Uri("https://github.com/Microsoft/vsts-agent"),
+                            new VssCredentials(),
+                            trace);
+                    });
+
+                    Assert.Null(exception);
+                }
+                finally
+                {
+                    Environment.SetEnvironmentVariable("AZP_AGENT_USE_LEGACY_HTTP", "");
+                }
+            }
+
+        }
     }
 }