UnitTestFrameworkPkg: Modify APIs in UnitTestPersistenceLib

[mirror_edk2.git] / UnitTestFrameworkPkg / Library / UnitTestLib / UnitTestLib.c

diff --git a/UnitTestFrameworkPkg/Library/UnitTestLib/UnitTestLib.c b/UnitTestFrameworkPkg/Library/UnitTestLib/UnitTestLib.c
index 64d5880783bf8a4713a34ed2e686681f6915d5bf..5b442ed122ea1d3fc5f4a1a79829fff07f4413ab 100644 (file)
--- a/UnitTestFrameworkPkg/Library/UnitTestLib/UnitTestLib.c
+++ b/UnitTestFrameworkPkg/Library/UnitTestLib/UnitTestLib.c
@@ -2,6 +2,7 @@
   Implement UnitTestLib\r
 \r
   Copyright (c) Microsoft Corporation.\r
+  Copyright (c) 2022, Intel Corporation. All rights reserved.<BR>\r
   SPDX-License-Identifier: BSD-2-Clause-Patent\r
 **/\r
 \r
@@ -210,6 +211,7 @@ InitUnitTestFramework (
   EFI_STATUS                  Status;\r
   UNIT_TEST_FRAMEWORK_HANDLE  NewFrameworkHandle;\r
   UNIT_TEST_FRAMEWORK         *NewFramework;\r
+  UINTN                       SaveStateSize;\r
 \r
   Status       = EFI_SUCCESS;\r
   NewFramework = NULL;\r
@@ -267,7 +269,7 @@ InitUnitTestFramework (
   // If there is a persisted context, load it now.\r
   //\r
   if (DoesCacheExist (NewFrameworkHandle)) {\r
-    Status = LoadUnitTestCache (NewFrameworkHandle, (UNIT_TEST_SAVE_HEADER **)(&NewFramework->SavedState));\r
+    Status = LoadUnitTestCache (NewFrameworkHandle, (VOID **)(&NewFramework->SavedState), &SaveStateSize);\r
     if (EFI_ERROR (Status)) {\r
       //\r
       // Don't actually report it as an error, but emit a warning.\r
@@ -852,7 +854,7 @@ SaveFrameworkState (
   //\r
   // All that should be left to do is save it using the associated persistence lib.\r
   //\r
-  Status = SaveUnitTestCache (FrameworkHandle, Header);\r
+  Status = SaveUnitTestCache (FrameworkHandle, Header, Header->SaveStateSize);\r
   if (EFI_ERROR (Status)) {\r
     DEBUG ((DEBUG_ERROR, "%a - Could not save state! %r\n", __FUNCTION__, Status));\r
     Status = EFI_DEVICE_ERROR;\r