requester/test/mock_mctp_discovery_handler_intf.hpp

#pragma once

#include "requester/mctp_endpoint_discovery.hpp"

#include <gmock/gmock.h>
#include <gtest/gtest.h>

namespace pldm
{

class MockManager : public pldm::MctpDiscoveryHandlerIntf
{
  public:
    MOCK_METHOD(void, handleMctpEndpoints, (const MctpInfos& mctpInfos),
                (override));
    MOCK_METHOD(void, handleRemovedMctpEndpoints, (const MctpInfos& mctpInfos),
                (override));
    MOCK_METHOD(void, updateMctpEndpointAvailability,
                (const MctpInfo& mctpInfo, Availability availability),
                (override));
};

} // namespace pldm