mirror of
https://github.com/dragonflydb/dragonfly.git
synced 2025-05-11 10:25:47 +02:00
16b6b11cf2
* feat(server): Implement MOVE command Signed-off-by: Vladislav Oleshko <vlad@dragonflydb.io> Co-authored-by: Vladislav Oleshko <vlad@dragonflydb.io>
96 lines
2.8 KiB
C++
96 lines
2.8 KiB
C++
// Copyright 2021, Roman Gershman. All rights reserved.
|
|
// See LICENSE for licensing terms.
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <gmock/gmock.h>
|
|
|
|
#include "facade/dragonfly_connection.h"
|
|
#include "facade/memcache_parser.h"
|
|
#include "facade/redis_parser.h"
|
|
#include "io/io.h"
|
|
#include "server/conn_context.h"
|
|
#include "server/main_service.h"
|
|
#include "util/proactor_pool.h"
|
|
|
|
namespace dfly {
|
|
using namespace facade;
|
|
|
|
class TestConnection : public facade::Connection {
|
|
public:
|
|
TestConnection(Protocol protocol);
|
|
|
|
void SendMsgVecAsync(const PubMessage& pmsg, util::fibers_ext::BlockingCounter bc) final;
|
|
|
|
std::vector<PubMessage> messages;
|
|
|
|
private:
|
|
std::vector<std::unique_ptr<std::string>> backing_str_;
|
|
};
|
|
|
|
class BaseFamilyTest : public ::testing::Test {
|
|
protected:
|
|
BaseFamilyTest();
|
|
~BaseFamilyTest();
|
|
|
|
static void SetUpTestSuite();
|
|
|
|
void SetUp() override;
|
|
void TearDown() override;
|
|
|
|
protected:
|
|
class TestConnWrapper;
|
|
|
|
RespExpr Run(std::initializer_list<const std::string_view> list) {
|
|
return Run(ArgSlice{list.begin(), list.size()});
|
|
}
|
|
|
|
RespExpr Run(ArgSlice list);
|
|
|
|
RespExpr Run(std::string_view id, ArgSlice list);
|
|
|
|
using MCResponse = std::vector<std::string>;
|
|
MCResponse RunMC(MemcacheParser::CmdType cmd_type, std::string_view key, std::string_view value,
|
|
uint32_t flags = 0, std::chrono::seconds ttl = std::chrono::seconds{});
|
|
MCResponse RunMC(MemcacheParser::CmdType cmd_type, std::string_view key = std::string_view{});
|
|
MCResponse GetMC(MemcacheParser::CmdType cmd_type, std::initializer_list<std::string_view> list);
|
|
|
|
int64_t CheckedInt(std::initializer_list<std::string_view> list);
|
|
|
|
bool IsLocked(DbIndex db_index, std::string_view key) const;
|
|
ConnectionContext::DebugInfo GetDebugInfo(const std::string& id) const;
|
|
|
|
ConnectionContext::DebugInfo GetDebugInfo() const {
|
|
return GetDebugInfo("IO0");
|
|
}
|
|
|
|
TestConnWrapper* AddFindConn(Protocol proto, std::string_view id);
|
|
static std::vector<std::string> StrArray(const RespExpr& expr);
|
|
|
|
// ts is ms
|
|
void UpdateTime(uint64_t ms);
|
|
|
|
// Wait for a locked key to unlock. Aborts after timeout seconds passed.
|
|
void WaitUntilLocked(DbIndex db_index, std::string_view key, double timeout = 3);
|
|
|
|
std::string GetId() const;
|
|
size_t SubscriberMessagesLen(std::string_view conn_id) const;
|
|
|
|
// Returns message parts as returned by RESP:
|
|
// pmessage, pattern, channel, message
|
|
facade::Connection::PubMessage GetPublishedMessage(std::string_view conn_id, size_t index) const;
|
|
|
|
std::unique_ptr<util::ProactorPool> pp_;
|
|
std::unique_ptr<Service> service_;
|
|
unsigned num_threads_ = 3;
|
|
|
|
absl::flat_hash_map<std::string, std::unique_ptr<TestConnWrapper>> connections_;
|
|
::boost::fibers::mutex mu_;
|
|
ConnectionContext::DebugInfo last_cmd_dbg_info_;
|
|
uint64_t expire_now_;
|
|
std::vector<RespVec*> resp_vec_;
|
|
bool single_response_ = true;
|
|
};
|
|
|
|
} // namespace dfly
|