MdePkg/MockUefiLib: Added new mock functions.

Signed-off-by: v-sbolisetti <[email protected]>
NVIDIA · Oct 28, 2024 · ea59e39 · ea59e39
1 parent 87e2ee6
commit ea59e39
Show file tree

Hide file tree

Showing 2 changed files with 382 additions and 4 deletions.
diff --git a/MdePkg/Test/Mock/Include/GoogleTest/Library/MockUefiLib.h b/MdePkg/Test/Mock/Include/GoogleTest/Library/MockUefiLib.h
@@ -2,6 +2,7 @@
   Google Test mocks for UefiLib
 
   Copyright (c) 2022, Intel Corporation. All rights reserved.
+  Copyright (c) Microsoft Corporation.
   SPDX-License-Identifier: BSD-2-Clause-Patent
 **/
 
@@ -18,21 +19,354 @@ extern "C" {
 struct MockUefiLib {
   MOCK_INTERFACE_DECLARATION (MockUefiLib);
 
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiGetSystemConfigurationTable,
+    (IN  EFI_GUID  *TableGuid,
+     OUT VOID      **Table)
+    );
+
+  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)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiNamedEventListen,
+    (IN CONST EFI_GUID          *Name,
+     IN       EFI_TPL           NotifyTpl,
+     IN       EFI_EVENT_NOTIFY  NotifyFunction,
+     IN CONST VOID              *NotifyContext OPTIONAL,
+     OUT      VOID              *Registration OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiNamedEventSignal,
+    (IN CONST EFI_GUID  *Name)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiEventGroupSignal,
+    (IN CONST EFI_GUID  *EventGroup)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    VOID,
+    EfiEventEmptyFunction,
+    (IN EFI_EVENT  Event,
+     IN VOID       *Context)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_TPL,
+    EfiGetCurrentTpl,
+    ()
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_LOCK *,
+    EfiInitializeLock,
+    (IN OUT EFI_LOCK  *Lock,
+     IN     EFI_TPL   Priority)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    VOID,
+    EfiAcquireLock,
+    (IN EFI_LOCK  *Lock)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiAcquireLockOrFail,
+    (IN EFI_LOCK  *Lock)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    VOID,
+    EfiReleaseLock,
+    (IN EFI_LOCK  *Lock)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiTestManagedDevice,
+    (IN CONST EFI_HANDLE  ControllerHandle,
+     IN CONST EFI_HANDLE  DriverBindingHandle,
+     IN CONST EFI_GUID    *ProtocolGuid)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiTestChildHandle,
+    (IN CONST EFI_HANDLE  ControllerHandle,
+     IN CONST EFI_HANDLE  ChildHandle,
+     IN CONST EFI_GUID    *ProtocolGuid)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    IsLanguageSupported,
+    (IN CONST CHAR8  *SupportedLanguages,
+     IN CONST CHAR8  *TargetLanguage)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    LookupUnicodeString,
+    (IN CONST CHAR8                     *Language,
+     IN CONST CHAR8                     *SupportedLanguages,
+     IN CONST EFI_UNICODE_STRING_TABLE  *UnicodeStringTable,
+     OUT      CHAR16                    **UnicodeString)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    LookupUnicodeString2,
+    (IN CONST CHAR8                     *Language,
+     IN CONST CHAR8                     *SupportedLanguages,
+     IN CONST EFI_UNICODE_STRING_TABLE  *UnicodeStringTable,
+     OUT      CHAR16                    **UnicodeString,
+     IN       BOOLEAN                   Iso639Language)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    AddUnicodeString,
+    (IN CONST CHAR8                     *Language,
+     IN CONST CHAR8                     *SupportedLanguages,
+     IN OUT   EFI_UNICODE_STRING_TABLE  **UnicodeStringTable,
+     IN CONST CHAR16                    *UnicodeString)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    AddUnicodeString2,
+    (IN CONST CHAR8                     *Language,
+     IN CONST CHAR8                     *SupportedLanguages,
+     IN OUT   EFI_UNICODE_STRING_TABLE  **UnicodeStringTable,
+     IN CONST CHAR16                    *UnicodeString,
+     IN       BOOLEAN                   Iso639Language)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    FreeUnicodeStringTable,
+    (IN EFI_UNICODE_STRING_TABLE  *UnicodeStringTable)
+    );
+
   MOCK_FUNCTION_DECLARATION (
     EFI_STATUS,
     GetVariable2,
     (IN CONST CHAR16    *Name,
      IN CONST EFI_GUID  *Guid,
-     OUT VOID           **Value,
-     OUT UINTN          *Size OPTIONAL)
+     OUT      VOID      **Value,
+     OUT      UINTN     *Size OPTIONAL)
     );
 
   MOCK_FUNCTION_DECLARATION (
     EFI_STATUS,
     GetEfiGlobalVariable2,
     (IN CONST CHAR16  *Name,
-     OUT VOID         **Value,
-     OUT UINTN        *Size OPTIONAL)
+     OUT      VOID    **Value,
+     OUT      UINTN   *Size OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    GetVariable3,
+    (IN CONST CHAR16    *Name,
+     IN CONST EFI_GUID  *Guid,
+     OUT      VOID      **Value,
+     OUT      UINTN     *Size OPTIONAL,
+     OUT      UINT32    *Attr OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    UINTN,
+    GetGlyphWidth,
+    (IN CHAR16  UnicodeChar)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    UINTN,
+    UnicodeStringDisplayLength,
+    (IN CONST CHAR16  *String)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    VOID,
+    EfiSignalEventReadyToBoot,
+    ()
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    VOID,
+    EfiSignalEventLegacyBoot,
+    ()
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiCreateEventLegacyBoot,
+    (OUT EFI_EVENT  *LegacyBootEvent)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiCreateEventLegacyBootEx,
+    (IN  EFI_TPL           NotifyTpl,
+     IN  EFI_EVENT_NOTIFY  NotifyFunction OPTIONAL,
+     IN  VOID              *NotifyContext OPTIONAL,
+     OUT EFI_EVENT         *LegacyBootEvent)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiCreateEventReadyToBoot,
+    (OUT EFI_EVENT  *ReadyToBootEvent)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiCreateEventReadyToBootEx,
+    (IN  EFI_TPL           NotifyTpl,
+     IN  EFI_EVENT_NOTIFY  NotifyFunction OPTIONAL,
+     IN  VOID              *NotifyContext OPTIONAL,
+     OUT EFI_EVENT         *ReadyToBootEvent)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    VOID,
+    EfiInitializeFwVolDevicepathNode,
+    (IN OUT   MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  *FvDevicePathNode,
+     IN CONST EFI_GUID                           *NameGuid)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_GUID *,
+    EfiGetNameGuidFromFwVolDevicePathNode,
+    (IN CONST MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  *FvDevicePathNode)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibInstallDriverBinding,
+    (IN CONST EFI_HANDLE                   ImageHandle,
+     IN CONST EFI_SYSTEM_TABLE             *SystemTable,
+     IN       EFI_DRIVER_BINDING_PROTOCOL  *DriverBinding,
+     IN       EFI_HANDLE                   DriverBindingHandle)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibUninstallDriverBinding,
+    (IN EFI_DRIVER_BINDING_PROTOCOL  *DriverBinding)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibInstallAllDriverProtocols,
+    (IN CONST EFI_HANDLE                         ImageHandle,
+     IN CONST EFI_SYSTEM_TABLE                   *SystemTable,
+     IN       EFI_DRIVER_BINDING_PROTOCOL        *DriverBinding,
+     IN       EFI_HANDLE                         DriverBindingHandle,
+     IN CONST EFI_COMPONENT_NAME_PROTOCOL        *ComponentName OPTIONAL,
+     IN CONST EFI_DRIVER_CONFIGURATION_PROTOCOL  *DriverConfiguration OPTIONAL,
+     IN CONST EFI_DRIVER_DIAGNOSTICS_PROTOCOL    *DriverDiagnostics OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibUninstallAllDriverProtocols,
+    (IN       EFI_DRIVER_BINDING_PROTOCOL        *DriverBinding,
+     IN CONST EFI_COMPONENT_NAME_PROTOCOL        *ComponentName OPTIONAL,
+     IN CONST EFI_DRIVER_CONFIGURATION_PROTOCOL  *DriverConfiguration OPTIONAL,
+     IN CONST EFI_DRIVER_DIAGNOSTICS_PROTOCOL    *DriverDiagnostics OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibInstallDriverBindingComponentName2,
+    (IN CONST EFI_HANDLE                    ImageHandle,
+     IN CONST EFI_SYSTEM_TABLE              *SystemTable,
+     IN       EFI_DRIVER_BINDING_PROTOCOL   *DriverBinding,
+     IN       EFI_HANDLE                    DriverBindingHandle,
+     IN CONST EFI_COMPONENT_NAME_PROTOCOL   *ComponentName OPTIONAL,
+     IN CONST EFI_COMPONENT_NAME2_PROTOCOL  *ComponentName2 OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibUninstallDriverBindingComponentName2,
+    (IN       EFI_DRIVER_BINDING_PROTOCOL   *DriverBinding,
+     IN CONST EFI_COMPONENT_NAME_PROTOCOL   *ComponentName OPTIONAL,
+     IN CONST EFI_COMPONENT_NAME2_PROTOCOL  *ComponentName2 OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibInstallAllDriverProtocols2,
+    (IN CONST EFI_HANDLE                          ImageHandle,
+     IN CONST EFI_SYSTEM_TABLE                    *SystemTable,
+     IN       EFI_DRIVER_BINDING_PROTOCOL         *DriverBinding,
+     IN       EFI_HANDLE                          DriverBindingHandle,
+     IN CONST EFI_COMPONENT_NAME_PROTOCOL         *ComponentName OPTIONAL,
+     IN CONST EFI_COMPONENT_NAME2_PROTOCOL        *ComponentName2 OPTIONAL,
+     IN CONST EFI_DRIVER_CONFIGURATION_PROTOCOL   *DriverConfiguration OPTIONAL,
+     IN CONST EFI_DRIVER_CONFIGURATION2_PROTOCOL  *DriverConfiguration2 OPTIONAL,
+     IN CONST EFI_DRIVER_DIAGNOSTICS_PROTOCOL     *DriverDiagnostics OPTIONAL,
+     IN CONST EFI_DRIVER_DIAGNOSTICS2_PROTOCOL    *DriverDiagnostics2 OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLibUninstallAllDriverProtocols2,
+    (IN       EFI_DRIVER_BINDING_PROTOCOL         *DriverBinding,
+     IN CONST EFI_COMPONENT_NAME_PROTOCOL         *ComponentName OPTIONAL,
+     IN CONST EFI_COMPONENT_NAME2_PROTOCOL        *ComponentName2 OPTIONAL,
+     IN CONST EFI_DRIVER_CONFIGURATION_PROTOCOL   *DriverConfiguration OPTIONAL,
+     IN CONST EFI_DRIVER_CONFIGURATION2_PROTOCOL  *DriverConfiguration2 OPTIONAL,
+     IN CONST EFI_DRIVER_DIAGNOSTICS_PROTOCOL     *DriverDiagnostics OPTIONAL,
+     IN CONST EFI_DRIVER_DIAGNOSTICS2_PROTOCOL    *DriverDiagnostics2 OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiLocateProtocolBuffer,
+    (IN  EFI_GUID  *Protocol,
+     OUT UINTN     *NoProtocols,
+     OUT VOID      ***Buffer)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_STATUS,
+    EfiOpenFileByDevicePath,
+    (IN OUT EFI_DEVICE_PATH_PROTOCOL  **FilePath,
+     OUT    EFI_FILE_PROTOCOL         **File,
+     IN     UINT64                    OpenMode,
+     IN     UINT64                    Attributes)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_ACPI_COMMON_HEADER *,
+    EfiLocateNextAcpiTable,
+    (IN UINT32                  Signature,
+     IN EFI_ACPI_COMMON_HEADER  *PreviousTable OPTIONAL)
+    );
+
+  MOCK_FUNCTION_DECLARATION (
+    EFI_ACPI_COMMON_HEADER *,
+    EfiLocateFirstAcpiTable,
+    (IN UINT32  Signature)
     );
 };