summaryrefslogtreecommitdiff
path: root/MdePkg/Test/Mock/Include/GoogleTest/Library/MockPerformanceLib.h
blob: 2523cee8beac391470825fa1c824f2822dd50031 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/** @file
  Google Test mocks for PerformanceLib

  Copyright (c) 2023, Intel Corporation. All rights reserved.
  Copyright (C) Microsoft Corporation.
  SPDX-License-Identifier: BSD-2-Clause-Patent
**/

#pragma once

#include <Library/GoogleTestLib.h>
#include <Library/FunctionMockLib.h>
extern "C" {
  #include <Uefi.h>
  #include <Library/PerformanceLib.h>
}

struct MockPerformanceLib {
  MOCK_INTERFACE_DECLARATION (MockPerformanceLib);

  MOCK_FUNCTION_DECLARATION (
    RETURN_STATUS,
    StartPerformanceMeasurement,
    (IN CONST VOID   *Handle   OPTIONAL,
     IN CONST CHAR8  *Token    OPTIONAL,
     IN CONST CHAR8  *Module   OPTIONAL,
     IN UINT64       TimeStamp)
    );

  MOCK_FUNCTION_DECLARATION (
    RETURN_STATUS,
    EndPerformanceMeasurement,
    (IN CONST VOID   *Handle   OPTIONAL,
     IN CONST CHAR8  *Token    OPTIONAL,
     IN CONST CHAR8  *Module   OPTIONAL,
     IN UINT64       TimeStamp)
    );

  MOCK_FUNCTION_DECLARATION (
    UINTN,
    GetPerformanceMeasurement,
    (IN  UINTN        LogEntryKey,
     OUT CONST VOID   **Handle,
     OUT CONST CHAR8  **Token,
     OUT CONST CHAR8  **Module,
     OUT UINT64       *StartTimeStamp,
     OUT UINT64       *EndTimeStamp)
    );

  MOCK_FUNCTION_DECLARATION (
    RETURN_STATUS,
    StartPerformanceMeasurementEx,
    (IN CONST VOID   *Handle   OPTIONAL,
     IN CONST CHAR8  *Token    OPTIONAL,
     IN CONST CHAR8  *Module   OPTIONAL,
     IN UINT64       TimeStamp,
     IN UINT32       Identifier)
    );

  MOCK_FUNCTION_DECLARATION (
    RETURN_STATUS,
    EndPerformanceMeasurementEx,
    (IN CONST VOID   *Handle   OPTIONAL,
     IN CONST CHAR8  *Token    OPTIONAL,
     IN CONST CHAR8  *Module   OPTIONAL,
     IN UINT64       TimeStamp,
     IN UINT32       Identifier)
    );

  MOCK_FUNCTION_DECLARATION (
    UINTN,
    GetPerformanceMeasurementEx,
    (IN  UINTN        LogEntryKey,
     OUT CONST VOID   **Handle,
     OUT CONST CHAR8  **Token,
     OUT CONST CHAR8  **Module,
     OUT UINT64       *StartTimeStamp,
     OUT UINT64       *EndTimeStamp,
     OUT UINT32       *Identifier)
    );

  MOCK_FUNCTION_DECLARATION (
    BOOLEAN,
    PerformanceMeasurementEnabled,
    ()
    );

  MOCK_FUNCTION_DECLARATION (
    BOOLEAN,
    LogPerformanceMeasurementEnabled,
    (IN  CONST UINTN  Type)
    );

  MOCK_FUNCTION_DECLARATION (
    RETURN_STATUS,
    LogPerformanceMeasurement,
    (IN CONST VOID   *CallerIdentifier  OPTIONAL,
     IN CONST VOID   *Guid     OPTIONAL,
     IN CONST CHAR8  *String   OPTIONAL,
     IN UINT64       Address   OPTIONAL,
     IN UINT32       Identifier)
    );
};