lib/mvfst
1
0
Fork 0
mirror of https://github.com/facebookincubator/mvfst.git synced 2025-08-08 09:42:06 +03:00
Code Activity

Move TestPacketBatchWriter to QUIC TestUtils

Browse Source 
Summary:
will be used in other test cases too

(Note: this ignores all push blocking failures!)

Reviewed By: mjoras

Differential Revision: D27786212

fbshipit-source-id: 2e21073962f20bd69ba1d16ae6d5752c9ee75b74
This commit is contained in:
Yang Chi
2021-04-29 08:41:42 -07:00
committed by Facebook GitHub Bot
parent 1943d4047b
commit fd554e17ff
3 changed files with 50 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
quic/api/test/IoBufQuicBatchTest.cpp
View File

@@ -10,6 +10,7 @@
#include <gtest/gtest.h>
#include <quic/client/state/ClientStateMachine.h>
#include <quic/common/test/TestUtils.h>
#include <quic/fizz/client/handshake/FizzClientQuicHandshakeContext.h>
#include <quic/state/StateData.h>
@@ -18,45 +19,11 @@ constexpr const auto kMaxBufs = 10;
namespace quic {
namespace testing {
class TestPacketBatchWriter : public IOBufBatchWriter {
public:
explicit TestPacketBatchWriter(int maxBufs) : maxBufs_(maxBufs) {}
~TestPacketBatchWriter() override {
CHECK_EQ(bufNum_, 0);
CHECK_EQ(bufSize_, 0);
}
void reset() override {
bufNum_ = 0;
bufSize_ = 0;
}
bool append(
std::unique_ptr<folly::IOBuf>&& /*unused*/,
size_t size,
const folly::SocketAddress& /*unused*/,
folly::AsyncUDPSocket* /*unused*/) override {
bufNum_++;
bufSize_ += size;
return ((maxBufs_ < 0) || (bufNum_ >= maxBufs_));
}
ssize_t write(
folly::AsyncUDPSocket& /*unused*/,
const folly::SocketAddress& /*unused*/) override {
return bufSize_;
}
private:
int maxBufs_{0};
int bufNum_{0};
size_t bufSize_{0};
};
void RunTest(int numBatch) {
folly::EventBase evb;
folly::AsyncUDPSocket sock(&evb);
auto batchWriter = BatchWriterPtr(new TestPacketBatchWriter(numBatch));
auto batchWriter = BatchWriterPtr(new test::TestPacketBatchWriter(numBatch));
folly::SocketAddress peerAddress{"127.0.0.1", 1234};
QuicClientConnectionState conn(
FizzClientQuicHandshakeContext::Builder().build());