mirror of
https://github.com/facebookincubator/mvfst.git
synced 2025-04-18 17:24:03 +03:00
00e67c1bf9
Reviewed By: lnicco Differential Revision: D33587012 fbshipit-source-id: 972eb440f0156c9c04aa6e8787561b18295c1a97
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#include <folly/Optional.h>
|
|
#include <folly/portability/GTest.h>
|
|
|
|
#include <quic/QuicException.h>
|
|
#include <quic/codec/QuicHeaderCodec.h>
|
|
#include <quic/common/test/TestUtils.h>
|
|
|
|
using namespace testing;
|
|
|
|
namespace quic {
|
|
namespace test {
|
|
|
|
class QuicHeaderCodecTest : public Test {};
|
|
|
|
TEST_F(QuicHeaderCodecTest, EmptyBuffer) {
|
|
auto emptyBuffer = folly::IOBuf::create(0);
|
|
EXPECT_FALSE(parseHeader(*emptyBuffer).hasValue());
|
|
}
|
|
|
|
TEST_F(QuicHeaderCodecTest, TooSmallBuffer) {
|
|
auto smallBuffer = folly::IOBuf::create(1);
|
|
smallBuffer->append(1);
|
|
folly::io::RWPrivateCursor wcursor(smallBuffer.get());
|
|
wcursor.writeBE<uint8_t>(0x01);
|
|
EXPECT_FALSE(parseHeader(*smallBuffer).hasValue());
|
|
}
|
|
|
|
TEST_F(QuicHeaderCodecTest, VersionNegotiationPacketTest) {
|
|
ConnectionId srcConnId = getTestConnectionId(0),
|
|
destConnId = getTestConnectionId(1);
|
|
QuicVersion version = MVFST1;
|
|
std::vector<QuicVersion> versions{version};
|
|
VersionNegotiationPacketBuilder builder(srcConnId, destConnId, versions);
|
|
auto packet = std::move(builder).buildPacket();
|
|
auto result = parseHeader(*packet.second);
|
|
EXPECT_TRUE(result->isVersionNegotiation);
|
|
}
|
|
|
|
TEST_F(QuicHeaderCodecTest, ShortHeaderTest) {
|
|
PacketNum packetNum = 1;
|
|
RegularQuicPacketBuilder builder(
|
|
kDefaultUDPSendPacketLen,
|
|
ShortHeader(
|
|
ProtectionType::KeyPhaseZero, getTestConnectionId(), packetNum),
|
|
0 /* largestAcked */);
|
|
builder.encodePacketHeader();
|
|
auto packet = std::move(builder).buildPacket();
|
|
auto result = parseHeader(*packet.header);
|
|
auto& header = result->parsedHeader;
|
|
LongHeader* longHeader = header->asLong();
|
|
if (longHeader) {
|
|
EXPECT_EQ(getTestConnectionId(), longHeader->getDestinationConnId());
|
|
} else {
|
|
EXPECT_EQ(getTestConnectionId(), header->asShort()->getConnectionId());
|
|
}
|
|
}
|
|
} // namespace test
|
|
} // namespace quic
|