Refactored the VS add-in

visual_studio/NativeClientVSAddIn/UnitTests/WebServerTest.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 EnvDTE;
+  using EnvDTE80;
+  using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+  using NativeClientVSAddIn;
+
+  /// <summary>
+  /// This is a test class for WebServerTest and is intended
+  /// to contain all WebServer Unit Tests
+  /// </summary>
+  [TestClass]
+  public class WebServerTest
+  {
+    /// <summary>
+    /// The main visual studio object.
+    /// </summary>
+    private DTE2 dte_;
+
+    /// <summary>
+    /// Gets or sets the test context which provides
+    /// information about and functionality for the current test run.
+    /// </summary>
+    public TestContext TestContext { get; set; }
+
+    /// <summary>
+    /// This is run before each test to create test resources.
+    /// </summary>
+    [TestInitialize]
+    public void TestSetup()
+    {
+      dte_ = TestUtilities.StartVisualStudioInstance();
+      try
+      {
+        TestUtilities.AssertAddinLoaded(dte_, NativeClientVSAddIn.Strings.AddInName);
+      }
+      catch
+      {
+        TestUtilities.CleanUpVisualStudioInstance(dte_);
+        throw;
+      }
+    }
+
+    /// <summary>
+    /// This is run after each test to clean up things created in TestSetup().
+    /// </summary>
+    [TestCleanup]
+    public void TestCleanup()
+    {
+      TestUtilities.CleanUpVisualStudioInstance(dte_);
+    }
+
+    /// <summary>
+    /// A test for WebServer Constructor. Starts the web server.
+    /// </summary>
+    [TestMethod]
+    public void WebServerConstructorTest()
+    {
+      OutputWindowPane outputWindowPane = dte_.ToolWindows.OutputWindow.OutputWindowPanes.Add(
+          Strings.WebServerOutputWindowTitle);
+
+      // Set up mock property manager to return the desired property values.
+      MockPropertyManager properties = new MockPropertyManager(
+        PropertyManager.ProjectPlatformType.Pepper,
+        delegate(string page, string name)
+        {
+          switch (page)
+          {
+            case "ConfigurationGeneral":
+              switch (name)
+              {
+                case "VSNaClSDKRoot": return System.Environment.GetEnvironmentVariable(
+                    NativeClientVSAddIn.Strings.SDKPathEnvironmentVariable);
+                case "NaClWebServerPort": return "5105";
+              }
+
+              break;
+            case "Property":
+              switch (name)
+              {
+                case "ProjectDirectory": return TestContext.DeploymentDirectory;
+              }
+
+              break;
+          }
+
+          return null;
+        },
+        null);
+
+      WebServer target = null;
+      try
+      {
+        target = new WebServer(outputWindowPane, properties);
+
+        TestUtilities.AssertTrueWithTimeout(
+          () => !string.IsNullOrEmpty(TestUtilities.GetPaneText(outputWindowPane)),
+          TimeSpan.FromMilliseconds(500),
+          20,
+          "Pane text never appeared");
+
+        TestUtilities.AssertTrueWithTimeout(
+            () => TestUtilities.DoesProcessExist("python.exe", "5105", "httpd.py"),
+            TimeSpan.FromMilliseconds(500),
+            20,
+            "Web server failed to start.");
+
+        target.Dispose();
+
+        TestUtilities.AssertTrueWithTimeout(
+            () => !TestUtilities.DoesProcessExist("python.exe", "5105", "httpd.py"),
+            TimeSpan.FromMilliseconds(500),
+            20,
+            "Web server failed to shut down.");
+      }
+      finally
+      {
+        if (target != null)
+        {
+          target.Dispose();
+        }
+      }
+    }
+  }
+}