813a3b9eed
Summary: Moved server and worker from bolt to communication. Started templatization. Started removal of Bolt class. Removed unnecessary files from network. Converted states to template functions. Bolt::Session is now a template. Merge remote-tracking branch 'origin/dev' into mg_refactor_network Merged bolt_serializer.cpp into hpp. Removed obsolete include. Initial version of bolt session unit test. Uncommented leftover log commands. Reimplemented io::Socket. Added client-stress.sh script. Reviewers: dgleich, buda Reviewed By: dgleich, buda Subscribers: pullbot, mferencevic, buda Differential Revision: https://phabricator.memgraph.io/D64
118 lines
3.4 KiB
C++
118 lines
3.4 KiB
C++
#include <array>
|
|
#include <cassert>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "logging/streams/stdout.hpp"
|
|
|
|
#include "communication/bolt/v1/serialization/record_stream.hpp"
|
|
#include "communication/bolt/v1/session.hpp"
|
|
#include "dbms/dbms.hpp"
|
|
#include "query/engine.hpp"
|
|
|
|
class TestSocket {
|
|
public:
|
|
TestSocket(int socket) : socket(socket) {}
|
|
TestSocket(const TestSocket& s) : socket(s.id()){};
|
|
TestSocket(TestSocket&& other) { *this = std::forward<TestSocket>(other); }
|
|
|
|
TestSocket& operator=(TestSocket&& other) {
|
|
this->socket = other.socket;
|
|
other.socket = -1;
|
|
return *this;
|
|
}
|
|
|
|
void Close() { socket = -1; }
|
|
bool IsOpen() { return socket != -1; }
|
|
|
|
int id() const { return socket; }
|
|
|
|
int Write(const std::string& str) { return Write(str.c_str(), str.size()); }
|
|
int Write(const char* data, size_t len) {
|
|
return Write(reinterpret_cast<const uint8_t*>(data), len);
|
|
}
|
|
int Write(const uint8_t* data, size_t len) {
|
|
for (int i = 0; i < len; ++i) output.push_back(data[i]);
|
|
return len;
|
|
}
|
|
|
|
std::vector<uint8_t> output;
|
|
|
|
protected:
|
|
int socket;
|
|
};
|
|
|
|
const uint8_t handshake_req[] =
|
|
"\x60\x60\xb0\x17\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
"\x00\x00";
|
|
const uint8_t handshake_resp[] = "\x00\x00\x00\x01";
|
|
const uint8_t init_req[] =
|
|
"\x00\x3f\xb2\x01\xd0\x15\x6c\x69\x62\x6e\x65\x6f\x34\x6a\x2d\x63\x6c\x69"
|
|
"\x65\x6e\x74\x2f\x31\x2e\x32\x2e\x31\xa3\x86\x73\x63\x68\x65\x6d\x65\x85"
|
|
"\x62\x61\x73\x69\x63\x89\x70\x72\x69\x6e\x63\x69\x70\x61\x6c\x80\x8b\x63"
|
|
"\x72\x65\x64\x65\x6e\x74\x69\x61\x6c\x73\x80\x00\x00";
|
|
const uint8_t init_resp[] = "\x00\x03\xb1\x70\xa0\x00\x00";
|
|
const uint8_t run_req[] =
|
|
"\x00\x26\xb2\x10\xd0\x21\x43\x52\x45\x41\x54\x45\x20\x28\x6e\x20\x7b\x6e"
|
|
"\x61\x6d\x65\x3a\x20\x32\x39\x33\x38\x33\x7d\x29\x20\x52\x45\x54\x55\x52"
|
|
"\x4e\x20\x6e\xa0\x00\x00";
|
|
|
|
void print_output(std::vector<uint8_t>& output) {
|
|
fprintf(stderr, "output: ");
|
|
for (int i = 0; i < output.size(); ++i) {
|
|
fprintf(stderr, "%02X ", output[i]);
|
|
}
|
|
fprintf(stderr, "\n");
|
|
}
|
|
|
|
void check_output(std::vector<uint8_t>& output, const uint8_t* data,
|
|
uint64_t len) {
|
|
EXPECT_EQ(len, output.size());
|
|
for (int i = 0; i < len; ++i) {
|
|
EXPECT_EQ(output[i], data[i]);
|
|
}
|
|
output.clear();
|
|
}
|
|
|
|
TEST(Bolt, Session) {
|
|
Dbms dbms;
|
|
TestSocket socket(10);
|
|
QueryEngine<bolt::RecordStream<TestSocket>> query_engine;
|
|
bolt::Session<TestSocket> session(std::move(socket), dbms, query_engine);
|
|
std::vector<uint8_t>& output = session.socket.output;
|
|
|
|
// execute handshake
|
|
session.execute(handshake_req, 20);
|
|
ASSERT_EQ(session.state, bolt::INIT);
|
|
print_output(output);
|
|
check_output(output, handshake_resp, 4);
|
|
|
|
// execute init
|
|
session.execute(init_req, 67);
|
|
ASSERT_EQ(session.state, bolt::EXECUTOR);
|
|
print_output(output);
|
|
check_output(output, init_resp, 7);
|
|
|
|
// execute run
|
|
session.execute(run_req, 42);
|
|
// TODO: query engine doesn't currently work,
|
|
// we should test the query output here and the next state
|
|
// ASSERT_EQ(session.state, bolt::EXECUTOR);
|
|
// print_output(output);
|
|
// check_output(output, run_resp, len);
|
|
|
|
// TODO: add more tests
|
|
|
|
session.close();
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
logging::init_sync();
|
|
logging::log->pipe(std::make_unique<Stdout>());
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|