Add mocks for isBufferMovable

Summary: Add mocks for the new isBufferMovable methods.

Reviewed By: @yfeldblum

Differential Revision: D2222642

diff --git a/folly/io/async/test/MockAsyncTransport.h b/folly/io/async/test/MockAsyncTransport.h
index e4f559c5c0423ef2aaaf6013f5b699881d293a16..84ce7b936ab851d628fc04ed44fe09f001fdd218 100644 (file)
--- a/folly/io/async/test/MockAsyncTransport.h
+++ b/folly/io/async/test/MockAsyncTransport.h
@@ -17,6 +17,7 @@
 
 #include <gmock/gmock.h>
 
+#include <folly/Memory.h>
 #include <folly/io/async/AsyncTransport.h>
 
 namespace folly { namespace test {
@@ -75,9 +76,18 @@ class MockReadCallback: public AsyncTransportWrapper::ReadCallback {
  public:
   MOCK_METHOD2(getReadBuffer, void(void**, size_t*));
   GMOCK_METHOD1_(, noexcept, , readDataAvailable, void(size_t));
+  GMOCK_METHOD0_(, noexcept, , isBufferMovable, bool());
+  GMOCK_METHOD1_(, noexcept, ,
+      readBufferAvailableInternal, void(std::shared_ptr<folly::IOBuf>));
   GMOCK_METHOD0_(, noexcept, , readEOF, void());
   GMOCK_METHOD1_(, noexcept, , readErr,
                  void(const AsyncSocketException&));
+
+  void readBufferAvailable(std::unique_ptr<folly::IOBuf> readBuf)
+    noexcept override {
+    readBufferAvailableInternal(
+        folly::to_shared_ptr(std::move(readBuf)));
+  }
 };
 
 class MockWriteCallback: public AsyncTransportWrapper::WriteCallback {