src/sdeventplus/test/sdevent.hpp

#pragma once

#include <gmock/gmock.h>
#include <sdeventplus/internal/sdevent.hpp>
#include <systemd/sd-event.h>

namespace sdeventplus
{
namespace test
{

class SdEventMock : public internal::SdEvent
{
  public:
    MOCK_CONST_METHOD1(sd_event_default, int(sd_event**));
    MOCK_CONST_METHOD1(sd_event_new, int(sd_event**));
    MOCK_CONST_METHOD1(sd_event_ref, sd_event*(sd_event*));
    MOCK_CONST_METHOD1(sd_event_unref, sd_event*(sd_event*));

    MOCK_CONST_METHOD1(sd_event_prepare, int(sd_event*));
    MOCK_CONST_METHOD2(sd_event_wait, int(sd_event*, uint64_t));
    MOCK_CONST_METHOD1(sd_event_dispatch, int(sd_event*));
    MOCK_CONST_METHOD2(sd_event_run, int(sd_event*, uint64_t));
    MOCK_CONST_METHOD1(sd_event_loop, int(sd_event*));
    MOCK_CONST_METHOD2(sd_event_exit, int(sd_event*, int));

    MOCK_CONST_METHOD3(sd_event_now, int(sd_event*, clockid_t, uint64_t*));

    MOCK_CONST_METHOD2(sd_event_get_exit_code, int(sd_event*, int*));
    MOCK_CONST_METHOD1(sd_event_get_watchdog, int(sd_event*));
    MOCK_CONST_METHOD2(sd_event_set_watchdog, int(sd_event*, int b));

    MOCK_CONST_METHOD1(sd_event_source_ref, sd_event_source*(sd_event_source*));
    MOCK_CONST_METHOD1(sd_event_source_unref,
                       sd_event_source*(sd_event_source*));

    MOCK_CONST_METHOD2(sd_event_source_set_userdata,
                       void*(sd_event_source*, void*));

    MOCK_CONST_METHOD2(sd_event_source_get_description,
                       int(sd_event_source*, const char**));
    MOCK_CONST_METHOD2(sd_event_source_set_description,
                       int(sd_event_source*, const char*));
    MOCK_CONST_METHOD2(sd_event_source_set_prepare,
                       int(sd_event_source*, sd_event_handler_t));
    MOCK_CONST_METHOD1(sd_event_source_get_pending, int(sd_event_source*));
    MOCK_CONST_METHOD2(sd_event_source_get_priority,
                       int(sd_event_source*, int64_t*));
    MOCK_CONST_METHOD2(sd_event_source_set_priority,
                       int(sd_event_source*, int64_t));
    MOCK_CONST_METHOD2(sd_event_source_get_enabled,
                       int(sd_event_source*, int*));
    MOCK_CONST_METHOD2(sd_event_source_set_enabled, int(sd_event_source*, int));
};

} // namespace test
} // namespace sdeventplus