Skip to content

Commit 52c57be

Browse files
YiTa-AMIVivianNK
authored andcommitted
Match Edk2 design
1 parent f996cc6 commit 52c57be

File tree

1 file changed

+10
-13
lines changed

1 file changed

+10
-13
lines changed

MdePkg/Test/Mock/Include/GoogleTest/Protocol/MockMmReportStatusCodeHandler.h

Lines changed: 10 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ extern "C" {
1717
#include <Protocol/MmReportStatusCodeHandler.h>
1818
}
1919

20-
struct MockReportStatusCodeHandler {
21-
MOCK_INTERFACE_DECLARATION (MockReportStatusCodeHandler);
20+
struct MockEfiMmRscHandlerProtocol {
21+
MOCK_INTERFACE_DECLARATION (MockEfiMmRscHandlerProtocol);
2222

2323
MOCK_FUNCTION_DECLARATION (
2424
EFI_STATUS,
@@ -34,17 +34,14 @@ struct MockReportStatusCodeHandler {
3434
);
3535
};
3636

37-
MOCK_INTERFACE_DEFINITION (MockReportStatusCodeHandler);
38-
MOCK_FUNCTION_DEFINITION (MockReportStatusCodeHandler, Register, 1, EFIAPI);
39-
MOCK_FUNCTION_DEFINITION (MockReportStatusCodeHandler, Unregister, 1, EFIAPI);
37+
MOCK_INTERFACE_DEFINITION (MockEfiMmRscHandlerProtocol);
38+
MOCK_FUNCTION_DEFINITION (MockEfiMmRscHandlerProtocol, Register, 1, EFIAPI);
39+
MOCK_FUNCTION_DEFINITION (MockEfiMmRscHandlerProtocol, Unregister, 1, EFIAPI);
4040

41-
EFI_MM_RSC_HANDLER_PROTOCOL MmRscHandlerProtocol = {
42-
Register,
43-
Unregister
44-
};
45-
46-
extern "C" {
47-
EFI_MM_RSC_HANDLER_PROTOCOL *MmRscHandlerProtocolServices = &MmRscHandlerProtocol;
48-
}
41+
#define MOCK_EFI_MM_RSC_HANDLER_PROTOCOL_INSTANCE(NAME) \
42+
EFI_MM_RSC_HANDLER_PROTOCOL NAME##_INSTANCE = { \
43+
Register, \
44+
Unregister }; \
45+
EFI_MM_RSC_HANDLER_PROTOCOL *NAME = &NAME##_INSTANCE;
4946

5047
#endif

0 commit comments

Comments
 (0)