Added mock functions on UefiLib

Added mock functions on UefiLib Signed-off-by: TsunFeng <[email protected]>
microsoft · Jul 23, 2024 · c41e694 · c41e694
1 parent 285f5d8
commit c41e694
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/MdePkg/Test/Mock/Include/GoogleTest/Library/MockUefiLib.h b/MdePkg/Test/Mock/Include/GoogleTest/Library/MockUefiLib.h
@@ -34,6 +34,16 @@ struct MockUefiLib {
      OUT VOID         **Value,
      OUT UINTN        *Size OPTIONAL)
     );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_EVENT,
+    EfiCreateProtocolNotifyEvent,
+    (IN EFI_GUID          *ProtocolGuid,
+     IN EFI_TPL           NotifyTpl,
+     IN EFI_EVENT_NOTIFY  NotifyFunction,
+     IN VOID              *NotifyContext OPTIONAL,
+     OUT VOID             **Registration)
+    );
 };
 
 #endif
diff --git a/MdePkg/Test/Mock/Library/GoogleTest/MockUefiLib/MockUefiLib.cpp b/MdePkg/Test/Mock/Library/GoogleTest/MockUefiLib/MockUefiLib.cpp
@@ -10,3 +10,4 @@ MOCK_INTERFACE_DEFINITION (MockUefiLib);
 
 MOCK_FUNCTION_DEFINITION (MockUefiLib, GetVariable2, 4, EFIAPI);
 MOCK_FUNCTION_DEFINITION (MockUefiLib, GetEfiGlobalVariable2, 3, EFIAPI);
+MOCK_FUNCTION_DEFINITION (MockUefiLib, EfiCreateProtocolNotifyEvent, 5, EFIAPI);
Original file line number	Diff line number	Diff line change
Expand Up		@@ -10,3 +10,4 @@ MOCK_INTERFACE_DEFINITION (MockUefiLib);

		MOCK_FUNCTION_DEFINITION (MockUefiLib, GetVariable2, 4, EFIAPI);
		MOCK_FUNCTION_DEFINITION (MockUefiLib, GetEfiGlobalVariable2, 3, EFIAPI);
		MOCK_FUNCTION_DEFINITION (MockUefiLib, EfiCreateProtocolNotifyEvent, 5, EFIAPI);