Refactored the VS add-in

visual_studio/NativeClientVSAddIn/UnitTests/TestUtilities.cs

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 namespace UnitTests
 {
   using System;
   using System.Collections.Generic;
   using System.IO;
   using System.Linq;
@@ skipping 18 matching lines @@
     /// Uniquename of the NaCl project in BlankValidSolution.
     /// </summary>
     public const string BlankNaClProjectUniqueName = @"NaClProject\NaClProject.vcxproj";
 
     /// <summary>
     /// Uniquename of the non-NaCl project in BlankValidSolution.
     /// </summary>
     public const string NotNaClProjectUniqueName = @"NotNaCl\NotNaCl.csproj";
 
     /// <summary>
+    /// A generic boolean statement to be used with RetryWithTimeout.
+    /// </summary>
+    /// <returns>True if the statement is true, false if false.</returns>
+    public delegate bool RetryStatement();
+
+    /// <summary>
     /// This starts an instance of Visual Studio and get its DTE object.
     /// </summary>
     /// <returns>DTE of the started instance.</returns>
     public static DTE2 StartVisualStudioInstance()
     {
       // Set up filter to handle threading events and automatically retry calls
       // to dte which fail because dte is busy.
       ComMessageFilter.Register();
 
       Type visualStudioType = Type.GetTypeFromProgID("VisualStudio.DTE.10.0");
@@ skipping 99 matching lines @@
         {
           found = true;
           break;
         }
       }
 
       Assert.IsTrue(found, "Add-in is not configured to load on start.");
     }
 
     /// <summary>
+    /// Will retry the given statement up to maxRetry times while pausing between each try for
+    /// the given interval.
+    /// </summary>
+    /// <param name="test">Generic boolean statement.</param>
+    /// <param name="interval">Amount of time to wait between each retry.</param>
+    /// <param name="maxRetry">Maximum number of retries.</param>
+    /// <param name="message">Message to print on failure.</param>
+    public static void AssertTrueWithTimeout(
+        RetryStatement test, TimeSpan interval, int maxRetry, string message)
+    {
+      for (int tryCount = 0; tryCount <= maxRetry; tryCount++)
+      {
+        if (test.Invoke())
+        {
+          return;
+        }
+
+        System.Threading.Thread.Sleep(interval);
+      }
+
+      throw new Exception(string.Format("Statement timed out. {0}", message));
+    }
+
+    /// <summary>
     /// This returns the text contained in the given output window pane.
     /// </summary>
     /// <param name="pane">Pane to get text from.</param>
     /// <returns>Text in the window.</returns>
     public static string GetPaneText(OutputWindowPane pane)
     {
       TextSelection selection = pane.TextDocument.Selection;
       selection.StartOfDocument(false);
       selection.EndOfDocument(true);
       return selection.Text;
@@ skipping 256 matching lines @@
       }
 
       foreach (DirectoryInfo subdir in dir.GetDirectories())
       {
         string path = Path.Combine(dest, subdir.Name);
         CopyDirectory(subdir.FullName, path);
       }
     }
   }
 }