MockRng.h

Go to the documentation of this file.

 
#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_