4acad5795b
Summary: For proper client interaction, we need to expose the (term_id, log_index) pair for the transaction that's about to be replicated and we need to be able to retrieve the status of a transaction defined by that pair. Transaction status can be one of the following: 1) REPLICATED (self-explanatory) 2) WAITING (waiting for replication) 3) ABORTED (self-explanatory) 4) INVALID (received request with either invalid term_id or invalid log_index) Reviewers: mferencevic Reviewed By: mferencevic Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D2201
77 lines
1.8 KiB
C++
77 lines
1.8 KiB
C++
#include "gtest/gtest.h"
|
|
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "durability/single_node_ha/state_delta.hpp"
|
|
#include "raft/raft_interface.hpp"
|
|
#include "transactions/single_node_ha/engine.hpp"
|
|
#include "transactions/transaction.hpp"
|
|
|
|
using namespace tx;
|
|
|
|
class RaftMock final : public raft::RaftInterface {
|
|
public:
|
|
raft::DeltaStatus Emplace(const database::StateDelta &delta) override {
|
|
log_[delta.transaction_id].emplace_back(std::move(delta));
|
|
return {true, std::nullopt};
|
|
}
|
|
|
|
bool SafeToCommit(const tx::TransactionId &) override {
|
|
return true;
|
|
}
|
|
|
|
bool IsLeader() override { return true; }
|
|
|
|
uint64_t TermId() override { return 1; }
|
|
|
|
raft::TxStatus TransactionStatus(uint64_t term_id,
|
|
uint64_t log_index) override {
|
|
return raft::TxStatus::REPLICATED;
|
|
}
|
|
|
|
std::vector<database::StateDelta> GetLogForTx(
|
|
const tx::TransactionId &tx_id) {
|
|
return log_[tx_id];
|
|
}
|
|
|
|
std::mutex &WithLock() override { return lock_; }
|
|
|
|
private:
|
|
std::unordered_map<tx::TransactionId, std::vector<database::StateDelta>> log_;
|
|
std::mutex lock_;
|
|
};
|
|
|
|
TEST(Engine, Reset) {
|
|
RaftMock raft;
|
|
Engine engine{&raft};
|
|
|
|
auto t0 = engine.Begin();
|
|
EXPECT_EQ(t0->id_, 1);
|
|
engine.Commit(*t0);
|
|
|
|
engine.Reset();
|
|
|
|
auto t1 = engine.Begin();
|
|
EXPECT_EQ(t1->id_, 1);
|
|
engine.Commit(*t1);
|
|
}
|
|
|
|
TEST(Engine, TxStateDelta) {
|
|
RaftMock raft;
|
|
Engine engine{&raft};
|
|
|
|
auto t0 = engine.Begin();
|
|
tx::TransactionId tx_id = t0->id_;
|
|
engine.Commit(*t0);
|
|
|
|
auto t0_log = raft.GetLogForTx(tx_id);
|
|
EXPECT_EQ(t0_log.size(), 2);
|
|
|
|
using Type = enum database::StateDelta::Type;
|
|
EXPECT_EQ(t0_log[0].type, Type::TRANSACTION_BEGIN);
|
|
EXPECT_EQ(t0_log[0].transaction_id, tx_id);
|
|
EXPECT_EQ(t0_log[1].type, Type::TRANSACTION_COMMIT);
|
|
EXPECT_EQ(t0_log[1].transaction_id, tx_id);
|
|
}
|