SocketPeekerTest.cpp File Reference

#include <wangle/acceptor/SocketPeeker.h>
#include <thread>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <folly/io/async/test/MockAsyncSocket.h>

Go to the source code of this file.

Classes
class	MockSocketPeekerCallback
class	SocketPeekerTest

Functions
	MATCHER_P2 (BufMatches, buf, len,"")
	MATCHER_P2 (IOBufMatches, buf, len,"")
	TEST_F (SocketPeekerTest, TestPeekSuccess)
	TEST_F (SocketPeekerTest, TestEOFDuringPeek)
	TEST_F (SocketPeekerTest, TestNotEnoughDataError)
	TEST_F (SocketPeekerTest, TestMultiplePeeks)
	TEST_F (SocketPeekerTest, TestDestoryWhilePeeking)
	TEST_F (SocketPeekerTest, TestNoPeekSuccess)

Function Documentation

MATCHER_P2	(	BufMatches	,
		buf	,
		len	,
		""
	)

Definition at line 59 of file SocketPeekerTest.cpp.

                                     {
  if (arg.size() != size_t(len)) {
    return false;
  } else if (len == 0) {
    return true;
  }
  return memcmp(buf, arg.data(), len) == 0;
}

MATCHER_P2	(	IOBufMatches	,
		buf	,
		len	,
		""
	)

Definition at line 68 of file SocketPeekerTest.cpp.

References folly::IOBuf::copyBuffer().

                                       {
  return folly::IOBufEqualTo()(arg, folly::IOBuf::copyBuffer(buf, len));
}

TEST_F	(	SocketPeekerTest	,
		TestPeekSuccess
	)

Definition at line 72 of file SocketPeekerTest.cpp.

References testing::_, EXPECT_CALL, EXPECT_EQ, and uint8_t.

                                          {
  EXPECT_CALL(*sock, setReadCB(_));
  SocketPeeker::UniquePtr peeker(new SocketPeeker(*sock, &callback, 2));
  peeker->start();

  uint8_t* buf = nullptr;
  size_t len = 0;
  peeker->getReadBuffer(reinterpret_cast<void**>(&buf), &len);
  EXPECT_EQ(2, len);
  // first 2 bytes of SSL3+.
  buf[0] = 0x16;
  buf[1] = 0x03;
  EXPECT_CALL(*sock, _setPreReceivedData(IOBufMatches(buf, 2)));
  EXPECT_CALL(callback, peekSuccess_(BufMatches(buf, 2)));
  // once after peeking, and once during destruction.
  EXPECT_CALL(*sock, setReadCB(nullptr));
  peeker->readDataAvailable(2);
}

TEST_F	(	SocketPeekerTest	,
		TestEOFDuringPeek
	)

Definition at line 91 of file SocketPeekerTest.cpp.

References testing::_, and EXPECT_CALL.

                                            {
  EXPECT_CALL(*sock, setReadCB(_));
  SocketPeeker::UniquePtr peeker(new SocketPeeker(*sock, &callback, 2));
  peeker->start();

  EXPECT_CALL(callback, peekError_(_));
  EXPECT_CALL(*sock, setReadCB(nullptr));
  peeker->readEOF();
}

TEST_F	(	SocketPeekerTest	,
		TestNotEnoughDataError
	)

Definition at line 101 of file SocketPeekerTest.cpp.

References testing::_, EXPECT_CALL, EXPECT_EQ, and uint8_t.

                                                 {
  EXPECT_CALL(*sock, setReadCB(_));
  SocketPeeker::UniquePtr peeker(new SocketPeeker(*sock, &callback, 2));
  peeker->start();

  uint8_t* buf = nullptr;
  size_t len = 0;
  peeker->getReadBuffer(reinterpret_cast<void**>(&buf), &len);
  EXPECT_EQ(2, len);
  buf[0] = 0x16;
  peeker->readDataAvailable(1);

  EXPECT_CALL(callback, peekError_(_));
  EXPECT_CALL(*sock, setReadCB(nullptr));
  peeker->readEOF();
}

TEST_F	(	SocketPeekerTest	,
		TestMultiplePeeks
	)

Definition at line 118 of file SocketPeekerTest.cpp.

References testing::_, EXPECT_CALL, EXPECT_EQ, and uint8_t.

                                            {
  EXPECT_CALL(*sock, setReadCB(_));
  SocketPeeker::UniquePtr peeker(new SocketPeeker(*sock, &callback, 2));
  peeker->start();

  uint8_t* buf = nullptr;
  size_t len = 0;
  peeker->getReadBuffer(reinterpret_cast<void**>(&buf), &len);
  EXPECT_EQ(2, len);
  buf[0] = 0x16;
  peeker->readDataAvailable(1);

  peeker->getReadBuffer(reinterpret_cast<void**>(&buf), &len);
  EXPECT_EQ(1, len);
  buf[0] = 0x03;

  EXPECT_CALL(*sock, _setPreReceivedData(IOBufMatches("\x16\x03", 2)));
  EXPECT_CALL(callback, peekSuccess_(BufMatches("\x16\x03", 2)));
  EXPECT_CALL(*sock, setReadCB(nullptr));
  peeker->readDataAvailable(1);
}

TEST_F	(	SocketPeekerTest	,
		TestDestoryWhilePeeking
	)

Definition at line 140 of file SocketPeekerTest.cpp.

References testing::_, and EXPECT_CALL.

                                                  {
  EXPECT_CALL(*sock, setReadCB(_));
  SocketPeeker::UniquePtr peeker(new SocketPeeker(*sock, &callback, 2));
  peeker->start();
  peeker = nullptr;
}

TEST_F	(	SocketPeekerTest	,
		TestNoPeekSuccess
	)

Definition at line 147 of file SocketPeekerTest.cpp.

References EXPECT_CALL.

                                            {
  SocketPeeker::UniquePtr peeker(new SocketPeeker(*sock, &callback, 0));

  char buf = '\0';
  EXPECT_CALL(callback, peekSuccess_(BufMatches(&buf, 0)));
  peeker->start();
}