proxygen
Public Member Functions | Protected Member Functions | Static Protected Member Functions | Protected Attributes | List of all members
fizz::client::test::AsyncFizzClientTest Class Reference
Inheritance diagram for fizz::client::test::AsyncFizzClientTest:
testing::Test

Public Member Functions

void SetUp () override
 
- Public Member Functions inherited from testing::Test
virtual ~Test ()
 
virtual ~Test ()
 
virtual ~Test ()
 

Protected Member Functions

void expectTransportReadCallback ()
 
void expectAppClose ()
 
void connect ()
 
void fullHandshakeSuccess (bool acceptEarlyData, std::string alpn="h2", std::shared_ptr< const Cert > clientCert=nullptr, std::shared_ptr< const Cert > serverCert=nullptr, bool pskResumed=false)
 
void completeHandshake ()
 
void completeEarlyHandshake (EarlyDataParams params=getEarlyDataParams())
 
- Protected Member Functions inherited from testing::Test
 Test ()
 
virtual void TearDown ()
 
 Test ()
 
virtual void TearDown ()
 
 Test ()
 
virtual void TearDown ()
 

Static Protected Member Functions

static EarlyDataParams getEarlyDataParams ()
 

Protected Attributes

AsyncFizzClientT< MockClientStateMachineInstance >::UniquePtr client_
 
std::shared_ptr< FizzClientContext > context_
 
std::shared_ptr< MockPskCache > mockPskCache_
 
MockAsyncTransport * socket_
 
MockClientStateMachineInstance * machine_
 
AsyncTransportWrapper::ReadCallback * socketReadCallback_
 
MockHandshakeCallbackT< MockClientStateMachineInstance > handshakeCallback_
 
MockReadCallback readCallback_
 
MockWriteCallback writeCallback_
 
EventBase evb_
 
MockReplaySafetyCallback mockReplayCallback_
 
folly::Optional< std::string > pskIdentity_ {"pskIdentity"}
 

Additional Inherited Members

- Public Types inherited from testing::Test
typedef internal::SetUpTestCaseFunc SetUpTestCaseFunc
 
typedef internal::TearDownTestCaseFunc TearDownTestCaseFunc
 
typedef internal::SetUpTestCaseFunc SetUpTestCaseFunc
 
typedef internal::TearDownTestCaseFunc TearDownTestCaseFunc
 
typedef internal::SetUpTestCaseFunc SetUpTestCaseFunc
 
typedef internal::TearDownTestCaseFunc TearDownTestCaseFunc
 
- Static Public Member Functions inherited from testing::Test
static void SetUpTestCase ()
 
static void TearDownTestCase ()
 
static bool HasFatalFailure ()
 
static bool HasNonfatalFailure ()
 
static bool HasFailure ()
 
static void RecordProperty (const std::string &key, const std::string &value)
 
static void RecordProperty (const std::string &key, int value)
 
static void SetUpTestCase ()
 
static void TearDownTestCase ()
 
static bool HasFatalFailure ()
 
static bool HasNonfatalFailure ()
 
static bool HasFailure ()
 
static void RecordProperty (const std::string &key, const std::string &value)
 
static void RecordProperty (const std::string &key, int value)
 
static void SetUpTestCase ()
 
static void TearDownTestCase ()
 
static bool HasFatalFailure ()
 
static bool HasNonfatalFailure ()
 
static bool HasFailure ()
 
static void RecordProperty (const std::string &key, const std::string &value)
 
static void RecordProperty (const std::string &key, int value)
 

Detailed Description

Definition at line 52 of file AsyncFizzClientTest.cpp.

Member Function Documentation

void fizz::client::test::AsyncFizzClientTest::completeEarlyHandshake ( EarlyDataParams  params = getEarlyDataParams())
inlineprotected

Definition at line 145 of file AsyncFizzClientTest.cpp.

References testing::_, fizz::client::detail::actions(), folly::netops::connect(), folly::IOBuf::copyBuffer(), EXPECT_CALL, EXPECT_FALSE, testing::InvokeWithoutArgs(), fizz::client::ReportEarlyHandshakeSuccess::maxEarlyDataSize, folly::gen::move, and params.

145  {
146  connect();
147  EXPECT_CALL(*machine_, _processSocketData(_, _))
148  .WillOnce(InvokeWithoutArgs([&params]() {
149  auto addParamsToState =
150  [params = std::move(params)](State& newState) mutable {
151  newState.earlyDataParams() = std::move(params);
152  };
153  ReportEarlyHandshakeSuccess reportSuccess;
154  reportSuccess.maxEarlyDataSize = 1000;
155  return detail::actions(
156  std::move(addParamsToState),
157  std::move(reportSuccess),
158  WaitForData());
159  }));
160  EXPECT_CALL(handshakeCallback_, _fizzHandshakeSuccess());
161  socketReadCallback_->readBufferAvailable(IOBuf::copyBuffer("ServerData"));
162  EXPECT_FALSE(client_->isReplaySafe());
163  }
folly::AsyncReader::ReadCallback::readBufferAvailable
virtual void readBufferAvailable(std::unique_ptr< IOBuf >) noexcept
Definition: AsyncTransport.h:586
folly::gen::move
constexpr detail::Map< Move > move
Definition: Base-inl.h:2567
fizz::client::test::AsyncFizzClientTest::handshakeCallback_
MockHandshakeCallbackT< MockClientStateMachineInstance > handshakeCallback_
Definition: AsyncFizzClientTest.cpp:171
testing::InvokeWithoutArgs
PolymorphicAction< internal::InvokeWithoutArgsAction< FunctionImpl > > InvokeWithoutArgs(FunctionImpl function_impl)
Definition: gmock-actions.h:1168
params
constexpr Params params[]
Definition: FlatCombiningTest.cpp:94
fizz::client::test::AsyncFizzClientTest::socketReadCallback_
AsyncTransportWrapper::ReadCallback * socketReadCallback_
Definition: AsyncFizzClientTest.cpp:170
fizz::client::test::AsyncFizzClientTest::machine_
MockClientStateMachineInstance * machine_
Definition: AsyncFizzClientTest.cpp:169
fizz::client::detail::actions
Actions actions(Args &&...act)
Definition: Actions.h:86
EXPECT_CALL
#define EXPECT_CALL(obj, call)
Definition: gmock-spec-builders.h:1845
testing::_
const internal::AnythingMatcher _
Definition: gmock-matchers.h:3729
EXPECT_FALSE
#define EXPECT_FALSE(condition)
Definition: gtest.h:1862
fizz::client::test::AsyncFizzClientTest::client_
AsyncFizzClientT< MockClientStateMachineInstance >::UniquePtr client_
Definition: AsyncFizzClientTest.cpp:165
folly::IOBuf::copyBuffer
static std::unique_ptr< IOBuf > copyBuffer(const void *buf, std::size_t size, std::size_t headroom=0, std::size_t minTailroom=0)
Definition: IOBuf.h:1587
void fizz::client::test::AsyncFizzClientTest::completeHandshake ( )
inlineprotected

Definition at line 131 of file AsyncFizzClientTest.cpp.

References folly::netops::connect(), and EXPECT_CALL.

131  {
132  connect();
133  EXPECT_CALL(handshakeCallback_, _fizzHandshakeSuccess());
134  fullHandshakeSuccess(false);
135  }
fizz::client::test::AsyncFizzClientTest::fullHandshakeSuccess
void fullHandshakeSuccess(bool acceptEarlyData, std::string alpn="h2", std::shared_ptr< const Cert > clientCert=nullptr, std::shared_ptr< const Cert > serverCert=nullptr, bool pskResumed=false)
Definition: AsyncFizzClientTest.cpp:97
fizz::client::test::AsyncFizzClientTest::handshakeCallback_
MockHandshakeCallbackT< MockClientStateMachineInstance > handshakeCallback_
Definition: AsyncFizzClientTest.cpp:171
EXPECT_CALL
#define EXPECT_CALL(obj, call)
Definition: gmock-spec-builders.h:1845
void fizz::client::test::AsyncFizzClientTest::connect ( )
inlineprotected

Definition at line 89 of file AsyncFizzClientTest.cpp.

References testing::_, EXPECT_CALL, testing::InvokeWithoutArgs(), sni, and string.

89  {
90  expectTransportReadCallback();
91  EXPECT_CALL(*machine_, _processConnect(_, _, _, _, _, _))
92  .WillOnce(InvokeWithoutArgs([]() { return Actions(); }));
93  const auto sni = std::string("www.example.com");
94  client_->connect(&handshakeCallback_, nullptr, sni, pskIdentity_);
95  }
fizz::client::test::AsyncFizzClientTest::handshakeCallback_
MockHandshakeCallbackT< MockClientStateMachineInstance > handshakeCallback_
Definition: AsyncFizzClientTest.cpp:171
testing::InvokeWithoutArgs
PolymorphicAction< internal::InvokeWithoutArgsAction< FunctionImpl > > InvokeWithoutArgs(FunctionImpl function_impl)
Definition: gmock-actions.h:1168
fizz::client::Actions
std::vector< Action > Actions
Definition: Actions.h:81
fizz::client::test::AsyncFizzClientTest::pskIdentity_
folly::Optional< std::string > pskIdentity_
Definition: AsyncFizzClientTest.cpp:176
fizz::client::test::AsyncFizzClientTest::machine_
MockClientStateMachineInstance * machine_
Definition: AsyncFizzClientTest.cpp:169
sni
StringPiece sni
Definition: ExtensionsTest.cpp:23
string
const char * string
Definition: Conv.cpp:212
EXPECT_CALL
#define EXPECT_CALL(obj, call)
Definition: gmock-spec-builders.h:1845
testing::_
const internal::AnythingMatcher _
Definition: gmock-matchers.h:3729
fizz::client::test::AsyncFizzClientTest::client_
AsyncFizzClientT< MockClientStateMachineInstance >::UniquePtr client_
Definition: AsyncFizzClientTest.cpp:165
void fizz::client::test::AsyncFizzClientTest::expectAppClose ( )
inlineprotected

Definition at line 74 of file AsyncFizzClientTest.cpp.

References testing::_, fizz::client::detail::actions(), fizz::WriteToSocket::contents, fizz::TLSContent::contentType, folly::IOBuf::copyBuffer(), fizz::TLSContent::data, fizz::TLSContent::encryptionLevel, fizz::client::Error, EXPECT_CALL, fizz::Handshake, fizz::handshake, testing::InvokeWithoutArgs(), folly::gen::move, fizz::test::State::state(), and fizz::detail::write().

74  {
75  EXPECT_CALL(*machine_, _processAppClose(_))
76  .WillOnce(InvokeWithoutArgs([]() {
77  TLSContent record;
78  record.contentType = ContentType::handshake;
79  record.data = IOBuf::copyBuffer("closenotify");
80  record.encryptionLevel = EncryptionLevel::Handshake;
81  WriteToSocket write;
82  write.contents.emplace_back(std::move(record));
83  return detail::actions(
84  [](State& newState) { newState.state() = StateEnum::Error; },
85  std::move(write));
86  }));
87  }
fizz::detail::write
void write(const T &in, folly::io::Appender &appender)
Definition: Types-inl.h:112
folly::gen::move
constexpr detail::Map< Move > move
Definition: Base-inl.h:2567
testing::InvokeWithoutArgs
PolymorphicAction< internal::InvokeWithoutArgsAction< FunctionImpl > > InvokeWithoutArgs(FunctionImpl function_impl)
Definition: gmock-actions.h:1168
fizz::client::test::AsyncFizzClientTest::machine_
MockClientStateMachineInstance * machine_
Definition: AsyncFizzClientTest.cpp:169
fizz::client::detail::actions
Actions actions(Args &&...act)
Definition: Actions.h:86
fizz::test::State::state
StateEnum state() const
Definition: FizzBaseTest.cpp:47
EXPECT_CALL
#define EXPECT_CALL(obj, call)
Definition: gmock-spec-builders.h:1845
testing::_
const internal::AnythingMatcher _
Definition: gmock-matchers.h:3729
folly::IOBuf::copyBuffer
static std::unique_ptr< IOBuf > copyBuffer(const void *buf, std::size_t size, std::size_t headroom=0, std::size_t minTailroom=0)
Definition: IOBuf.h:1587
void fizz::client::test::AsyncFizzClientTest::expectTransportReadCallback ( )
inlineprotected

Definition at line 69 of file AsyncFizzClientTest.cpp.

References testing::_, EXPECT_CALL, and socket_.

69  {
70  EXPECT_CALL(*socket_, setReadCB(_))
71  .WillRepeatedly(SaveArg<0>(&socketReadCallback_));
72  }
fizz::client::test::AsyncFizzClientTest::socketReadCallback_
AsyncTransportWrapper::ReadCallback * socketReadCallback_
Definition: AsyncFizzClientTest.cpp:170
EXPECT_CALL
#define EXPECT_CALL(obj, call)
Definition: gmock-spec-builders.h:1845
testing::_
const internal::AnythingMatcher _
Definition: gmock-matchers.h:3729
void fizz::client::test::AsyncFizzClientTest::fullHandshakeSuccess ( bool  acceptEarlyData,
std::string  alpn = "h2",
std::shared_ptr< const Cert >  clientCert = nullptr,
std::shared_ptr< const Cert >  serverCert = nullptr,
bool  pskResumed = false 
)
inlineprotected

Definition at line 97 of file AsyncFizzClientTest.cpp.

References testing::_, fizz::client::detail::actions(), alpn, clientCert, folly::IOBuf::copyBuffer(), fizz::client::ReportHandshakeSuccess::earlyDataAccepted, folly::Range< Iter >::empty(), EXPECT_CALL, testing::InvokeWithoutArgs(), folly::gen::move, folly::none, fizz::psk_ke, fizz::Resumption, serverCert, fizz::tls_1_3, and fizz::TLS_AES_128_GCM_SHA256.

102  {
103  EXPECT_CALL(*machine_, _processSocketData(_, _))
104  .WillOnce(InvokeWithoutArgs([=]() {
105  auto addToState = [=](State& newState) {
106  newState.exporterMasterSecret() =
107  folly::IOBuf::copyBuffer("12345678901234567890123456789012");
108  newState.cipher() = CipherSuite::TLS_AES_128_GCM_SHA256;
109  newState.version() = ProtocolVersion::tls_1_3;
110  if (alpn.empty()) {
111  newState.alpn() = none;
112  } else {
113  newState.alpn() = alpn;
114  }
115  newState.clientCert() = clientCert;
116  newState.serverCert() = serverCert;
117 
118  if (acceptEarlyData || pskResumed) {
119  newState.pskMode() = PskKeyExchangeMode::psk_ke;
120  newState.pskType() = PskType::Resumption;
121  }
122  };
123  ReportHandshakeSuccess reportSuccess;
124  reportSuccess.earlyDataAccepted = acceptEarlyData;
125  return detail::actions(
126  std::move(addToState), std::move(reportSuccess), WaitForData());
127  }));
128  socketReadCallback_->readBufferAvailable(IOBuf::copyBuffer("ServerData"));
129  }
folly::AsyncReader::ReadCallback::readBufferAvailable
virtual void readBufferAvailable(std::unique_ptr< IOBuf >) noexcept
Definition: AsyncTransport.h:586
folly::gen::move
constexpr detail::Map< Move > move
Definition: Base-inl.h:2567
alpn
StringPiece alpn
Definition: ExtensionsTest.cpp:22
testing::InvokeWithoutArgs
PolymorphicAction< internal::InvokeWithoutArgsAction< FunctionImpl > > InvokeWithoutArgs(FunctionImpl function_impl)
Definition: gmock-actions.h:1168
folly::Range::empty
constexpr bool empty() const
Definition: Range.h:443
fizz::client::test::AsyncFizzClientTest::socketReadCallback_
AsyncTransportWrapper::ReadCallback * socketReadCallback_
Definition: AsyncFizzClientTest.cpp:170
serverCert
std::shared_ptr< const Cert > serverCert
Definition: ClientProtocol.cpp:779
fizz::client::test::AsyncFizzClientTest::machine_
MockClientStateMachineInstance * machine_
Definition: AsyncFizzClientTest.cpp:169
fizz::client::detail::actions
Actions actions(Args &&...act)
Definition: Actions.h:86
clientCert
std::shared_ptr< const Cert > clientCert
Definition: ClientProtocol.cpp:780
EXPECT_CALL
#define EXPECT_CALL(obj, call)
Definition: gmock-spec-builders.h:1845
testing::_
const internal::AnythingMatcher _
Definition: gmock-matchers.h:3729
folly::IOBuf::copyBuffer
static std::unique_ptr< IOBuf > copyBuffer(const void *buf, std::size_t size, std::size_t headroom=0, std::size_t minTailroom=0)
Definition: IOBuf.h:1587
folly::none
constexpr None none
Definition: Optional.h:87
static EarlyDataParams fizz::client::test::AsyncFizzClientTest::getEarlyDataParams ( )
inlinestaticprotected

Definition at line 137 of file AsyncFizzClientTest.cpp.

References fizz::client::EarlyDataParams::alpn, fizz::client::EarlyDataParams::cipher, params, fizz::tls_1_3, fizz::TLS_AES_128_GCM_SHA256, and fizz::client::EarlyDataParams::version.

137  {
138  EarlyDataParams params;
139  params.version = ProtocolVersion::tls_1_3;
140  params.cipher = CipherSuite::TLS_AES_128_GCM_SHA256;
141  params.alpn = "h2";
142  return params;
143  }
params
constexpr Params params[]
Definition: FlatCombiningTest.cpp:94
void fizz::client::test::AsyncFizzClientTest::SetUp ( )
inlineoverridevirtual

Reimplemented from testing::Test.

Definition at line 54 of file AsyncFizzClientTest.cpp.

References fizz::client::test::MockClientStateMachineInstance::instance, folly::gen::move, ON_CALL, testing::Return(), and socket_.

54  {
55  context_ = std::make_shared<FizzClientContext>();
56  context_->setSendEarlyData(true);
57  mockPskCache_ = std::make_shared<MockPskCache>();
58  context_->setPskCache(mockPskCache_);
59  socket_ = new MockAsyncTransport();
60  auto transport = AsyncTransportWrapper::UniquePtr(socket_);
61  client_.reset(new AsyncFizzClientT<MockClientStateMachineInstance>(
62  std::move(transport), context_));
63  machine_ = MockClientStateMachineInstance::instance;
64  ON_CALL(*socket_, good()).WillByDefault(Return(true));
65  ON_CALL(readCallback_, isBufferMovable_()).WillByDefault(Return(true));
66  }
folly::gen::move
constexpr detail::Map< Move > move
Definition: Base-inl.h:2567
folly::AsyncTransportWrapper::UniquePtr
std::unique_ptr< AsyncTransportWrapper, Destructor > UniquePtr
Definition: AsyncTransport.h:701
fizz::client::test::AsyncFizzClientTest::machine_
MockClientStateMachineInstance * machine_
Definition: AsyncFizzClientTest.cpp:169
ON_CALL
#define ON_CALL(obj, call)
Definition: gmock-spec-builders.h:1841
fizz::client::test::AsyncFizzClientTest::mockPskCache_
std::shared_ptr< MockPskCache > mockPskCache_
Definition: AsyncFizzClientTest.cpp:167
fizz::client::test::AsyncFizzClientTest::client_
AsyncFizzClientT< MockClientStateMachineInstance >::UniquePtr client_
Definition: AsyncFizzClientTest.cpp:165
fizz::client::test::MockClientStateMachineInstance::instance
static MockClientStateMachineInstance * instance
Definition: AsyncFizzClientTest.cpp:34
fizz::client::test::AsyncFizzClientTest::context_
std::shared_ptr< FizzClientContext > context_
Definition: AsyncFizzClientTest.cpp:166
testing::Return
internal::ReturnAction< R > Return(R value)
Definition: gmock-actions.h:1061

Member Data Documentation

AsyncFizzClientT<MockClientStateMachineInstance>::UniquePtr fizz::client::test::AsyncFizzClientTest::client_
protected

Definition at line 165 of file AsyncFizzClientTest.cpp.

std::shared_ptr<FizzClientContext> fizz::client::test::AsyncFizzClientTest::context_
protected

Definition at line 166 of file AsyncFizzClientTest.cpp.

EventBase fizz::client::test::AsyncFizzClientTest::evb_
protected

Definition at line 174 of file AsyncFizzClientTest.cpp.

MockHandshakeCallbackT<MockClientStateMachineInstance> fizz::client::test::AsyncFizzClientTest::handshakeCallback_
protected

Definition at line 171 of file AsyncFizzClientTest.cpp.

MockClientStateMachineInstance* fizz::client::test::AsyncFizzClientTest::machine_
protected

Definition at line 169 of file AsyncFizzClientTest.cpp.

std::shared_ptr<MockPskCache> fizz::client::test::AsyncFizzClientTest::mockPskCache_
protected

Definition at line 167 of file AsyncFizzClientTest.cpp.

MockReplaySafetyCallback fizz::client::test::AsyncFizzClientTest::mockReplayCallback_
protected

Definition at line 175 of file AsyncFizzClientTest.cpp.

folly::Optional<std::string> fizz::client::test::AsyncFizzClientTest::pskIdentity_ {"pskIdentity"}
protected

Definition at line 176 of file AsyncFizzClientTest.cpp.

MockReadCallback fizz::client::test::AsyncFizzClientTest::readCallback_
protected

Definition at line 172 of file AsyncFizzClientTest.cpp.

MockAsyncTransport* fizz::client::test::AsyncFizzClientTest::socket_
protected

Definition at line 168 of file AsyncFizzClientTest.cpp.

AsyncTransportWrapper::ReadCallback* fizz::client::test::AsyncFizzClientTest::socketReadCallback_
protected

Definition at line 170 of file AsyncFizzClientTest.cpp.

MockWriteCallback fizz::client::test::AsyncFizzClientTest::writeCallback_
protected

Definition at line 173 of file AsyncFizzClientTest.cpp.

The documentation for this class was generated from the following file: