diff --git a/MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockRng.h b/MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockRng.h index b77ec134c1..5c1d833cf0 100644 --- a/MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockRng.h +++ b/MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockRng.h @@ -20,23 +20,24 @@ 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 - ) + 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 *RNGAlgorithmList, - IN UINTN RNGAlgorithmListSize, - IN OUT UINT8 *RandomNumber, - IN UINTN RandomNumberLength - ) + EFI_STATUS, + GetRng, + ( + IN EFI_RNG_PROTOCOL *This, + IN EFI_RNG_ALGORITHM *RNGAlgorithm, + IN UINTN RNGValueLength, + OUT UINT8 *RNGValue + ) ); }; diff --git a/MdePkg/Test/Mock/Library/GoogleTest/Protocol/MockRng.cpp b/MdePkg/Test/Mock/Library/GoogleTest/Protocol/MockRng.cpp index f8680088c6..7d8b69cc86 100644 --- a/MdePkg/Test/Mock/Library/GoogleTest/Protocol/MockRng.cpp +++ b/MdePkg/Test/Mock/Library/GoogleTest/Protocol/MockRng.cpp @@ -9,7 +9,7 @@ MOCK_INTERFACE_DEFINITION (MockRng); MOCK_FUNCTION_DEFINITION (MockRng, GetInfo, 3, EFIAPI); -MOCK_FUNCTION_DEFINITION (MockRng, GetRng, 5, EFIAPI); +MOCK_FUNCTION_DEFINITION (MockRng, GetRng, 4, EFIAPI); EFI_RNG_PROTOCOL RNG_PROTOCOL_INSTANCE = { GetInfo, // EFI_RNG_GET_INFO