net/quic/core/quic_version_manager_test.cc

// Copyright (c) 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/quic/core/quic_version_manager.h"

#include "net/quic/core/quic_flags.h"
#include "net/quic/core/quic_versions.h"
#include "net/quic/test_tools/quic_test_utils.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {
namespace test {
namespace {

TEST(QuicVersionManagerTest, QuicVersionManager) {
  QuicFlagSaver flags;
  FLAGS_quic_reloadable_flag_quic_enable_version_36_v3 = false;
  FLAGS_quic_reloadable_flag_quic_enable_version_37 = false;
  FLAGS_quic_reloadable_flag_quic_disable_version_34 = false;
  QuicVersionManager manager(AllSupportedVersions());
  EXPECT_EQ(FilterSupportedVersions(AllSupportedVersions()),
            manager.GetSupportedVersions());

  FLAGS_quic_reloadable_flag_quic_enable_version_36_v3 = true;
  EXPECT_EQ(FilterSupportedVersions(AllSupportedVersions()),
            manager.GetSupportedVersions());
  ASSERT_EQ(3u, manager.GetSupportedVersions().size());
  EXPECT_EQ(QUIC_VERSION_36, manager.GetSupportedVersions()[0]);
  EXPECT_EQ(QUIC_VERSION_35, manager.GetSupportedVersions()[1]);
  EXPECT_EQ(QUIC_VERSION_34, manager.GetSupportedVersions()[2]);

  FLAGS_quic_reloadable_flag_quic_enable_version_37 = true;
  EXPECT_EQ(FilterSupportedVersions(AllSupportedVersions()),
            manager.GetSupportedVersions());
  ASSERT_EQ(4u, manager.GetSupportedVersions().size());
  EXPECT_EQ(QUIC_VERSION_37, manager.GetSupportedVersions()[0]);
  EXPECT_EQ(QUIC_VERSION_36, manager.GetSupportedVersions()[1]);
  EXPECT_EQ(QUIC_VERSION_35, manager.GetSupportedVersions()[2]);
  EXPECT_EQ(QUIC_VERSION_34, manager.GetSupportedVersions()[3]);

  FLAGS_quic_reloadable_flag_quic_disable_version_34 = true;
  EXPECT_EQ(FilterSupportedVersions(AllSupportedVersions()),
            manager.GetSupportedVersions());
  ASSERT_EQ(3u, manager.GetSupportedVersions().size());
  EXPECT_EQ(QUIC_VERSION_37, manager.GetSupportedVersions()[0]);
  EXPECT_EQ(QUIC_VERSION_36, manager.GetSupportedVersions()[1]);
  EXPECT_EQ(QUIC_VERSION_35, manager.GetSupportedVersions()[2]);
}

}  // namespace
}  // namespace test
}  // namespace net