Try synchronizing dispose and output (#20341)

Author: John Luo

src/Shared/Process/ProcessEx.cs

@@ -26,9 +26,11 @@ internal class ProcessEx : IDisposable
         private readonly StringBuilder _stderrCapture;
         private readonly StringBuilder _stdoutCapture;
         private readonly object _pipeCaptureLock = new object();
+        private readonly object _testOutputLock = new object();
         private BlockingCollection<string> _stdoutLines;
         private TaskCompletionSource<int> _exited;
         private CancellationTokenSource _stdoutLinesCancellationSource = new CancellationTokenSource(TimeSpan.FromMinutes(5));
+        private bool _disposed = false;
 
         public ProcessEx(ITestOutputHelper output, Process proc)
         {
@@ -135,7 +137,13 @@ private void OnErrorData(object sender, DataReceivedEventArgs e)
                 _stderrCapture.AppendLine(e.Data);
             }
 
-            _output.WriteLine("[ERROR] " + e.Data);
+            lock (_testOutputLock)
+            {
+                if (!_disposed)
+                {
+                    _output.WriteLine("[ERROR] " + e.Data);
+                }
+            }
         }
 
         private void OnOutputData(object sender, DataReceivedEventArgs e)
@@ -150,7 +158,13 @@ private void OnOutputData(object sender, DataReceivedEventArgs e)
                 _stdoutCapture.AppendLine(e.Data);
             }
 
-            _output.WriteLine(e.Data);
+            lock (_testOutputLock)
+            {
+                if (!_disposed)
+                {
+                    _output.WriteLine(e.Data);
+                }
+            }
 
             if (_stdoutLines != null)
             {
@@ -204,6 +218,11 @@ private static string GetNugetPackagesRestorePath() => (string.IsNullOrEmpty(Env
 
         public void Dispose()
         {
+            lock (_testOutputLock)
+            {
+                _disposed = true;
+            }
+
             if (_process != null && !_process.HasExited)
             {
                 _process.KillTree();