1 #pragma once
2 
3 #include "requester/mctp_endpoint_discovery.hpp"
4 
5 #include <gmock/gmock.h>
6 #include <gtest/gtest.h>
7 
8 namespace pldm
9 {
10 
11 class MockManager : public pldm::MctpDiscoveryHandlerIntf
12 {
13   public:
14     MOCK_METHOD(void, handleMctpEndpoints, (const MctpInfos& mctpInfos),
15                 (override));
16     MOCK_METHOD(void, handleRemovedMctpEndpoints, (const MctpInfos& mctpInfos),
17                 (override));
18 };
19 
20 } // namespace pldm
21