async_stream_test_helper.hxx

#include "utils/async_datagram_test_helper.hxx"
#include "openlcb/StreamReceiver.hxx"
 
namespace openlcb
{
 
static constexpr uint8_t LOCAL_STREAM_ID = 0x3a;
static constexpr uint8_t SRC_STREAM_ID = 0xa7;
 
string get_payload_data(size_t length)
{
    string r(length, 0);
    for (size_t i = 0; i < length; ++i)
    {
        r[i] = i & 0xff;
    }
    return r;
}
 
struct CollectData : public ByteSink
{
    string data;
    Q q;
    bool keepBuffers_ {false};
 
    void send(ByteBuffer *msg, unsigned prio) override
    {
        auto rb = get_buffer_deleter(msg);
        data.append((char *)msg->data()->data_, msg->data()->size());
        if (keepBuffers_)
        {
            q.insert(msg->ref());
        }
    }
 
    string qtake()
    {
        ByteBuffer *b = (ByteBuffer *)q.next(0);
        HASSERT(b != nullptr);
        string ret((char *)b->data()->data_, b->data()->size());
        b->unref();
        return ret;
    }
};
 
class StreamTestBase : public TwoNodeDatagramTest
{
protected:
    StreamTestBase()
    {
        mainBufferPool->alloc(&recvRequest_);
    }
 
    BufferPtr<StreamReceiveRequest> recvRequest_;
    CollectData sink_;
};
 
} // namespace openlcb