oceanbase/unittest/election/mock_election.h

/**
 * Copyright (c) 2021 OceanBase
 * OceanBase CE is licensed under Mulan PubL v2.
 * You can use this software according to the terms and conditions of the Mulan PubL v2.
 * You may obtain a copy of Mulan PubL v2 at:
 *          http://license.coscl.org.cn/MulanPubL-2.0
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PubL v2 for more details.
 */

#ifndef _OB_MOCK_ELECTION_H
#define _OB_MOCK_ELECTION_H

#include "election/ob_election.h"

namespace oceanbase {
namespace unittest {
class MockElection : public election::ObIElection {
public:
  virtual int init(const common::ObPartitionKey&, const common::ObAddr&, election::ObIElectionRpc*,
      common::ObTimeWheel*, const int64_t, election::ObIElectionCallback*, election::ObIElectionGroupMgr*,
      election::ObElectionEventHistoryArray*)
  {
    return 0;
  }
  virtual void destroy()
  {}
  virtual int start()
  {
    return 0;
  }
  virtual int stop()
  {
    return 0;
  }

public:
  virtual int handle_devote_prepare(const election::ObElectionMsgDEPrepare&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_devote_vote(const election::ObElectionMsgDEVote&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_devote_success(const election::ObElectionMsgDESuccess&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_vote_prepare(const election::ObElectionMsgPrepare&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_vote_vote(const election::ObElectionMsgVote&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_vote_success(const election::ObElectionMsgSuccess&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_query_leader(const election::ObElectionQueryLeader&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }
  virtual int handle_query_leader_response(const election::ObElectionQueryLeaderResponse&, obrpc::ObElectionRpcResult&)
  {
    return 0;
  }

public:
  virtual int set_candidate(const int64_t, const common::ObMemberList&, const int64_t)
  {
    return 0;
  }
  virtual int change_leader_async(const common::ObAddr&, common::ObTsWindows&)
  {
    return 0;
  }
  virtual int change_leader_to_self() override
  {
    return 0;
  }
  virtual int force_leader_async()
  {
    return 0;
  }
  virtual int get_curr_candidate(common::ObMemberList&) const
  {
    return 0;
  }
  virtual int get_valid_candidate(common::ObMemberList&) const
  {
    return 0;
  }
  virtual int get_leader(common::ObAddr&, common::ObAddr&, int64_t&, bool&) const
  {
    return 0;
  }
  virtual int get_leader(common::ObAddr&, int64_t&, bool&, common::ObTsWindows&) const
  {
    return 0;
  }
  virtual int get_current_leader(common::ObAddr&) const
  {
    return 0;
  }
  virtual int get_election_info(election::ObElectionInfo&) const
  {
    return 0;
  }
  virtual void leader_revoke(const uint32_t)
  {}
  virtual int inc_replica_num()
  {
    return 0;
  }
  virtual int dec_replica_num()
  {
    return 0;
  }
  virtual int set_replica_num(const int64_t)
  {
    return 0;
  }
  virtual int64_t get_election_time_offset() const
  {
    return 0;
  }
  virtual int set_offline()
  {
    return 0;
  }

public:
  virtual int leader_takeover(const common::ObAddr&, const int64_t, int64_t&)
  {
    return 0;
  }
  virtual int64_t get_current_ts() const
  {
    return ts_;
  }
  virtual int get_priority(election::ObElectionPriority&) const
  {
    return 0;
  }
  virtual int get_timestamp(int64_t&, common::ObAddr&) const
  {
    return 0;
  }
  virtual int64_t get_last_leader_revoke_time() const
  {
    return 0;
  }

public:
  // for election group
  virtual int move_out_election_group(const election::ObElectionGroupId&)
  {
    return 0;
  }
  virtual int move_into_election_group(const election::ObElectionGroupId&)
  {
    return 0;
  }
  virtual const election::lease_t& get_leader_lease() const
  {
    return lease_;
  }
  virtual int64_t get_replica_num() const
  {
    return 0;
  }
  virtual uint64_t get_eg_id_hash() const
  {
    return 0;
  }
  virtual bool is_offline() const
  {
    return 0;
  }
  void set_current_ts(int64_t ts)
  {
    ts_ = ts;
  }

private:
  election::lease_t lease_;
  int64_t ts_;
};
}  // namespace unittest
}  // namespace oceanbase
#endif