third_party/blink/common/client_hints/client_hints_unittest.cc

// Copyright 2019 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 "third_party/blink/public/common/client_hints/client_hints.h"

#include <iostream>

#include "base/test/scoped_feature_list.h"
#include "services/network/public/mojom/web_client_hints_types.mojom-shared.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/features.h"
#include "url/gurl.h"

using testing::UnorderedElementsAre;

namespace blink {

// Checks that the removed header list doesn't include legacy headers nor the
// on-by-default ones, when the kAllowClientHintsToThirdParty flag is on.
TEST(ClientHintsTest, FindClientHintsToRemoveLegacy) {
  base::test::ScopedFeatureList scoped_feature_list;
  scoped_feature_list.InitAndEnableFeature(
      features::kAllowClientHintsToThirdParty);
  std::vector<std::string> removed_headers;
  FindClientHintsToRemove(nullptr, GURL(), &removed_headers);
  EXPECT_THAT(
      removed_headers,
      UnorderedElementsAre(
          "rtt", "downlink", "ect", "sec-ch-ua-arch", "sec-ch-ua-model",
          "sec-ch-ua-full-version", "sec-ch-ua-platform-version",
          "sec-ch-prefers-color-scheme", "sec-ch-ua-bitness",
          "sec-ch-ua-reduced", "sec-ch-viewport-height", "sec-ch-device-memory",
          "sec-ch-dpr", "sec-ch-width", "sec-ch-viewport-width",
          "sec-ch-ua-full-version-list", "sec-ch-ua-full", "sec-ch-ua-wow64"));
}

// Checks that the removed header list includes legacy headers but not the
// on-by-default ones, when the kAllowClientHintsToThirdParty flag is off.
TEST(ClientHintsTest, FindClientHintsToRemoveNoLegacy) {
  base::test::ScopedFeatureList scoped_feature_list;
  scoped_feature_list.InitAndDisableFeature(
      features::kAllowClientHintsToThirdParty);
  std::vector<std::string> removed_headers;
  FindClientHintsToRemove(nullptr, GURL(), &removed_headers);
  EXPECT_THAT(
      removed_headers,
      UnorderedElementsAre(
          "device-memory", "dpr", "width", "viewport-width", "rtt", "downlink",
          "ect", "sec-ch-ua-arch", "sec-ch-ua-model", "sec-ch-ua-full-version",
          "sec-ch-ua-platform-version", "sec-ch-prefers-color-scheme",
          "sec-ch-ua-bitness", "sec-ch-ua-reduced", "sec-ch-viewport-height",
          "sec-ch-device-memory", "sec-ch-dpr", "sec-ch-width",
          "sec-ch-viewport-width", "sec-ch-ua-full-version-list",
          "sec-ch-ua-full", "sec-ch-ua-wow64"));
}
}  // namespace blink