mock_crud_base.hpp

// Copyright (C) 2018 Bluzelle
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License, version 3,
// as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

#pragma once

#include <crud/crud_base.hpp>
#include <gmock/gmock.h>

namespace bzn {

    class mock_crud_base : public crud_base {
    public:
        MOCK_METHOD3(handle_request,
            void(const bzn::caller_id_t& caller_id, const database_msg& request, std::shared_ptr<bzn::session_base> session));
        MOCK_METHOD1(start,
            void(std::shared_ptr<bzn::pbft_base> pbft));
        MOCK_METHOD0(save_state,
            bool());
        MOCK_METHOD0(get_saved_state,
            std::shared_ptr<std::string>());
        MOCK_METHOD1(load_state,
            bool(const std::string&));
    };

}  // namespace bzn