Remove per-write buffer callback from AsyncSocket

[folly.git] / folly / io / async / test / MockAsyncTransport.h

diff --git a/folly/io/async/test/MockAsyncTransport.h b/folly/io/async/test/MockAsyncTransport.h
index bd091056514341e8348212be7622cd81eb5f9c00..8cff00f7ab214ca6c29f70a64e25673f795a023b 100644 (file)
--- a/folly/io/async/test/MockAsyncTransport.h
+++ b/folly/io/async/test/MockAsyncTransport.h
@@ -27,31 +27,15 @@ class MockAsyncTransport: public AsyncTransportWrapper {
   MOCK_METHOD1(setReadCB, void(ReadCallback*));
   MOCK_CONST_METHOD0(getReadCallback, ReadCallback*());
   MOCK_CONST_METHOD0(getReadCB, ReadCallback*());
-  MOCK_METHOD5(write, void(WriteCallback*,
-                           const void*, size_t,
-                           WriteFlags,
-                           BufferCallback*));
-  MOCK_METHOD5(writev, void(WriteCallback*,
-                            const iovec*, size_t,
-                            WriteFlags,
-                            BufferCallback*));
-  MOCK_METHOD4(writeChain,
-               void(WriteCallback*,
-                    std::shared_ptr<folly::IOBuf>,
-                    WriteFlags,
-                    BufferCallback*));
-
+  MOCK_METHOD4(write, void(WriteCallback*, const void*, size_t, WriteFlags));
+  MOCK_METHOD4(writev, void(WriteCallback*, const iovec*, size_t, WriteFlags));
+  MOCK_METHOD3(writeChain,
+               void(WriteCallback*, std::shared_ptr<folly::IOBuf>, WriteFlags));
 
   void writeChain(WriteCallback* callback,
                   std::unique_ptr<folly::IOBuf>&& iob,
-                  WriteFlags flags =
-                  WriteFlags::NONE,
-                  BufferCallback* bufCB = nullptr) override {
-    writeChain(
-        callback,
-        std::shared_ptr<folly::IOBuf>(iob.release()),
-        flags,
-        bufCB);
+                  WriteFlags flags = WriteFlags::NONE) override {
+    writeChain(callback, std::shared_ptr<folly::IOBuf>(iob.release()), flags);
   }
 
   MOCK_METHOD0(close, void());