f1a8d7cd3d
Summary: The network layer now has a `Session` that handles all things that should be done before the `Execute` method is called on sessions. Also, all sessions now communicate using streams instead of holding the input buffer and writing to the `Socket`. This design will allow implementation of a SSL middleware. Reviewers: buda, dgleich Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1314
59 lines
1.9 KiB
C++
59 lines
1.9 KiB
C++
#include "bolt_common.hpp"
|
|
|
|
#include "communication/bolt/v1/encoder/chunked_encoder_buffer.hpp"
|
|
#include "communication/bolt/v1/encoder/encoder.hpp"
|
|
#include "communication/bolt/v1/encoder/result_stream.hpp"
|
|
#include "query/typed_value.hpp"
|
|
|
|
using query::TypedValue;
|
|
|
|
using BufferT = communication::bolt::ChunkedEncoderBuffer<TestOutputStream>;
|
|
using EncoderT = communication::bolt::Encoder<BufferT>;
|
|
using ResultStreamT = communication::bolt::ResultStream<EncoderT>;
|
|
|
|
const uint8_t header_output[] =
|
|
"\x00\x29\xB1\x70\xA1\x86\x66\x69\x65\x6C\x64\x73\x9A\x82\x61\x61\x82\x62"
|
|
"\x62\x82\x63\x63\x82\x64\x64\x82\x65\x65\x82\x66\x66\x82\x67\x67\x82\x68"
|
|
"\x68\x82\x69\x69\x82\x6A\x6A\x00\x00";
|
|
const uint8_t result_output[] =
|
|
"\x00\x0A\xB1\x71\x92\x05\x85\x68\x65\x6C\x6C\x6F\x00\x00";
|
|
const uint8_t summary_output[] =
|
|
"\x00\x0C\xB1\x70\xA1\x87\x63\x68\x61\x6E\x67\x65\x64\x0A\x00\x00";
|
|
|
|
TEST(Bolt, ResultStream) {
|
|
TestOutputStream output_stream;
|
|
BufferT buffer(output_stream);
|
|
EncoderT encoder(buffer);
|
|
ResultStreamT result_stream(encoder);
|
|
std::vector<uint8_t> &output = output_stream.output;
|
|
|
|
std::vector<std::string> headers;
|
|
for (int i = 0; i < 10; ++i)
|
|
headers.push_back(std::string(2, (char)('a' + i)));
|
|
|
|
result_stream.Header(headers);
|
|
buffer.FlushFirstChunk();
|
|
PrintOutput(output);
|
|
CheckOutput(output, header_output, 45);
|
|
|
|
std::vector<TypedValue> result{TypedValue(5),
|
|
TypedValue(std::string("hello"))};
|
|
result_stream.Result(result);
|
|
buffer.Flush();
|
|
PrintOutput(output);
|
|
CheckOutput(output, result_output, 14);
|
|
|
|
std::map<std::string, TypedValue> summary;
|
|
summary.insert(std::make_pair(std::string("changed"), TypedValue(10)));
|
|
result_stream.Summary(summary);
|
|
buffer.Flush();
|
|
PrintOutput(output);
|
|
CheckOutput(output, summary_output, 16);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
google::InitGoogleLogging(argv[0]);
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|