folly/io/async/test/MockAsyncTransport.h

   1 /*
   2  * Copyright 2015 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 <gmock/gmock.h>
  19
  20 #include <folly/Memory.h>
  21 #include <folly/io/async/AsyncTransport.h>
  22
  23 namespace folly { namespace test {
  24
  25 class MockAsyncTransport: public AsyncTransportWrapper {
  26  public:
  27   MOCK_METHOD1(setReadCB, void(ReadCallback*));
  28   MOCK_CONST_METHOD0(getReadCallback, ReadCallback*());
  29   MOCK_CONST_METHOD0(getReadCB, ReadCallback*());
  30   MOCK_METHOD5(write, void(WriteCallback*,
  31                            const void*, size_t,
  32                            WriteFlags,
  33                            BufferCallback*));
  34   MOCK_METHOD5(writev, void(WriteCallback*,
  35                             const iovec*, size_t,
  36                             WriteFlags,
  37                             BufferCallback*));
  38   MOCK_METHOD4(writeChain,
  39                void(WriteCallback*,
  40                     std::shared_ptr<folly::IOBuf>,
  41                     WriteFlags,
  42                     BufferCallback*));
  43
  44
  45   void writeChain(WriteCallback* callback,
  46                   std::unique_ptr<folly::IOBuf>&& iob,
  47                   WriteFlags flags =
  48                   WriteFlags::NONE,
  49                   BufferCallback* bufCB = nullptr) override {
  50     writeChain(
  51         callback,
  52         std::shared_ptr<folly::IOBuf>(iob.release()),
  53         flags,
  54         bufCB);
  55   }
  56
  57   MOCK_METHOD0(close, void());
  58   MOCK_METHOD0(closeNow, void());
  59   MOCK_METHOD0(closeWithReset, void());
  60   MOCK_METHOD0(shutdownWrite, void());
  61   MOCK_METHOD0(shutdownWriteNow, void());
  62   MOCK_CONST_METHOD0(good, bool());
  63   MOCK_CONST_METHOD0(readable, bool());
  64   MOCK_CONST_METHOD0(connecting, bool());
  65   MOCK_CONST_METHOD0(error, bool());
  66   MOCK_METHOD1(attachEventBase, void(EventBase*));
  67   MOCK_METHOD0(detachEventBase, void());
  68   MOCK_CONST_METHOD0(isDetachable, bool());
  69   MOCK_CONST_METHOD0(getEventBase, EventBase*());
  70   MOCK_METHOD1(setSendTimeout, void(uint32_t));
  71   MOCK_CONST_METHOD0(getSendTimeout, uint32_t());
  72   MOCK_CONST_METHOD1(getLocalAddress, void(folly::SocketAddress*));
  73   MOCK_CONST_METHOD1(getPeerAddress, void(folly::SocketAddress*));
  74   MOCK_CONST_METHOD0(getAppBytesWritten, size_t());
  75   MOCK_CONST_METHOD0(getRawBytesWritten, size_t());
  76   MOCK_CONST_METHOD0(getAppBytesReceived, size_t());
  77   MOCK_CONST_METHOD0(getRawBytesReceived, size_t());
  78   MOCK_CONST_METHOD0(isEorTrackingEnabled, bool());
  79   MOCK_METHOD1(setEorTracking, void(bool));
  80   MOCK_CONST_METHOD0(getWrappedTransport, AsyncTransportWrapper*());
  81
  82 };
  83
  84 class MockReadCallback: public AsyncTransportWrapper::ReadCallback {
  85  public:
  86   MOCK_METHOD2(getReadBuffer, void(void**, size_t*));
  87   GMOCK_METHOD1_(, noexcept, , readDataAvailable, void(size_t));
  88   GMOCK_METHOD0_(, noexcept, , isBufferMovable, bool());
  89   GMOCK_METHOD1_(, noexcept, ,
  90       readBufferAvailableInternal, void(std::shared_ptr<folly::IOBuf>));
  91   GMOCK_METHOD0_(, noexcept, , readEOF, void());
  92   GMOCK_METHOD1_(, noexcept, , readErr,
  93                  void(const AsyncSocketException&));
  94
  95   void readBufferAvailable(std::unique_ptr<folly::IOBuf> readBuf)
  96     noexcept override {
  97     readBufferAvailableInternal(
  98         folly::to_shared_ptr(std::move(readBuf)));
  99   }
 100 };
 101
 102 class MockWriteCallback: public AsyncTransportWrapper::WriteCallback {
 103  public:
 104   GMOCK_METHOD0_(, noexcept, , writeSuccess, void());
 105   GMOCK_METHOD2_(, noexcept, , writeErr,
 106                  void(size_t, const AsyncSocketException&));
 107 };
 108
 109 }}