62 lines
1.6 KiB
C++
62 lines
1.6 KiB
C++
#include <array>
|
|
#include <cassert>
|
|
#include <cstring>
|
|
#include <deque>
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "communication/bolt/v1/transport/chunked_decoder.hpp"
|
|
|
|
using byte = unsigned char;
|
|
|
|
void print_hex(byte x) { printf("%02X ", static_cast<byte>(x)); }
|
|
|
|
struct DummyStream {
|
|
void write(const byte *values, size_t n) {
|
|
data.insert(data.end(), values, values + n);
|
|
}
|
|
|
|
std::vector<byte> data;
|
|
};
|
|
|
|
using Decoder = bolt::ChunkedDecoder<DummyStream>;
|
|
|
|
std::vector<byte> chunks[] = {
|
|
{0x00, 0x08, 'A', ' ', 'q', 'u', 'i', 'c', 'k', ' ', 0x00, 0x06, 'b', 'r',
|
|
'o', 'w', 'n', ' '},
|
|
{0x00, 0x0A, 'f', 'o', 'x', ' ', 'j', 'u', 'm', 'p', 's', ' '},
|
|
{0x00, 0x07, 'o', 'v', 'e', 'r', ' ', 'a', ' '},
|
|
{0x00, 0x08, 'l', 'a', 'z', 'y', ' ', 'd', 'o', 'g', 0x00, 0x00}};
|
|
|
|
static constexpr size_t N = std::extent<decltype(chunks)>::value;
|
|
|
|
std::string decoded = "A quick brown fox jumps over a lazy dog";
|
|
|
|
TEST(ChunkedDecoderTest, WriteString) {
|
|
DummyStream stream;
|
|
Decoder decoder(stream);
|
|
|
|
for (size_t i = 0; i < N; ++i) {
|
|
auto &chunk = chunks[i];
|
|
logging::info("Chunk size: {}", chunk.size());
|
|
|
|
const byte *start = chunk.data();
|
|
auto finished = decoder.decode(start, chunk.size());
|
|
|
|
// break early if finished
|
|
if (finished) break;
|
|
}
|
|
|
|
// check validity
|
|
ASSERT_EQ(decoded.size(), stream.data.size());
|
|
for (size_t i = 0; i < decoded.size(); ++i)
|
|
ASSERT_EQ(decoded[i], stream.data[i]);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|