From f43d176f95601ff7524e06247166d4f3b6e61c05 Mon Sep 17 00:00:00 2001 From: Sean Hall Date: Fri, 2 Jul 2021 10:18:08 -0500 Subject: Make the BA responsible for parsing restart prompt behavior. Fixes #4975 --- .../burn/test/BalUtilUnitTest/TestBootstrapperApplication.cpp | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'src/api/burn/test/BalUtilUnitTest') diff --git a/src/api/burn/test/BalUtilUnitTest/TestBootstrapperApplication.cpp b/src/api/burn/test/BalUtilUnitTest/TestBootstrapperApplication.cpp index 13d22e72..daa1d690 100644 --- a/src/api/burn/test/BalUtilUnitTest/TestBootstrapperApplication.cpp +++ b/src/api/burn/test/BalUtilUnitTest/TestBootstrapperApplication.cpp @@ -8,9 +8,8 @@ class CTestBootstrapperApplication : public CBalBaseBootstrapperApplication { public: CTestBootstrapperApplication( - __in IBootstrapperEngine* pEngine, - __in const BOOTSTRAPPER_CREATE_ARGS* pArgs - ) : CBalBaseBootstrapperApplication(pEngine, pArgs) + __in IBootstrapperEngine* pEngine + ) : CBalBaseBootstrapperApplication(pEngine) { } }; @@ -25,9 +24,12 @@ HRESULT CreateBootstrapperApplication( HRESULT hr = S_OK; CTestBootstrapperApplication* pApplication = NULL; - pApplication = new CTestBootstrapperApplication(pEngine, pArgs); + pApplication = new CTestBootstrapperApplication(pEngine); ExitOnNull(pApplication, hr, E_OUTOFMEMORY, "Failed to create new test bootstrapper application object."); + hr = pApplication->Initialize(pArgs); + ExitOnFailure(hr, "CTestBootstrapperApplication initialization failed."); + pResults->pfnBootstrapperApplicationProc = BalBaseBootstrapperApplicationProc; pResults->pvBootstrapperApplicationProcContext = pApplication; *ppApplication = pApplication; -- cgit v1.2.3-55-g6feb