1 /*
2  * Copyright (c) Facebook, Inc. and its affiliates.
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 
17 #pragma once
18 
19 #include <folly/io/async/AsyncServerSocket.h>
20 #include <folly/io/async/EventBase.h>
21 #include <folly/portability/GMock.h>
22 
23 namespace folly {
24 
25 namespace test {
26 
27 class MockAsyncServerSocket : public AsyncServerSocket {
28  public:
29   typedef std::unique_ptr<MockAsyncServerSocket, Destructor> UniquePtr;
30 
31   // We explicitly do not mock destroy(), since the base class implementation
32   // in DelayedDestruction is what actually deletes the object.
33   // MOCK_METHOD0(destroy,
34   //             void());
35   MOCK_METHOD1(bind, void(const folly::SocketAddress& address));
36   MOCK_METHOD2(
37       bind,
38       void(const std::vector<folly::IPAddress>& ipAddresses, uint16_t port));
39   MOCK_METHOD1(bind, void(uint16_t port));
40   MOCK_METHOD1(listen, void(int backlog));
41   MOCK_METHOD0(startAccepting, void());
42   MOCK_METHOD3(
43       addAcceptCallback,
44       void(AcceptCallback* callback, EventBase* eventBase, uint32_t maxAtOnce));
45 };
46 
47 } // namespace test
48 } // namespace folly
49