From eea310d41ff3661508aff0b904491c9d9fedd3e9 Mon Sep 17 00:00:00 2001
From: David Hall <dahall@byu.net>
Date: Mon, 23 Nov 2020 15:02:43 -0700
Subject: [PATCH] Added tests for `IContextMenu`

---
 UnitTests/PInvoke/Shell32/ContextMenuTests.cs | 75 +++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 UnitTests/PInvoke/Shell32/ContextMenuTests.cs

diff --git a/UnitTests/PInvoke/Shell32/ContextMenuTests.cs b/UnitTests/PInvoke/Shell32/ContextMenuTests.cs
new file mode 100644
index 00000000..d74174dc
--- /dev/null
+++ b/UnitTests/PInvoke/Shell32/ContextMenuTests.cs
@@ -0,0 +1,75 @@
+using NUnit.Framework;
+using System;
+using Vanara.InteropServices;
+using static Vanara.PInvoke.User32;
+using static Vanara.PInvoke.Shell32;
+using System.Runtime.InteropServices;
+using Vanara.Extensions;
+
+namespace Vanara.PInvoke.Tests
+{
+	[TestFixture]
+	public class ContextMenuTests
+	{
+		[Test]
+		public void QueryTest([Values] CMF cmf)
+		{
+			using var pshi = ComReleaserFactory.Create(SHCreateItemFromParsingName<IShellItem>(TestCaseSources.WordDoc));
+			using var pcm = ComReleaserFactory.Create(pshi.Item.BindToHandler<IContextMenu>(null, BHID.BHID_SFUIObject.Guid()));
+			using var hmenu = CreatePopupMenu();
+			Assert.That(pcm.Item.QueryContextMenu(hmenu, 0, 1, 0x7FFF, cmf), ResultIs.Successful);
+			var miis = MenuItemInfo.GetMenuItems(hmenu);
+			for (int i = 0; i < miis.Length; i++)
+				ShowMII(miis[i], i);
+
+			static void ShowMII(MenuItemInfo mii, int c, int indent = 0)
+			{
+				TestContext.WriteLine($"{new string(' ', indent * 3)}{c+1}) {mii.Text} (#{mii.Id}) - Type={mii.Type}; State={mii.State}");
+				for (int j = 0; j < mii.SubMenus.Length; j++)
+					ShowMII(mii.SubMenus[j], j, indent + 1);
+			}
+		}
+
+		public class MenuItemInfo
+		{
+			internal MenuItemInfo(HMENU hMenu, uint idx)
+			{
+				using var strmem = new SafeHGlobalHandle(512);
+				var mii = new MENUITEMINFO
+				{
+					cbSize = (uint)Marshal.SizeOf(typeof(MENUITEMINFO)),
+					fMask = MenuItemInfoMask.MIIM_ID | MenuItemInfoMask.MIIM_SUBMENU | MenuItemInfoMask.MIIM_FTYPE | MenuItemInfoMask.MIIM_STRING | MenuItemInfoMask.MIIM_STATE | MenuItemInfoMask.MIIM_BITMAP,
+					fType = MenuItemType.MFT_STRING,
+					dwTypeData = (IntPtr)strmem,
+					cch = strmem.Size / (uint)StringHelper.GetCharSize()
+				};
+				Win32Error.ThrowLastErrorIfFalse(GetMenuItemInfo(hMenu, idx, true, ref mii));
+				Id = mii.wID;
+				Text = mii.fType.IsFlagSet(MenuItemType.MFT_SEPARATOR) ? "-" : mii.fType.IsFlagSet(MenuItemType.MFT_STRING) ? strmem.ToString(-1, CharSet.Auto) : "";
+				Type = mii.fType;
+				State = mii.fState;
+				BitmapHandle = mii.hbmpItem;
+				SubMenus = GetMenuItems(mii.hSubMenu);
+			}
+
+			public static MenuItemInfo[] GetMenuItems(HMENU hMenu)
+			{
+				if (hMenu.IsNull)
+					return new MenuItemInfo[0];
+
+				var SubMenus = new MenuItemInfo[GetMenuItemCount(hMenu)];
+				for (uint i = 0; i < SubMenus.Length; i++)
+					SubMenus[i] = new MenuItemInfo(hMenu, i);
+				return SubMenus;
+			}
+
+			public uint Id { get; }
+			public string Text { get; }
+			public MenuItemType Type { get; }
+			public MenuItemState State { get; }
+			public MenuItemInfo[] SubMenus { get; }
+			public HBITMAP BitmapHandle { get; }
+		}
+
+	}
+}
\ No newline at end of file