| #include <transport/TTransportUtils.h> |
| #include <transport/TBufferTransports.h> |
| using facebook::thrift::transport::TTransportException; |
| using facebook::thrift::transport::TPipedTransport; |
| using facebook::thrift::transport::TMemoryBuffer; |
| shared_ptr<TMemoryBuffer> underlying(new TMemoryBuffer); |
| shared_ptr<TMemoryBuffer> pipe(new TMemoryBuffer); |
| shared_ptr<TPipedTransport> trans(new TPipedTransport(underlying, pipe)); |
| underlying->write((uint8_t*)"abcd", 4); |
| trans->readAll(buffer, 2); |
| assert( string((char*)buffer, 2) == "ab" ); |
| assert( pipe->getBufferAsString() == "ab" ); |
| underlying->write((uint8_t*)"ef", 2); |
| trans->readAll(buffer, 2); |
| assert( string((char*)buffer, 2) == "cd" ); |
| trans->readAll(buffer, 2); |
| assert( string((char*)buffer, 2) == "ef" ); |
| assert( pipe->getBufferAsString() == "cdef" ); |