Subversion Repositories SmartDukaan

/*

 * Licensed to the Apache Software Foundation (ASF) under one

 * or more contributor license agreements. See the NOTICE file

 * distributed with this work for additional information

 * regarding copyright ownership. The ASF licenses this file

 * to you under the Apache License, Version 2.0 (the

 * "License"); you may not use this file except in compliance

 * with the License. You may obtain a copy of the License at

 *

 *   http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing,

 * software distributed under the License is distributed on an

 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

 * KIND, either express or implied. See the License for the

 * specific language governing permissions and limitations

 * under the License.

 */

#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 apache::thrift::transport::TMemoryBuffer;

    using apache::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 apache::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 apache::thrift::transport::TTransportException;

    using apache::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()