chrome/test/chromedriver/chrome/network_conditions_override_manager_unittest.cc

// Copyright (c) 2015 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 "base/values.h"
#include "chrome/test/chromedriver/chrome/network_conditions.h"
#include "chrome/test/chromedriver/chrome/network_conditions_override_manager.h"
#include "chrome/test/chromedriver/chrome/recorder_devtools_client.h"
#include "chrome/test/chromedriver/chrome/status.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

void AssertNetworkConditionsCommand(
    const Command& command,
    const NetworkConditions& network_conditions) {
  ASSERT_EQ("Network.emulateNetworkConditions", command.method);
  bool offline;
  double latency, download_throughput, upload_throughput;
  ASSERT_TRUE(command.params.GetBoolean("offline", &offline));
  ASSERT_TRUE(command.params.GetDouble("latency", &latency));
  ASSERT_TRUE(command.params.GetDouble("downloadThroughput",
                                       &download_throughput));
  ASSERT_TRUE(command.params.GetDouble("uploadThroughput",
                                       &upload_throughput));
  ASSERT_EQ(network_conditions.offline, offline);
  ASSERT_EQ(network_conditions.latency, latency);
  ASSERT_EQ(network_conditions.download_throughput, download_throughput);
  ASSERT_EQ(network_conditions.upload_throughput, upload_throughput);
}

}  // namespace

TEST(NetworkConditionsOverrideManager, OverrideSendsCommand) {
  RecorderDevToolsClient client;
  NetworkConditionsOverrideManager manager(&client);
  NetworkConditions network_conditions = {false, 100, 750*1024, 750*1024};
  manager.OverrideNetworkConditions(network_conditions);
  ASSERT_EQ(3u, client.commands_.size());
  ASSERT_NO_FATAL_FAILURE(
     AssertNetworkConditionsCommand(client.commands_[2], network_conditions));

  network_conditions.latency = 200;
  manager.OverrideNetworkConditions(network_conditions);
  ASSERT_EQ(6u, client.commands_.size());
  ASSERT_NO_FATAL_FAILURE(
      AssertNetworkConditionsCommand(client.commands_[5], network_conditions));
}

TEST(NetworkConditionsOverrideManager, SendsCommandOnConnect) {
  RecorderDevToolsClient client;
  NetworkConditionsOverrideManager manager(&client);
  ASSERT_EQ(0u, client.commands_.size());
  ASSERT_EQ(kOk, manager.OnConnected(&client).code());

  NetworkConditions network_conditions = {false, 100, 750*1024, 750*1024};
  manager.OverrideNetworkConditions(network_conditions);
  ASSERT_EQ(3u, client.commands_.size());
  ASSERT_EQ(kOk, manager.OnConnected(&client).code());
  ASSERT_EQ(6u, client.commands_.size());
  ASSERT_NO_FATAL_FAILURE(
      AssertNetworkConditionsCommand(client.commands_[5], network_conditions));
}

TEST(NetworkConditionsOverrideManager, SendsCommandOnNavigation) {
  RecorderDevToolsClient client;
  NetworkConditionsOverrideManager manager(&client);
  base::DictionaryValue main_frame_params;
  ASSERT_EQ(kOk,
            manager.OnEvent(&client, "Page.frameNavigated", main_frame_params)
                .code());
  ASSERT_EQ(0u, client.commands_.size());

  NetworkConditions network_conditions = {false, 100, 750*1024, 750*1024};
  manager.OverrideNetworkConditions(network_conditions);
  ASSERT_EQ(3u, client.commands_.size());
  ASSERT_EQ(kOk,
            manager.OnEvent(&client, "Page.frameNavigated", main_frame_params)
                .code());
  ASSERT_EQ(6u, client.commands_.size());
  ASSERT_NO_FATAL_FAILURE(
      AssertNetworkConditionsCommand(client.commands_[2], network_conditions));

  base::DictionaryValue sub_frame_params;
  sub_frame_params.SetString("frame.parentId", "id");
  ASSERT_EQ(
      kOk,
      manager.OnEvent(&client, "Page.frameNavigated", sub_frame_params).code());
  ASSERT_EQ(6u, client.commands_.size());
  ASSERT_NO_FATAL_FAILURE(
      AssertNetworkConditionsCommand(client.commands_[5], network_conditions));
}