//
// Copyright (C) 2018 Codership Oy <info@codership.com>
//

#ifndef WSREP_MOCK_SERVER_CONTEXT_HPP
#define WSREP_MOCK_SERVER_CONTEXT_HPP

#include "wsrep/server_state.hpp"
#include "mock_client_state.hpp"
#include "mock_provider.hpp"

#include "wsrep/compiler.hpp"

namespace wsrep
{
    class mock_server_state : public wsrep::server_state
    {
    public:
        mock_server_state(const std::string& name,
                            const std::string& id,
                            enum wsrep::server_state::rollback_mode rollback_mode)
            : wsrep::server_state(mutex_, cond_,
                                    name, id, "", "./", rollback_mode)
            , mutex_()
            , cond_()
            , provider_(*this)
            , client_service_(provider_)
            , last_client_id_(0)
        { }
        wsrep::mock_provider& provider() const
        { return provider_; }
        wsrep::client_state* local_client_state()
        {
            return new wsrep::mock_client_state(
                *this, client_service_, ++last_client_id_,
                wsrep::client_state::m_local);
        }
        wsrep::client_state* streaming_applier_client_state()
        {
            return new wsrep::mock_client_state(
                *this, client_service_, ++last_client_id_,
                wsrep::client_state::m_high_priority);
        }
        void release_client_state(wsrep::client_state* client_state)
        {
            delete client_state;
        }

        void log_dummy_write_set(wsrep::client_state&,
                                 const wsrep::ws_meta&)
            WSREP_OVERRIDE
        {
        }
        void on_connect() WSREP_OVERRIDE { }
        void wait_until_connected() WSREP_OVERRIDE { }
        void on_view(const wsrep::view&) WSREP_OVERRIDE { }
        void on_sync() WSREP_OVERRIDE { }
        bool sst_before_init() const WSREP_OVERRIDE { return false; }
        std::string on_sst_required() WSREP_OVERRIDE { return ""; }
        void on_sst_request(const std::string&,
                            const wsrep::gtid&,
                            bool) WSREP_OVERRIDE { }
        void background_rollback(wsrep::client_state& client_state)
            WSREP_OVERRIDE
        {
            client_state.before_rollback();
            client_state.after_rollback();
        }
        // void sst_received(const wsrep_gtid_t&, int) WSREP_OVERRIDE { }
        // void on_apply(wsrep::transaction&) { }
        // void on_commit(wsrep::transaction&) { }
        wsrep::mock_client_service& client_service()
        {
            return client_service_;
        }
    private:
        wsrep::default_mutex mutex_;
        wsrep::default_condition_variable cond_;
        mutable wsrep::mock_provider provider_;
        wsrep::mock_client_service client_service_;
        unsigned long long last_client_id_;
    };
}

#endif // WSREP_MOCK_SERVER_CONTEXT_HPP