Skip to content

Commit

Permalink
create mock for ReadOnlyVariable2
Browse files Browse the repository at this point in the history
  • Loading branch information
VivianNK committed Sep 27, 2023
1 parent c3baf89 commit 2789263
Show file tree
Hide file tree
Showing 2 changed files with 68 additions and 0 deletions.
47 changes: 47 additions & 0 deletions MdePkg/Test/Mock/Include/GoogleTest/Ppi/MockReadOnlyVariable2.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
/** @file MockReadOnlyVariable2.h
This file declares a mock of Read-only Variable Service2 PPI.
This ppi permits read-only access to the UEFI variable store during the PEI phase.
Copyright (c) Microsoft Corporation.
SPDX-License-Identifier: BSD-2-Clause-Patent
**/

#ifndef MOCK_PEI_READ_ONLY_VARIABLE2_PPI_H
#define MOCK_PEI_READ_ONLY_VARIABLE2_PPI_H

#include <Library/GoogleTestLib.h>
#include <Library/FunctionMockLib.h>
extern "C" {
#include <Uefi.h>
#include <Pi/PiPeiCis.h>
#include <Ppi/ReadOnlyVariable2.h>
}

struct MockReadOnlyVariable2 {
MOCK_INTERFACE_DECLARATION (MockReadOnlyVariable2);

MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
GetVariable,
(IN CONST EFI_PEI_READ_ONLY_VARIABLE2_PPI *This,
IN CONST CHAR16 *VariableName,
IN CONST EFI_GUID *VariableGuid,
OUT UINT32 *Attributes,
IN OUT UINTN *DataSize,
OUT VOID *Data OPTIONAL)
);

MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
NextVariableName,
(IN CONST EFI_PEI_READ_ONLY_VARIABLE2_PPI *This,
IN OUT UINTN *VariableNameSize,
IN OUT CHAR16 *VariableName,
IN OUT EFI_GUID *VariableGuid)
);
};

extern "C" {
extern EFI_PEI_READ_ONLY_VARIABLE2_PPI *PPIReadOnlyVariableServices;
}

#endif
21 changes: 21 additions & 0 deletions MdePkg/Test/Mock/Library/GoogleTest/Ppi/MockReadOnlyVariable2.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
/** @file MockReadOnlyVariable2.cpp
Google Test mock for ReadOnlyVariable2
Copyright (c) Microsoft Corporation.
SPDX-License-Identifier: BSD-2-Clause-Patent
**/
#include <GoogleTest/Ppi/MockReadOnlyVariable2.h>

MOCK_INTERFACE_DEFINITION (MockReadOnlyVariable2);
MOCK_FUNCTION_DEFINITION (MockReadOnlyVariable2, GetVariable, 6, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockReadOnlyVariable2, NextVariableName, 4, EFIAPI);

// Normally PPIVariableServices is "found"
// This will be defined INSIDE the test, with its definition pointing to the mock function getVariable
EFI_PEI_READ_ONLY_VARIABLE2_PPI peiReadOnlyVariablePpi = {
GetVariable, // EFI_PEI_GET_VARIABLE2
NextVariableName // EFI_PEI_GET_NEXT_VARIABLE_NAME2
};

extern "C" {
EFI_PEI_READ_ONLY_VARIABLE2_PPI *PPIReadOnlyVariableServices = &peiReadOnlyVariablePpi;
}

0 comments on commit 2789263

Please sign in to comment.