2 * Copyright 2015 Facebook, Inc.
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
18 #include <gmock/gmock.h>
20 #include <folly/Memory.h>
21 #include <folly/io/async/AsyncTransport.h>
23 namespace folly { namespace test {
25 class MockAsyncTransport: public AsyncTransportWrapper {
27 MOCK_METHOD1(setReadCB, void(ReadCallback*));
28 MOCK_CONST_METHOD0(getReadCallback, ReadCallback*());
29 MOCK_CONST_METHOD0(getReadCB, ReadCallback*());
30 MOCK_METHOD4(write, void(WriteCallback*,
33 MOCK_METHOD4(writev, void(WriteCallback*,
36 MOCK_METHOD3(writeChain,
38 std::shared_ptr<folly::IOBuf>,
42 void writeChain(WriteCallback* callback,
43 std::unique_ptr<folly::IOBuf>&& iob,
45 WriteFlags::NONE) override {
46 writeChain(callback, std::shared_ptr<folly::IOBuf>(iob.release()), flags);
49 MOCK_METHOD0(close, void());
50 MOCK_METHOD0(closeNow, void());
51 MOCK_METHOD0(closeWithReset, void());
52 MOCK_METHOD0(shutdownWrite, void());
53 MOCK_METHOD0(shutdownWriteNow, void());
54 MOCK_CONST_METHOD0(good, bool());
55 MOCK_CONST_METHOD0(readable, bool());
56 MOCK_CONST_METHOD0(connecting, bool());
57 MOCK_CONST_METHOD0(error, bool());
58 MOCK_METHOD1(attachEventBase, void(EventBase*));
59 MOCK_METHOD0(detachEventBase, void());
60 MOCK_CONST_METHOD0(isDetachable, bool());
61 MOCK_CONST_METHOD0(getEventBase, EventBase*());
62 MOCK_METHOD1(setSendTimeout, void(uint32_t));
63 MOCK_CONST_METHOD0(getSendTimeout, uint32_t());
64 MOCK_CONST_METHOD1(getLocalAddress, void(folly::SocketAddress*));
65 MOCK_CONST_METHOD1(getPeerAddress, void(folly::SocketAddress*));
66 MOCK_CONST_METHOD0(getAppBytesWritten, size_t());
67 MOCK_CONST_METHOD0(getRawBytesWritten, size_t());
68 MOCK_CONST_METHOD0(getAppBytesReceived, size_t());
69 MOCK_CONST_METHOD0(getRawBytesReceived, size_t());
70 MOCK_CONST_METHOD0(isEorTrackingEnabled, bool());
71 MOCK_METHOD1(setEorTracking, void(bool));
75 class MockReadCallback: public AsyncTransportWrapper::ReadCallback {
77 MOCK_METHOD2(getReadBuffer, void(void**, size_t*));
78 GMOCK_METHOD1_(, noexcept, , readDataAvailable, void(size_t));
79 GMOCK_METHOD0_(, noexcept, , isBufferMovable, bool());
80 GMOCK_METHOD1_(, noexcept, ,
81 readBufferAvailableInternal, void(std::shared_ptr<folly::IOBuf>));
82 GMOCK_METHOD0_(, noexcept, , readEOF, void());
83 GMOCK_METHOD1_(, noexcept, , readErr,
84 void(const AsyncSocketException&));
86 void readBufferAvailable(std::unique_ptr<folly::IOBuf> readBuf)
88 readBufferAvailableInternal(
89 folly::to_shared_ptr(std::move(readBuf)));
93 class MockWriteCallback: public AsyncTransportWrapper::WriteCallback {
95 GMOCK_METHOD0_(, noexcept, , writeSuccess, void());
96 GMOCK_METHOD2_(, noexcept, , writeErr,
97 void(size_t, const AsyncSocketException&));