mirror of
https://github.com/facebookincubator/mvfst.git
synced 2025-08-09 20:42:44 +03:00
5f51f4436f
Summary: no more surprises in upper layer Reviewed By: mjoras Differential Revision: D19976510 fbshipit-source-id: 3487e9aa2cb28d7bc748f13bc2bbc393216b4a8a
71 lines
2.6 KiB
C++
71 lines
2.6 KiB
C++
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
#include <quic/server/state/ServerStateMachine.h>
|
|
#include <gtest/gtest.h>
|
|
#include <quic/codec/DefaultConnectionIdAlgo.h>
|
|
#include <quic/codec/QuicConnectionId.h>
|
|
|
|
namespace quic {
|
|
|
|
namespace {
|
|
void assertServerConnIdParamsEq(
|
|
ServerConnectionIdParams& first,
|
|
ServerConnectionIdParams& second) {
|
|
EXPECT_EQ(first.version, second.version);
|
|
EXPECT_EQ(first.hostId, second.hostId);
|
|
EXPECT_EQ(first.processId, second.processId);
|
|
EXPECT_EQ(first.workerId, second.workerId);
|
|
}
|
|
} // namespace
|
|
|
|
namespace test {
|
|
TEST(ServerStateMachineTest, TestAddConnId) {
|
|
QuicServerConnectionState serverState;
|
|
ServerConnectionIdParams originalParams(12, 1, 37);
|
|
|
|
auto algo = std::make_unique<DefaultConnectionIdAlgo>();
|
|
|
|
serverState.connIdAlgo = algo.get();
|
|
serverState.serverConnIdParams = originalParams;
|
|
serverState.serverAddr = folly::SocketAddress("0.0.0.0", 42069);
|
|
|
|
std::array<uint8_t, kStatelessResetTokenSecretLength> secret;
|
|
serverState.transportSettings.statelessResetTokenSecret = secret;
|
|
EXPECT_EQ(serverState.selfConnectionIds.size(), 0);
|
|
serverState.peerActiveConnectionIdLimit = 2;
|
|
auto newConnId1 = serverState.createAndAddNewSelfConnId();
|
|
auto newConnId2 = serverState.createAndAddNewSelfConnId();
|
|
auto newConnId3 = serverState.createAndAddNewSelfConnId();
|
|
|
|
// Sequence numbers correctly set.
|
|
EXPECT_EQ(newConnId1->sequenceNumber, 0);
|
|
EXPECT_EQ(newConnId2->sequenceNumber, 1);
|
|
EXPECT_EQ(newConnId3->sequenceNumber, 2);
|
|
|
|
// All three conn ids are different from each other.
|
|
EXPECT_NE(newConnId1->connId, newConnId2->connId);
|
|
EXPECT_NE(newConnId2->connId, newConnId3->connId);
|
|
EXPECT_NE(newConnId3->connId, newConnId1->connId);
|
|
|
|
EXPECT_EQ(newConnId1->token->size(), kStatelessResetTokenLength);
|
|
EXPECT_EQ(newConnId2->token->size(), kStatelessResetTokenLength);
|
|
EXPECT_EQ(newConnId3->token->size(), kStatelessResetTokenLength);
|
|
|
|
auto params1 = *serverState.connIdAlgo->parseConnectionId(newConnId1->connId);
|
|
auto params2 = *serverState.connIdAlgo->parseConnectionId(newConnId2->connId);
|
|
auto params3 = *serverState.connIdAlgo->parseConnectionId(newConnId3->connId);
|
|
|
|
// Server connection id params are correctly encoded/decoded.
|
|
assertServerConnIdParamsEq(originalParams, params1);
|
|
assertServerConnIdParamsEq(params1, params2);
|
|
assertServerConnIdParamsEq(params2, params3);
|
|
|
|
auto newConnId4 = serverState.createAndAddNewSelfConnId();
|
|
EXPECT_EQ(folly::none, newConnId4);
|
|
|
|
EXPECT_EQ(serverState.selfConnectionIds.size(), 3);
|
|
EXPECT_EQ(serverState.nextSelfConnectionIdSequence, 3);
|
|
}
|
|
} // namespace test
|
|
} // namespace quic
|