47 lines
1.2 KiB
C++
47 lines
1.2 KiB
C++
|
#define NDEBUG
|
||
|
|
#include "bolt_common.hpp"
|
||
|
|
|
||
|
|
#include "communication/bolt/v1/encoder/chunked_buffer.hpp"
|
||
|
|
|
||
|
|
|
||
|
|
constexpr const int SIZE = 131072;
|
||
|
|
uint8_t data[SIZE];
|
||
|
|
|
||
|
|
|
||
|
|
void verify_output(std::vector<uint8_t>& output, const uint8_t* data, uint64_t size) {
|
||
|
|
uint64_t len = 0, pos = 0;
|
||
|
|
uint8_t tail[2] = { 0, 0 };
|
||
|
|
uint16_t head;
|
||
|
|
while (size > 0) {
|
||
|
|
head = len = std::min(size, communication::bolt::CHUNK_SIZE);
|
||
|
|
head = bswap(head);
|
||
|
|
check_output(output, reinterpret_cast<uint8_t *>(&head), sizeof(head), false);
|
||
|
|
check_output(output, data + pos, len, false);
|
||
|
|
check_output(output, tail, 2, false);
|
||
|
|
size -= len;
|
||
|
|
pos += len;
|
||
|
|
}
|
||
|
|
check_output(output, nullptr, 0, true);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(Bolt, ChunkedBuffer) {
|
||
|
|
TestSocket socket(10);
|
||
|
|
communication::bolt::ChunkedBuffer<TestSocket> chunked_buffer(socket);
|
||
|
|
std::vector<uint8_t>& output = socket.output;
|
||
|
|
|
||
|
|
for (int i = 0; i <= SIZE; i += 16) {
|
||
|
|
chunked_buffer.Write(data, i);
|
||
|
|
chunked_buffer.Flush();
|
||
|
|
verify_output(output, data, i);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int main(int argc, char** argv) {
|
||
|
|
initialize_data(data, SIZE);
|
||
|
|
logging::init_sync();
|
||
|
|
logging::log->pipe(std::make_unique<Stdout>());
|
||
|
|
::testing::InitGoogleTest(&argc, argv);
|
||
|
|
return RUN_ALL_TESTS();
|
||
|
|
}
|