#pragma once

#include <vector>
#include <string>
#include <optional>

#include <gmock/gmock.h>
#include <mw/error.hpp>

#include "data.hpp"

class DataSourceMock : public DataSourceInterface
{
public:
    ~DataSourceMock() override = default;

    MOCK_METHOD(mw::E<int64_t>, getSchemaVersion, (), (const override));
    MOCK_METHOD(mw::E<std::optional<User>>, userByOpenIDUID,
                (const std::string& uid), (const override));
    MOCK_METHOD(mw::E<int64_t>, addUser, (User&& u), (const override));
    MOCK_METHOD(mw::E<std::optional<User>>, userByID, (const int64_t),
                (const override));
    MOCK_METHOD(mw::E<std::optional<User>>, userByName,
                (const std::string& name), (const override));
    MOCK_METHOD(mw::E<std::optional<LinkItem>>, itemByID, (int64_t id),
                (const override));
    MOCK_METHOD(mw::E<std::vector<LinkItem>>, itemsByParent, (int64_t parent),
                (const override));
    MOCK_METHOD(mw::E<std::vector<LinkItem>>, itemsTopLevelByUser,
                (int64_t user_id), (const override));
    MOCK_METHOD(mw::E<int64_t>, addLink, (LinkItem&& link), (const override));

protected:
    mw::E<void> setSchemaVersion([[maybe_unused]] int64_t v) const override
    {
        return {};
    }
};