sdbusplus/test/sdbus_mock.hpp

#pragma once

#include "gmock/gmock.h"

#include <sdbusplus/sdbus.hpp>

namespace sdbusplus
{

class SdBusMock : public SdBusInterface
{
  public:
    virtual ~SdBusMock(){};

    MOCK_METHOD3(sd_bus_add_object_manager,
                 int(sd_bus *, sd_bus_slot **, const char *));
    MOCK_METHOD6(sd_bus_add_object_vtable,
                 int(sd_bus *, sd_bus_slot **, const char *, const char *,
                     const sd_bus_vtable *, void *));
    MOCK_METHOD3(sd_bus_attach_event, int(sd_bus *, sd_event *, int));
    MOCK_METHOD5(sd_bus_call, int(sd_bus *, sd_bus_message *, uint64_t,
                                  sd_bus_error *, sd_bus_message **));
    MOCK_METHOD1(sd_bus_detach_event, int(sd_bus *));

    MOCK_METHOD3(sd_bus_emit_interfaces_added_strv,
                 int(sd_bus *, const char *, char **));
    MOCK_METHOD3(sd_bus_emit_interfaces_removed_strv,
                 int(sd_bus *, const char *, char **));
    MOCK_METHOD2(sd_bus_emit_object_added, int(sd_bus *, const char *));
    MOCK_METHOD2(sd_bus_emit_object_removed, int(sd_bus *, const char *));
    MOCK_METHOD4(sd_bus_emit_properties_changed_strv,
                 int(sd_bus *, const char *, const char *, char **));

    MOCK_METHOD1(sd_bus_get_event, sd_event *(sd_bus *));
    MOCK_METHOD2(sd_bus_get_unique_name, int(sd_bus *, const char **));

    MOCK_METHOD3(sd_bus_list_names, int(sd_bus *, char ***, char ***));

    MOCK_METHOD3(sd_bus_message_append_basic,
                 int(sd_bus_message *, char, const void *));
    MOCK_METHOD2(sd_bus_message_at_end, int(sd_bus_message *, int));
    MOCK_METHOD1(sd_bus_message_close_container, int(sd_bus_message *));
    MOCK_METHOD3(sd_bus_message_enter_container,
                 int(sd_bus_message *, char, const char *));
    MOCK_METHOD1(sd_bus_message_exit_container, int(sd_bus_message *));

    MOCK_METHOD1(sd_bus_message_get_bus, sd_bus *(sd_bus_message *));
    MOCK_METHOD2(sd_bus_message_get_cookie, int(sd_bus_message *, uint64_t *));
    MOCK_METHOD1(sd_bus_message_get_destination,
                 const char *(sd_bus_message *));
    MOCK_METHOD1(sd_bus_message_get_interface, const char *(sd_bus_message *));
    MOCK_METHOD1(sd_bus_message_get_member, const char *(sd_bus_message *));
    MOCK_METHOD1(sd_bus_message_get_path, const char *(sd_bus_message *));
    MOCK_METHOD1(sd_bus_message_get_sender, const char *(sd_bus_message *));
    MOCK_METHOD2(sd_bus_message_get_signature,
                 const char *(sd_bus_message *, int));

    MOCK_METHOD3(sd_bus_message_is_method_call,
                 int(sd_bus_message *, const char *, const char *));
    MOCK_METHOD2(sd_bus_message_is_method_error,
                 int(sd_bus_message *, const char *));
    MOCK_METHOD3(sd_bus_message_is_signal,
                 int(sd_bus_message *, const char *, const char *));

    MOCK_METHOD6(sd_bus_message_new_method_call,
                 int(sd_bus *, sd_bus_message **, const char *, const char *,
                     const char *, const char *));

    MOCK_METHOD2(sd_bus_message_new_method_return,
                 int(sd_bus_message *, sd_bus_message **));

    MOCK_METHOD5(sd_bus_message_new_signal,
                 int(sd_bus *, sd_bus_message **, const char *, const char *,
                     const char *));

    MOCK_METHOD3(sd_bus_message_open_container,
                 int(sd_bus_message *, char, const char *));

    MOCK_METHOD3(sd_bus_message_read_basic,
                 int(sd_bus_message *, char, void *));
    MOCK_METHOD1(sd_bus_message_ref, sd_bus_message *(sd_bus_message *));

    MOCK_METHOD2(sd_bus_message_skip, int(sd_bus_message *, const char *));
    MOCK_METHOD3(sd_bus_message_verify_type,
                 int(sd_bus_message *, char, const char *));

    MOCK_METHOD2(sd_bus_process, int(sd_bus *, sd_bus_message **));
    MOCK_METHOD1(sd_bus_ref, sd_bus *(sd_bus *));
    MOCK_METHOD3(sd_bus_request_name, int(sd_bus *, const char *, uint64_t));
    MOCK_METHOD3(sd_bus_send, int(sd_bus *, sd_bus_message *, uint64_t *));
    MOCK_METHOD1(sd_bus_unref, sd_bus *(sd_bus *));
    MOCK_METHOD2(sd_bus_wait, int(sd_bus *, uint64_t));
};

bus::bus get_mocked_new(SdBusMock *sdbus)
{
    using ::testing::IsNull;
    using ::testing::Return;

    EXPECT_CALL(*sdbus, sd_bus_ref(IsNull())).WillRepeatedly(Return(nullptr));
    bus::bus bus_mock(nullptr, sdbus);

    return bus_mock;
}

} // namespace sdbusplus