folly/io/async/test/MockAsyncTransport.h

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 #pragma once
  17
  18 #include <folly/Memory.h>
  19 #include <folly/io/async/AsyncTransport.h>
  20 #include <folly/portability/GMock.h>
  21
  22 namespace folly { namespace test {
  23
  24 class MockAsyncTransport: public AsyncTransportWrapper {
  25  public:
  26   MOCK_METHOD1(setReadCB, void(ReadCallback*));
  27   MOCK_CONST_METHOD0(getReadCallback, ReadCallback*());
  28   MOCK_CONST_METHOD0(getReadCB, ReadCallback*());
  29   MOCK_METHOD4(write, void(WriteCallback*, const void*, size_t, WriteFlags));
  30   MOCK_METHOD4(writev, void(WriteCallback*, const iovec*, size_t, WriteFlags));
  31   MOCK_METHOD3(writeChain,
  32                void(WriteCallback*, std::shared_ptr<folly::IOBuf>, WriteFlags));
  33
  34   void writeChain(WriteCallback* callback,
  35                   std::unique_ptr<folly::IOBuf>&& iob,
  36                   WriteFlags flags = WriteFlags::NONE) override {
  37     writeChain(callback, std::shared_ptr<folly::IOBuf>(iob.release()), flags);
  38   }
  39
  40   MOCK_METHOD0(close, void());
  41   MOCK_METHOD0(closeNow, void());
  42   MOCK_METHOD0(closeWithReset, void());
  43   MOCK_METHOD0(shutdownWrite, void());
  44   MOCK_METHOD0(shutdownWriteNow, void());
  45   MOCK_CONST_METHOD0(good, bool());
  46   MOCK_CONST_METHOD0(readable, bool());
  47   MOCK_CONST_METHOD0(connecting, bool());
  48   MOCK_CONST_METHOD0(error, bool());
  49   MOCK_METHOD1(attachEventBase, void(EventBase*));
  50   MOCK_METHOD0(detachEventBase, void());
  51   MOCK_CONST_METHOD0(isDetachable, bool());
  52   MOCK_CONST_METHOD0(getEventBase, EventBase*());
  53   MOCK_METHOD1(setSendTimeout, void(uint32_t));
  54   MOCK_CONST_METHOD0(getSendTimeout, uint32_t());
  55   MOCK_CONST_METHOD1(getLocalAddress, void(folly::SocketAddress*));
  56   MOCK_CONST_METHOD1(getPeerAddress, void(folly::SocketAddress*));
  57   MOCK_CONST_METHOD0(getAppBytesWritten, size_t());
  58   MOCK_CONST_METHOD0(getRawBytesWritten, size_t());
  59   MOCK_CONST_METHOD0(getAppBytesReceived, size_t());
  60   MOCK_CONST_METHOD0(getRawBytesReceived, size_t());
  61   MOCK_CONST_METHOD0(isEorTrackingEnabled, bool());
  62   MOCK_METHOD1(setEorTracking, void(bool));
  63   MOCK_CONST_METHOD0(getWrappedTransport, AsyncTransportWrapper*());
  64   MOCK_CONST_METHOD0(isReplaySafe, bool());
  65   MOCK_METHOD1(setReplaySafetyCallback,
  66                void(AsyncTransport::ReplaySafetyCallback*));
  67 };
  68
  69 class MockReplaySafetyCallback : public AsyncTransport::ReplaySafetyCallback {
  70  public:
  71   GMOCK_METHOD0_(, noexcept, , onReplaySafe, void());
  72 };
  73
  74 class MockReadCallback: public AsyncTransportWrapper::ReadCallback {
  75  public:
  76   MOCK_METHOD2(getReadBuffer, void(void**, size_t*));
  77   GMOCK_METHOD1_(, noexcept, , readDataAvailable, void(size_t));
  78   GMOCK_METHOD0_(, noexcept, , isBufferMovable, bool());
  79   GMOCK_METHOD1_(, noexcept, ,
  80       readBufferAvailableInternal,
  81       void(std::unique_ptr<folly::IOBuf>&));
  82   GMOCK_METHOD0_(, noexcept, , readEOF, void());
  83   GMOCK_METHOD1_(, noexcept, , readErr,
  84                  void(const AsyncSocketException&));
  85
  86   void readBufferAvailable(std::unique_ptr<folly::IOBuf> readBuf)
  87     noexcept override {
  88     readBufferAvailableInternal(readBuf);
  89   }
  90 };
  91
  92 class MockWriteCallback: public AsyncTransportWrapper::WriteCallback {
  93  public:
  94   GMOCK_METHOD0_(, noexcept, , writeSuccess, void());
  95   GMOCK_METHOD2_(, noexcept, , writeErr,
  96                  void(size_t, const AsyncSocketException&));
  97 };
  98
  99 } // namespace test
 100 } // namespace folly