1 files changed, 32 insertions, 3 deletions

diff --git a/src/test/examples/TestEngine/TestEngine.h b/src/test/examples/TestEngine/TestEngine.h
index cf1c8aac..14b69999 100644
--- a/src/test/examples/TestEngine/TestEngine.h
+++ b/src/test/examples/TestEngine/TestEngine.h
@@ -2,6 +2,20 @@
 // 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.
 
 
+enum WM_TESTENG
+{
+    WM_TESTENG_FIRST = WM_APP + 0xFFF, // this enum value must always be first.
+
+    WM_TESTENG_DETECT,
+    WM_TESTENG_PLAN,
+    WM_TESTENG_ELEVATE,
+    WM_TESTENG_APPLY,
+    WM_TESTENG_LAUNCH_APPROVED_EXE,
+    WM_TESTENG_QUIT,
+
+    WM_TESTENG_LAST, // this enum value must always be last.
+};
+
 class TestEngine
 {
 public:
@@ -17,19 +31,29 @@ public:
         __in LPCWSTR wzMessage
         );
 
+    HRESULT RunApplication();
+
     HRESULT SendShutdownEvent(
         __in BOOTSTRAPPER_SHUTDOWN_ACTION defaultAction
         );
 
     HRESULT SendStartupEvent();
 
+    HRESULT SimulateQuit(
+        __in DWORD dwExitCode
+        );
+
     void UnloadBA();
 
 private:
-    static HRESULT BAEngineLog(
-        __in TestEngine* pContext,
+    HRESULT BAEngineLog(
         __in BAENGINE_LOG_ARGS* pArgs,
-        __in BAENGINE_LOG_RESULTS* /*pResults*/
+        __in BAENGINE_LOG_RESULTS* pResults
+        );
+
+    HRESULT BAEngineQuit(
+        __in BAENGINE_QUIT_ARGS* pArgs,
+        __in BAENGINE_QUIT_RESULTS* pResults
         );
 
     static HRESULT WINAPI EngineProc(
@@ -39,6 +63,10 @@ private:
         __in_opt LPVOID pvContext
         );
 
+    HRESULT ProcessBAMessage(
+        __in const MSG* pmsg
+        );
+
 public:
     TestEngine();
 
@@ -47,4 +75,5 @@ public:
 private:
     HMODULE m_hBAModule;
     BOOTSTRAPPER_CREATE_RESULTS* m_pCreateResults;
+    DWORD m_dwThreadId;
 };
\ No newline at end of file