tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

MdePkg: Add MockHash2 Protocol for testing 

This commit adds a new MockHash2 protocol to the MdePkg. This allows
the unit tests to pick up the new protocol and use it for testing.

Cc: Michael D Kinney <michael.d.kinney@intel.com>
Cc: Liming Gao <gaoliming@byosoft.com.cn>
Cc: Zhiguang Liu <zhiguang.liu@intel.com>

Signed-off-by: Doug Flick [MSFT] <doug.edk2@gmail.com>
Reviewed-by: Liming Gao <gaoliming@byosoft.com.cn>
This commit is contained in:
Doug Flick 2024-05-08 22:56:32 -07:00 committed by mergify[bot]
parent 4afb939531
commit dff3d3811f
2 changed files with 94 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockHash2.h Normal file
View File

@ -0,0 +1,67 @@
/** @file
This file declares a mock of Hash2 Protocol.
Copyright (c) Microsoft Corporation.
SPDX-License-Identifier: BSD-2-Clause-Patent
**/
#ifndef MOCK_HASH2_H_
#define MOCK_HASH2_H_
#include <Library/GoogleTestLib.h>
#include <Library/FunctionMockLib.h>
extern "C" {
#include <Uefi.h>
#include <Protocol/Hash2.h>
}
struct MockHash2 {
MOCK_INTERFACE_DECLARATION (MockHash2);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
GetHashSize,
(IN CONST EFI_HASH2_PROTOCOL *This,
IN CONST EFI_GUID *HashAlgorithm,
OUT UINTN *HashSize)
);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
Hash,
(IN CONST EFI_HASH2_PROTOCOL *This,
IN CONST EFI_GUID *HashAlgorithm,
IN CONST UINT8 *Message,
IN UINTN MessageSize,
IN OUT EFI_HASH2_OUTPUT *Hash)
);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
HashInit,
(IN CONST EFI_HASH2_PROTOCOL *This,
IN CONST EFI_GUID *HashAlgorithm)
);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
HashUpdate,
(IN CONST EFI_HASH2_PROTOCOL *This,
IN CONST UINT8 *Message,
IN UINTN MessageSize)
);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
HashFinal,
(IN CONST EFI_HASH2_PROTOCOL *This,
IN OUT EFI_HASH2_OUTPUT *Hash)
);
};
extern "C" {
extern EFI_HASH2_PROTOCOL *gHash2Protocol;
}
#endif // MOCK_HASH2_H_

27
MdePkg/Test/Mock/Library/GoogleTest/Protocol/MockHash2.cpp Normal file
View File

@ -0,0 +1,27 @@
/** @file MockHash2.cpp
Google Test mock for Hash2 Protocol
Copyright (c) Microsoft Corporation.
SPDX-License-Identifier: BSD-2-Clause-Patent
**/
#include <GoogleTest/Protocol/MockHash2.h>
MOCK_INTERFACE_DEFINITION (MockHash2);
MOCK_FUNCTION_DEFINITION (MockHash2, GetHashSize, 3, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockHash2, Hash, 5, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockHash2, HashInit, 2, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockHash2, HashUpdate, 3, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockHash2, HashFinal, 2, EFIAPI);
EFI_HASH2_PROTOCOL HASH2_PROTOCOL_INSTANCE = {
GetHashSize, // EFI_HASH2_GET_HASH_SIZE
Hash, // EFI_HASH2_HASH
HashInit, // EFI_HASH2_HASH_INIT
HashUpdate, // EFI_HASH2_HASH_UPDATE
HashFinal // EFI_HASH2_HASH_FINAL
};
extern "C" {
EFI_HASH2_PROTOCOL *gHash2Protocol = &HASH2_PROTOCOL_INSTANCE;
}