Make MockAsyncUDPSocket usable with intermediate types

Summary: Make it possible to use `MockAsyncUDPSocket` in cases where the type to be mocked inherits from `AsyncUDPSocket` by templating the class and defining a default.

Differential Revision: D48717389

fbshipit-source-id: b0907d18b887ebe6703f8367b243d7f88414a0d5

folly/io/async/test/MockAsyncUDPSocket.h

@@ -22,9 +22,10 @@
 namespace folly {
 namespace test {
 
-struct MockAsyncUDPSocket : public AsyncUDPSocket {
-  explicit MockAsyncUDPSocket(EventBase* evb) : AsyncUDPSocket(evb) {}
-  ~MockAsyncUDPSocket() override {}
+template <typename Base = AsyncUDPSocket>
+struct MockAsyncUDPSocketT : public Base {
+  explicit MockAsyncUDPSocketT(EventBase* evb) : Base(evb) {}
+  ~MockAsyncUDPSocketT() override {}
 
   MOCK_METHOD(const SocketAddress&, address, (), (const));
   MOCK_METHOD(
@@ -48,15 +49,18 @@ struct MockAsyncUDPSocket : public AsyncUDPSocket {
        folly::AsyncUDPSocket::WriteOptions));
   MOCK_METHOD(
       ssize_t, writev, (const SocketAddress&, const struct iovec*, size_t));
-  MOCK_METHOD(void, resumeRead, (ReadCallback*));
+  MOCK_METHOD(void, resumeRead, (folly::AsyncUDPSocket::ReadCallback*));
   MOCK_METHOD(void, pauseRead, ());
   MOCK_METHOD(void, close, ());
   MOCK_METHOD(void, setDFAndTurnOffPMTU, ());
   MOCK_METHOD(NetworkSocket, getNetworkSocket, (), (const));
   MOCK_METHOD(void, setReusePort, (bool));
   MOCK_METHOD(void, setReuseAddr, (bool));
   MOCK_METHOD(void, dontFragment, (bool));
-  MOCK_METHOD(void, setErrMessageCallback, (ErrMessageCallback*));
+  MOCK_METHOD(
+      void,
+      setErrMessageCallback,
+      (folly::AsyncUDPSocket::ErrMessageCallback*));
   MOCK_METHOD(void, connect, (const SocketAddress&));
   MOCK_METHOD(bool, isBound, (), (const));
   MOCK_METHOD(int, getGSO, ());
@@ -78,5 +82,7 @@ struct MockAsyncUDPSocket : public AsyncUDPSocket {
       (const SocketNontrivialOptionMap&, SocketOptionKey::ApplyPos));
 };
 
+using MockAsyncUDPSocket = MockAsyncUDPSocketT<>;
+
 } // namespace test
 } // namespace folly