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