Commit

VSPackageInstall/VSPackageInstall_UnitTests/MyToolWindowTest/UIShellServiceMock.cs

@@ -0,0 +1,84 @@
+/***************************************************************************
+
+Copyright (c) Microsoft Corporation. All rights reserved.
+This code is licensed under the Visual Studio SDK license terms.
+THIS CODE IS PROVIDED *AS IS* WITHOUT WARRANTY OF
+ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING ANY
+IMPLIED WARRANTIES OF FITNESS FOR A PARTICULAR
+PURPOSE, MERCHANTABILITY, OR NON-INFRINGEMENT.
+
+***************************************************************************/
+
+using System;
+using Microsoft.VisualStudio;
+using Microsoft.VisualStudio.Shell.Interop;
+using Microsoft.VsSDK.UnitTestLibrary;
+
+namespace VSPackageInstall_UnitTests.MyToolWindowTest
+{
+    static class UIShellServiceMock
+    {
+        private static GenericMockFactory uiShellFactory;
+
+        #region UiShell Getters
+        /// <summary>
+        /// Returns an IVsUiShell that does not implement any methods
+        /// </summary>
+        /// <returns></returns>
+        internal static BaseMock GetUiShellInstance()
+        {
+            if (uiShellFactory == null)
+            {
+                uiShellFactory = new GenericMockFactory("UiShell", new Type[] { typeof(IVsUIShell), typeof(IVsUIShellOpenDocument) });
+            }
+            BaseMock uiShell = uiShellFactory.GetInstance();
+            return uiShell;
+        }
+
+        /// <summary>
+        /// Get an IVsUiShell that implement CreateToolWindow
+        /// </summary>
+        /// <returns>uishell mock</returns>
+        internal static BaseMock GetUiShellInstanceCreateToolWin()
+        {
+            BaseMock uiShell = GetUiShellInstance();
+            string name = string.Format("{0}.{1}", typeof(IVsUIShell).FullName, "CreateToolWindow");
+            uiShell.AddMethodCallback(name, new EventHandler<CallbackArgs>(CreateToolWindowCallBack));
+
+            return uiShell;
+        }
+
+        /// <summary>
+        /// Get an IVsUiShell that implement CreateToolWindow (negative test)
+        /// </summary>
+        /// <returns>uishell mock</returns>
+        internal static BaseMock GetUiShellInstanceCreateToolWinReturnsNull()
+        {
+            BaseMock uiShell = GetUiShellInstance();
+            string name = string.Format("{0}.{1}", typeof(IVsUIShell).FullName, "CreateToolWindow");
+            uiShell.AddMethodCallback(name, new EventHandler<CallbackArgs>(CreateToolWindowNegativeTestCallBack));
+
+            return uiShell;
+        }
+        #endregion
+
+        #region Callbacks
+        private static void CreateToolWindowCallBack(object caller, CallbackArgs arguments)
+        {
+            arguments.ReturnValue = VSConstants.S_OK;
+
+            // Create the output mock object for the frame
+            IVsWindowFrame frame = WindowFrameMock.GetBaseFrame();
+            arguments.SetParameter(9, frame);
+        }
+
+        private static void CreateToolWindowNegativeTestCallBack(object caller, CallbackArgs arguments)
+        {
+            arguments.ReturnValue = VSConstants.S_OK;
+
+            //set the windowframe object to null
+            arguments.SetParameter(9, null);
+        }
+        #endregion
+    }
+}