Update ResultsCommandExtension.cs

src/Agent.Worker/TestResults/ResultsCommandExtension.cs

@@ -311,17 +311,17 @@ private PublishOptions GetPublishOptions()
         private async Task PublishTestRunDataAsync(string teamProject, TestRunContext testRunContext)
         {
             bool isTestRunOutcomeFailed = false;
-
+        
             _telemetryProperties.Add("UsePublishTestResultsLib", _publishTestResultsLibFeatureState);
-            using (var connection = WorkerUtilities.GetVssConnection(_executionContext))
+            var connection = WorkerUtilities.GetVssConnection(_executionContext);
+
+            try
             {
-
-                //This check is to determine to use "Microsoft.TeamFoundation.PublishTestResults" Library or the agent code to parse and publish the test results.
                 if (_publishTestResultsLibFeatureState)
                 {
                     var publisher = _executionContext.GetHostContext().GetService<ITestDataPublisher>();
                     publisher.InitializePublisher(_executionContext, teamProject, connection, _testRunner);
-
+        
                     if (_enableAzureTestPlanFeatureState && !_testCaseResults.IsNullOrEmpty() && !_testPlanId.IsNullOrEmpty())
                     {
                         isTestRunOutcomeFailed = await publisher.PublishAsync(testRunContext, _testResultFiles, _testCaseResults, GetPublishOptions(), _executionContext.CancellationToken);
@@ -335,22 +335,26 @@ private async Task PublishTestRunDataAsync(string teamProject, TestRunContext te
                 {
                     var publisher = _executionContext.GetHostContext().GetService<ILegacyTestRunDataPublisher>();
                     publisher.InitializePublisher(_executionContext, teamProject, connection, _testRunner, _publishRunLevelAttachments);
-
+        
                     isTestRunOutcomeFailed = await publisher.PublishAsync(testRunContext, _testResultFiles, _runTitle, _executionContext.Variables.Build_BuildId, _mergeResults);
                 }
-
+        
                 if (isTestRunOutcomeFailed && _failTaskOnFailedTests)
                 {
                     _executionContext.Result = TaskResult.Failed;
                     _executionContext.Error(StringUtil.Loc("FailedTestsInResults"));
                 }
-
+        
                 await PublishEventsAsync(connection);
                 if (_triggerCoverageMergeJobFeatureState)
                 {
                     TriggerCoverageMergeJob(_testResultFiles, _executionContext);
                 }
             }
+            finally
+            {
+                connection?.Dispose();
+            }
         }
 
         // Queue code coverage merge job if code coverage attachments are published to avoid BQC timeout.