#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <sdeventplus/source.hpp>
#include <sdeventplus/test/sdevent.hpp>

namespace sdeventplus
{
namespace
{

using testing::Return;

class SourceTest : public testing::Test
{
  protected:
    testing::StrictMock<SdEventMock> mock;
    sd_event_source *const expected_source =
        reinterpret_cast<sd_event_source *>(1234);
};

TEST_F(SourceTest, NewSourceRef)
{
    EXPECT_CALL(mock, sd_event_source_ref(expected_source))
        .WillOnce(Return(expected_source));
    Source source(expected_source, &mock);

    EXPECT_CALL(mock, sd_event_source_unref(expected_source))
        .WillOnce(Return(nullptr));
}

TEST_F(SourceTest, NewSourceNoRef)
{
    Source source(expected_source, std::false_type(), &mock);

    EXPECT_CALL(mock, sd_event_source_unref(expected_source))
        .WillOnce(Return(nullptr));
}

} // namespace
} // namespace sdeventplus