content/browser/loader/resource_scheduler_browsertest.cc

// Copyright 2017 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 <map>
#include <string>

#include "base/memory/ptr_util.h"
#include "base/metrics/field_trial.h"
#include "base/metrics/field_trial_param_associator.h"
#include "base/metrics/field_trial_params.h"
#include "base/test/scoped_feature_list.h"
#include "content/public/test/browser_test_utils.h"
#include "content/public/test/content_browser_test.h"
#include "content/public/test/content_browser_test_utils.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "url/gurl.h"

namespace content {

namespace {

class ResourceSchedulerBrowserTest : public ContentBrowserTest {
 protected:
  ResourceSchedulerBrowserTest() : field_trial_list_(nullptr) {}

  void SetUpInProcessBrowserTestFixture() override {
    ASSERT_TRUE(embedded_test_server()->Start());
    InitializeMaxDelayableRequestsExperiment();
  }

  void InitializeMaxDelayableRequestsExperiment() {
    base::FieldTrialParamAssociator::GetInstance()->ClearAllParamsForTesting();
    const char kTrialName[] = "TrialName";
    const char kGroupName[] = "GroupName";
    const char kMaxDelayableRequestsNetworkOverride[] =
        "MaxDelayableRequestsNetworkOverride";

    std::map<std::string, std::string> params;
    params["MaxEffectiveConnectionType"] = "2G";
    params["MaxBDPKbits1"] = "130";
    params["MaxDelayableRequests1"] = "2";
    params["MaxBDPKbits2"] = "160";
    params["MaxDelayableRequests2"] = "4";

    base::AssociateFieldTrialParams(kTrialName, kGroupName, params);
    base::FieldTrial* field_trial =
        base::FieldTrialList::CreateFieldTrial(kTrialName, kGroupName);

    std::unique_ptr<base::FeatureList> feature_list(
        base::MakeUnique<base::FeatureList>());
    feature_list->RegisterFieldTrialOverride(
        kMaxDelayableRequestsNetworkOverride,
        base::FeatureList::OVERRIDE_ENABLE_FEATURE, field_trial);
    feature_list_.InitWithFeatureList(std::move(feature_list));
  }

 private:
  base::FieldTrialList field_trial_list_;
  base::test::ScopedFeatureList feature_list_;
};

IN_PROC_BROWSER_TEST_F(ResourceSchedulerBrowserTest,
                       ResourceLoadingExperimentIncognito) {
  GURL url(embedded_test_server()->GetURL(
      "/resource_loading/resource_loading_non_mobile.html"));

  Shell* otr_browser = CreateOffTheRecordBrowser();
  EXPECT_TRUE(NavigateToURL(otr_browser, url));
  int data = -1;
  EXPECT_TRUE(
      ExecuteScriptAndExtractInt(otr_browser, "getResourceNumber()", &data));
  EXPECT_EQ(9, data);
}

IN_PROC_BROWSER_TEST_F(ResourceSchedulerBrowserTest,
                       ResourceLoadingExperimentNormal) {
  GURL url(embedded_test_server()->GetURL(
      "/resource_loading/resource_loading_non_mobile.html"));
  Shell* browser = shell();
  EXPECT_TRUE(NavigateToURL(browser, url));
  int data = -1;
  EXPECT_TRUE(
      ExecuteScriptAndExtractInt(browser, "getResourceNumber()", &data));
  EXPECT_EQ(9, data);
}

}  // anonymous namespace

}  // namespace content