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 TerminusInfos& mctpInfos),
                (override));
    MOCK_METHOD(void, handleRemovedMctpEndpoints,
                (const TerminusInfos& mctpInfos), (override));
    MOCK_METHOD(void, updateMctpEndpointAvailability,
                (const MctpInfo& mctpInfo, Availability availability),
                (override));
    MOCK_METHOD(std::optional<pldm_tid_t>, allocateOrGetTid,
                (const MctpInfo& mctpInfo), (override));
    MOCK_METHOD(PldmTransport*, getTransport, (), (override));
    MOCK_METHOD(std::optional<mctp_eid_t>, getActiveEidByName,
                (const std::string& terminusName), (override));
};

} // namespace pldm