commit d3013f3fb5d83e201723cd97cab49d8203037c0d
author James E. King, III <jking@apache.org> Fri Sep 22 11:41:10 2017 -0700
committer James E. King, III <jking@apache.org> Fri Sep 22 15:15:30 2017 -0700
tree 4ee35620840b9b3b5440b8f4dfc30b1eaced5bfe
parent 4a6bff1c5bd7eabe71966454d90dddeb48ed4d37

THRIFT-3766: add zlib getUnderlyingTransport method
Client: C++

This closes #1370

lib/cpp/src/thrift/transport/TZlibTransport.h

diff --git a/lib/cpp/src/thrift/transport/TZlibTransport.h b/lib/cpp/src/thrift/transport/TZlibTransport.h
index 4803c61..a0fb464 100644
--- a/lib/cpp/src/thrift/transport/TZlibTransport.h
+++ b/lib/cpp/src/thrift/transport/TZlibTransport.h
@@ -180,6 +180,8 @@
   static const int DEFAULT_UWBUF_SIZE = 128;
   static const int DEFAULT_CWBUF_SIZE = 1024;
 
+  stdcxx::shared_ptr<TTransport> getUnderlyingTransport() const { return transport_; }
+
 protected:
   inline void checkZlibRv(int status, const char* msg);
   inline void checkZlibRvNothrow(int status, const char* msg);

lib/cpp/test/ZlibTest.cpp

diff --git a/lib/cpp/test/ZlibTest.cpp b/lib/cpp/test/ZlibTest.cpp
index a4479eb..ea54487 100644
--- a/lib/cpp/test/ZlibTest.cpp
+++ b/lib/cpp/test/ZlibTest.cpp
@@ -332,6 +332,12 @@
   BOOST_CHECK_EQUAL(membuf->available_read(), (uint32_t)0);
 }
 
+void test_get_underlying_transport() {
+  shared_ptr<TMemoryBuffer> membuf(new TMemoryBuffer());
+  shared_ptr<TZlibTransport> zlib_trans(new TZlibTransport(membuf));
+  BOOST_CHECK_EQUAL(membuf.get(), zlib_trans->getUnderlyingTransport().get());
+}
+
 /*
  * Initialization
  */
@@ -436,6 +442,7 @@
   add_tests(suite, gen_random_buffer(buf_len), buf_len, "random");
 
   suite->add(BOOST_TEST_CASE(test_no_write));
+  suite->add(BOOST_TEST_CASE(test_get_underlying_transport));
 
   return true;
 }