Add replay safety method and callback to AsyncTransport.

[folly.git] / folly / io / async / test / MockAsyncTransport.h

diff --git a/folly/io/async/test/MockAsyncTransport.h b/folly/io/async/test/MockAsyncTransport.h
index 5ad241c1fa41363b91170885d3c0731125ca753f..8b4e9228b4e55b82dd944b2ca9f01fbcab1b188b 100644 (file)
--- a/folly/io/async/test/MockAsyncTransport.h
+++ b/folly/io/async/test/MockAsyncTransport.h
@@ -65,6 +65,11 @@ class MockAsyncTransport: public AsyncTransportWrapper {
 
 };
 
+class MockReplaySafetyCallback : public AsyncTransport::ReplaySafetyCallback {
+ public:
+  GMOCK_METHOD0_(, noexcept, , onReplaySafe, void());
+};
+
 class MockReadCallback: public AsyncTransportWrapper::ReadCallback {
  public:
   MOCK_METHOD2(getReadBuffer, void(void**, size_t*));