chrome/common/google_url_loader_throttle_unittest.cc

// Copyright 2020 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 "chrome/common/google_url_loader_throttle.h"

#include "base/test/scoped_feature_list.h"
#include "build/build_config.h"
#include "chrome/common/chrome_features.h"
#include "chrome/common/renderer_configuration.mojom.h"
#include "testing/gtest/include/gtest/gtest.h"

class GoogleURLLoaderThrottleTest : public testing::Test {
 public:
  GoogleURLLoaderThrottleTest() {
    scoped_feature_list_.InitWithFeatureList(
        std::make_unique<base::FeatureList>());
  }

  ~GoogleURLLoaderThrottleTest() override = default;

  base::test::ScopedFeatureList& scoped_feature_list() {
    return scoped_feature_list_;
  }

 private:
  base::test::ScopedFeatureList scoped_feature_list_;
};

#if defined(OS_ANDROID)
TEST_F(GoogleURLLoaderThrottleTest, DarkSearchGoogleHomepage) {
  scoped_feature_list().Reset();
  scoped_feature_list().InitAndEnableFeature(features::kAndroidDarkSearch);
  GoogleURLLoaderThrottle throttle(/* client_header= */ "",
                                   /* night_mode_enabled= */ true,
                                   chrome::mojom::DynamicParams());

  network::ResourceRequest request;
  request.url = GURL("https://www.google.com");
  bool defer = false;

  throttle.WillStartRequest(&request, &defer);
  EXPECT_NE(std::string::npos, request.url.spec().find("cs=1"));
}

TEST_F(GoogleURLLoaderThrottleTest, DarkSearchGoogleSearch) {
  scoped_feature_list().Reset();
  scoped_feature_list().InitAndEnableFeature(features::kAndroidDarkSearch);
  GoogleURLLoaderThrottle throttle(/* client_header= */ "",
                                   /* night_mode_enabled= */ true,
                                   chrome::mojom::DynamicParams());

  network::ResourceRequest request;
  request.url = GURL("https://www.google.com/search?q=test");
  bool defer = false;

  throttle.WillStartRequest(&request, &defer);
  EXPECT_NE(std::string::npos, request.url.spec().find("cs=1"));
}

TEST_F(GoogleURLLoaderThrottleTest, DarkSearchGoogleSource) {
  scoped_feature_list().Reset();
  scoped_feature_list().InitAndEnableFeature(features::kAndroidDarkSearch);
  GoogleURLLoaderThrottle throttle(/* client_header= */ "",
                                   /* night_mode_enabled= */ true,
                                   chrome::mojom::DynamicParams());

  network::ResourceRequest request;
  request.url = GURL("https://code.google.com/");
  bool defer = false;

  throttle.WillStartRequest(&request, &defer);
  EXPECT_EQ(std::string::npos, request.url.spec().find("cs=1"));
}
#endif