1 files changed, 39 insertions, 0 deletions

diff --git a/src/test/burn/TestBA/MessagePump.cs b/src/test/burn/TestBA/MessagePump.cs
new file mode 100644
index 00000000..21a00349
--- /dev/null
+++ b/src/test/burn/TestBA/MessagePump.cs
@@ -0,0 +1,39 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Test.BA
+{
+    using System;
+    using System.Runtime.InteropServices;
+    using System.Windows.Forms;
+
+    public class MessagePump
+    {
+        const uint PM_REMOVE = 1;
+
+        [DllImport("user32.dll", ExactSpelling = true)]
+        [return: MarshalAs(UnmanagedType.Bool)]
+        public static extern bool PeekMessageW(ref Message pMsg, IntPtr hWnd, uint wMsgFilterMin, uint wMsgFilterMax, uint wRemoveMsg);
+
+        [DllImport("user32.dll", ExactSpelling = true)]
+        [return: MarshalAs(UnmanagedType.Bool)]
+        public static extern bool TranslateMessage(ref Message pMsg);
+
+        [DllImport("user32.dll", ExactSpelling = true)]
+        public static extern IntPtr DispatchMessageW(ref Message pMsg);
+
+        public static void ProcessMessages(int maxMessages)
+        {
+            for (int i = 0; i < maxMessages; i++)
+            {
+                Message message = new Message();
+                if (!PeekMessageW(ref message, IntPtr.Zero, 0, 0, PM_REMOVE))
+                {
+                    break;
+                }
+
+                TranslateMessage(ref message);
+                DispatchMessageW(ref message);
+            }
+        }
+    }
+}