From 5bb10c1ed568b5ee00e6699efdb86eae5561ddd7 Mon Sep 17 00:00:00 2001
From: Neel Goyal <ngoyal@fb.com>
Date: Tue, 26 Jul 2016 08:09:18 -0700
Subject: [PATCH] Refactor socket peeker

Summary:
Refactor the socket peeker into its
own class so that it can be used in
other places as well, for example
for peeking at plaintext traffic.

Reviewed By: knekritz

Differential Revision: D3521899

fbshipit-source-id: fcd5615dd5d7cd1ddece4c624efbdecfbbc88a09
---
 folly/io/async/test/MockAsyncSocket.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/folly/io/async/test/MockAsyncSocket.h b/folly/io/async/test/MockAsyncSocket.h
index f7f9fee1..32c692ad 100644
--- a/folly/io/async/test/MockAsyncSocket.h
+++ b/folly/io/async/test/MockAsyncSocket.h
@@ -46,6 +46,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*));
 };
 
 }}
-- 
2.34.1