X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=folly%2Fio%2Fasync%2Ftest%2FMockAsyncSocket.h;h=d8f58409b4041f5277c6122c88da41ba492d1516;hb=6a6ac91e1fda65d7871390f03be8d19b3591ba45;hp=f7f9fee165eb1b9b328a1c155ec336a4ca070060;hpb=321542683a01c3f334047531e9b487f047129775;p=folly.git

diff --git a/folly/io/async/test/MockAsyncSocket.h b/folly/io/async/test/MockAsyncSocket.h
index f7f9fee1..d8f58409 100644
--- a/folly/io/async/test/MockAsyncSocket.h
+++ b/folly/io/async/test/MockAsyncSocket.h
@@ -15,10 +15,9 @@
  */
 #pragma once
 
-#include <gmock/gmock.h>
-
 #include <folly/io/async/AsyncSocket.h>
 #include <folly/io/async/EventBase.h>
+#include <folly/portability/GMock.h>
 
 namespace folly {
 
@@ -46,6 +45,8 @@ class MockAsyncSocket : public AsyncSocket {
   MOCK_CONST_METHOD0(good, bool());
   MOCK_CONST_METHOD0(readable, bool());
   MOCK_CONST_METHOD0(hangup, bool());
+  MOCK_METHOD1(setPeek, void(bool));
+  MOCK_METHOD1(setReadCB, void(ReadCallback*));
 };
 
 }}