audk/MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockRng.h

49 lines
1022 B
C

/** @file
This file declares a mock of Rng Protocol.
Copyright (c) Microsoft Corporation.
SPDX-License-Identifier: BSD-2-Clause-Patent
**/
#ifndef MOCK_RNG_H_
#define MOCK_RNG_H_
#include <Library/GoogleTestLib.h>
#include <Library/FunctionMockLib.h>
extern "C" {
#include <Uefi.h>
#include <Protocol/Rng.h>
}
struct MockRng {
MOCK_INTERFACE_DECLARATION (MockRng);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
GetInfo,
(
IN EFI_RNG_PROTOCOL *This,
IN OUT UINTN *RNGAlgorithmListSize,
OUT EFI_RNG_ALGORITHM *RNGAlgorithmList
)
);
MOCK_FUNCTION_DECLARATION (
EFI_STATUS,
GetRng,
(
IN EFI_RNG_PROTOCOL *This,
IN EFI_RNG_ALGORITHM *RNGAlgorithm,
IN UINTN RNGValueLength,
OUT UINT8 *RNGValue
)
);
};
extern "C" {
extern EFI_RNG_PROTOCOL *gRngProtocol;
}
#endif // MOCK_RNG_H_