99 lines
3.0 KiB
C++
99 lines
3.0 KiB
C++
/************************************************************************
|
|
*
|
|
* Copyright (c) 2019 Alibaba.com, Inc. All Rights Reserved
|
|
* $Id: consensus-replicate-with-cache-log-t.cc,v 1.0 03/14/2019 11:06:00 AM aili.xp(aili.xp@alibaba-inc.com) $
|
|
*
|
|
************************************************************************/
|
|
|
|
/**
|
|
* @file consensus-replicate-with-cache-log-t.cc
|
|
* @author aili.xp(aili.xp@alibaba-inc.com)
|
|
* @date 03/14/2019 11:06:00 AM
|
|
* @version 1.0
|
|
* @brief unit test for alisql::Consensus
|
|
*
|
|
**/
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
#include <memory>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "paxos.h"
|
|
#include "paxos.pb.h"
|
|
#include "file_paxos_log.h"
|
|
#include "files.h"
|
|
|
|
using namespace alisql;
|
|
|
|
class FilePaxosLogTest : public FilePaxosLog {
|
|
public:
|
|
FilePaxosLogTest(const std::string &dataDir, LogTypeT type= LTMem):FilePaxosLog(dataDir, type) { }
|
|
void setLastCachedLogIndex(uint64_t ci) { cache_index = ci; }
|
|
virtual uint64_t getLastCachedLogIndex() { return cache_index; }
|
|
uint64_t cache_index = 0;
|
|
};
|
|
|
|
TEST(consensus, replicate_with_cache_log)
|
|
{
|
|
std::vector<std::string> strConfig;
|
|
strConfig.emplace_back("127.0.0.1:11001");
|
|
strConfig.emplace_back("127.0.0.1:11002");
|
|
strConfig.emplace_back("127.0.0.1:11003");
|
|
|
|
const uint64_t timeout= 2000;
|
|
std::string dir1 = std::string("paxosLogTestDir")+strConfig[0];
|
|
std::shared_ptr<PaxosLog> rlog1= std::make_shared<FilePaxosLogTest>(dir1, FilePaxosLog::LogType::LTMem);
|
|
Paxos *paxos1= new Paxos(timeout, rlog1);
|
|
paxos1->init(strConfig, 1);
|
|
std::string dir2 = std::string("paxosLogTestDir")+strConfig[1];
|
|
std::shared_ptr<PaxosLog> rlog2= std::make_shared<FilePaxosLogTest>(dir2, FilePaxosLog::LogType::LTMem);
|
|
Paxos *paxos2= new Paxos(timeout, rlog2);
|
|
paxos2->init(strConfig, 2);
|
|
std::string dir3 = std::string("paxosLogTestDir")+strConfig[2];
|
|
std::shared_ptr<PaxosLog> rlog3= std::make_shared<FilePaxosLogTest>(dir3, FilePaxosLog::LogType::LTMem);
|
|
Paxos *paxos3= new Paxos(timeout, rlog3);
|
|
paxos3->init(strConfig, 3);
|
|
|
|
sleep(3);
|
|
paxos1->requestVote();
|
|
sleep(1);
|
|
|
|
EXPECT_EQ(paxos1->getState(), Paxos::LEADER);
|
|
uint64_t base = rlog1->getLastLogIndex();
|
|
|
|
paxos1->setReplicateWithCacheLog(true);
|
|
|
|
LogEntry le;
|
|
le.set_optype(0);
|
|
le.set_ikey("a");
|
|
|
|
paxos1->replicateLog(le);
|
|
EXPECT_EQ(rlog1->getLastLogIndex(), base + 1);
|
|
sleep(2);
|
|
EXPECT_TRUE(rlog2->getLastLogIndex() == base && rlog3->getLastLogIndex() == base);
|
|
|
|
((FilePaxosLogTest *)rlog1.get())->setLastCachedLogIndex(rlog1->getLastLogIndex());
|
|
paxos1->appendLog(false);
|
|
sleep(2);
|
|
EXPECT_TRUE(rlog2->getLastLogIndex() == base + 1 && rlog3->getLastLogIndex() == base + 1);
|
|
|
|
paxos1->setReplicateWithCacheLog(false);
|
|
|
|
LogEntry le2;
|
|
le2.set_optype(0);
|
|
le2.set_ikey("a");
|
|
paxos1->replicateLog(le2);
|
|
sleep(2);
|
|
EXPECT_TRUE(rlog2->getLastLogIndex() == base + 2 && rlog3->getLastLogIndex() == base + 2);
|
|
|
|
delete paxos3;
|
|
delete paxos2;
|
|
delete paxos1;
|
|
|
|
deleteDir(dir1.c_str());
|
|
deleteDir(dir2.c_str());
|
|
deleteDir(dir3.c_str());
|
|
}
|
|
|