thrift/test/TMemoryBufferTest.cpp
David Reiss 28f298dd5d Memory-based transport rewrite.
The old implementations of the memory-based transports (TBufferedTransport,
TFramedTransport, and TMemoryBuffer) shared very little code and all worked
a bit differently.  This change unifies them all as subclasses of a single
base (TBufferBase) which handles the fast-path operations (when requests
can be satisfied by the buffer) with inline methods (that will eventually
be made nonvirtual in the template branch) and calls out to pure-virutal
methods to handle full/empty buffers.  All of the buffer-management is now
done in terms of "base and bound" pointers rather than "pos" integers.

These classes were moved to TBufferTransports.{h,cpp}.  The .h is included
in TTransportUtils for backwards compatibility.

Also added a "TShortReadTransport" to assist testing transports.


git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@665676 13f79535-47bb-0310-9956-ffa450edef68
2008-05-01 06:17:36 +00:00

89 lines
2.6 KiB
C++

#include <boost/test/unit_test.hpp>
#include <iostream>
#include <climits>
#include <cassert>
#include <transport/TBufferTransports.h>
#include <protocol/TBinaryProtocol.h>
#include "gen-cpp/ThriftTest_types.h"
BOOST_AUTO_TEST_SUITE( TMemoryBufferTest )
BOOST_AUTO_TEST_CASE( test_roundtrip ) {
using facebook::thrift::transport::TMemoryBuffer;
using facebook::thrift::protocol::TBinaryProtocol;
using boost::shared_ptr;
shared_ptr<TMemoryBuffer> strBuffer(new TMemoryBuffer());
shared_ptr<TBinaryProtocol> binaryProtcol(new TBinaryProtocol(strBuffer));
thrift::test::Xtruct a;
a.i32_thing = 10;
a.i64_thing = 30;
a.string_thing ="holla back a";
a.write(binaryProtcol.get());
std::string serialized = strBuffer->getBufferAsString();
shared_ptr<TMemoryBuffer> strBuffer2(new TMemoryBuffer());
shared_ptr<TBinaryProtocol> binaryProtcol2(new TBinaryProtocol(strBuffer2));
strBuffer2->resetBuffer((uint8_t*)serialized.data(), serialized.length());
thrift::test::Xtruct a2;
a2.read(binaryProtcol2.get());
assert(a == a2);
}
BOOST_AUTO_TEST_CASE( test_copy )
{
using facebook::thrift::transport::TMemoryBuffer;
using std::string;
using std::cout;
using std::endl;
string* str1 = new string("abcd1234");
const char* data1 = str1->data();
TMemoryBuffer buf((uint8_t*)str1->data(), str1->length(), TMemoryBuffer::COPY);
delete str1;
string* str2 = new string("plsreuse");
bool obj_reuse = (str1 == str2);
bool dat_reuse = (data1 == str2->data());
cout << "Object reuse: " << obj_reuse << " Data reuse: " << dat_reuse
<< ((obj_reuse && dat_reuse) ? " YAY!" : "") << endl;
delete str2;
string str3 = "wxyz", str4 = "6789";
buf.readAppendToString(str3, 4);
buf.readAppendToString(str4, INT_MAX);
assert(str3 == "wxyzabcd");
assert(str4 == "67891234");
}
BOOST_AUTO_TEST_CASE( test_exceptions )
{
using facebook::thrift::transport::TTransportException;
using facebook::thrift::transport::TMemoryBuffer;
using std::string;
char data[] = "foo\0bar";
TMemoryBuffer buf1((uint8_t*)data, 7, TMemoryBuffer::OBSERVE);
string str = buf1.getBufferAsString();
assert(str.length() == 7);
buf1.resetBuffer();
try {
buf1.write((const uint8_t*)"foo", 3);
assert(false);
} catch (TTransportException& ex) {}
TMemoryBuffer buf2((uint8_t*)data, 7, TMemoryBuffer::COPY);
try {
buf2.write((const uint8_t*)"bar", 3);
} catch (TTransportException& ex) {
assert(false);
}
}
BOOST_AUTO_TEST_SUITE_END()