[ChromeDriver] Implementing PerfLoggingPrefs for perf log

chrome/test/chromedriver/capabilities_unittest.cc

 // Copyright (c) 2013 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/test/chromedriver/capabilities.h"
 
 #include "base/values.h"
 #include "chrome/test/chromedriver/chrome/log.h"
 #include "chrome/test/chromedriver/chrome/status.h"
+#include "chrome/test/chromedriver/logging.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 TEST(Switches, Empty) {
   Switches switches;
   CommandLine cmd(CommandLine::NO_PROGRAM);
   switches.AppendToCommandLine(&cmd);
   ASSERT_EQ(0u, cmd.GetSwitches().size());
   ASSERT_EQ("", switches.ToString());
 }
 
@@ skipping 309 matching lines @@
 }
 
 TEST(ParseCapabilities, LoggingPrefsNotDict) {
   Capabilities capabilities;
   base::DictionaryValue caps;
   caps.SetString("loggingPrefs", "INFO");
   Status status = capabilities.Parse(caps);
   ASSERT_FALSE(status.IsOk());
 }
 
+TEST(ParseCapabilities, PerfLoggingPrefsInspectorDomainStatus) {
+  Capabilities capabilities;
+  // Perf log must be enabled if performance log preferences are specified.
+  base::DictionaryValue logging_prefs;
+  logging_prefs.SetString(WebDriverLog::kPerformanceType, "INFO");
+  base::DictionaryValue desired_caps;
+  desired_caps.Set("loggingPrefs", logging_prefs.DeepCopy());
+  ASSERT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kDefaultEnabled,
+            capabilities.perf_logging_prefs.network);
+  ASSERT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kDefaultEnabled,
+            capabilities.perf_logging_prefs.page);
+  ASSERT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kDefaultEnabled,
+            capabilities.perf_logging_prefs.timeline);
+  base::DictionaryValue perf_logging_prefs;
+  perf_logging_prefs.SetBoolean("enableNetwork", true);
+  perf_logging_prefs.SetBoolean("enablePage", false);
+  desired_caps.Set("chromeOptions.perfLoggingPrefs",
+                   perf_logging_prefs.DeepCopy());
+  Status status = capabilities.Parse(desired_caps);
+  ASSERT_TRUE(status.IsOk());
+  ASSERT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kExplicitlyEnabled,
+            capabilities.perf_logging_prefs.network);
+  ASSERT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kExplicitlyDisabled,
+            capabilities.perf_logging_prefs.page);
+  ASSERT_EQ(PerfLoggingPrefs::InspectorDomainStatus::kDefaultEnabled,
+            capabilities.perf_logging_prefs.timeline);
+}
+
+TEST(ParseCapabilities, PerfLoggingPrefsTraceCategories) {
+  Capabilities capabilities;
+  // Perf log must be enabled if performance log preferences are specified.
+  base::DictionaryValue logging_prefs;
+  logging_prefs.SetString(WebDriverLog::kPerformanceType, "INFO");
+  base::DictionaryValue desired_caps;
+  desired_caps.Set("loggingPrefs", logging_prefs.DeepCopy());
+  ASSERT_EQ("", capabilities.perf_logging_prefs.trace_categories);
+  base::DictionaryValue perf_logging_prefs;
+  perf_logging_prefs.SetString("traceCategories", "benchmark,webkit.console");
+  desired_caps.Set("chromeOptions.perfLoggingPrefs",
+                   perf_logging_prefs.DeepCopy());
+  Status status = capabilities.Parse(desired_caps);
+  ASSERT_TRUE(status.IsOk());
+  ASSERT_EQ("benchmark,webkit.console",
+            capabilities.perf_logging_prefs.trace_categories);
+}
+
+TEST(ParseCapabilities, PerfLoggingPrefsNotDict) {
+  Capabilities capabilities;
+  // Perf log must be enabled if performance log preferences are specified.
+  base::DictionaryValue logging_prefs;
+  logging_prefs.SetString(WebDriverLog::kPerformanceType, "INFO");
+  base::DictionaryValue desired_caps;
+  desired_caps.Set("loggingPrefs", logging_prefs.DeepCopy());
+  desired_caps.SetString("chromeOptions.perfLoggingPrefs", "traceCategories");
+  Status status = capabilities.Parse(desired_caps);
+  ASSERT_FALSE(status.IsOk());
+}
+
+TEST(ParseCapabilities, PerfLoggingPrefsNoPerfLogLevel) {
+  Capabilities capabilities;
+  base::DictionaryValue desired_caps;
+  base::DictionaryValue perf_logging_prefs;
+  perf_logging_prefs.SetBoolean("enableNetwork", true);
+  desired_caps.Set("chromeOptions.perfLoggingPrefs",
+                   perf_logging_prefs.DeepCopy());
+  // Should fail because perf log must be enabled if perf log prefs specified.
+  Status status = capabilities.Parse(desired_caps);
+  ASSERT_FALSE(status.IsOk());
+}
+
+TEST(ParseCapabilities, PerfLoggingPrefsPerfLogOff) {
+  Capabilities capabilities;
+  base::DictionaryValue logging_prefs;
+  // Disable performance log by setting logging level to OFF.
+  logging_prefs.SetString(WebDriverLog::kPerformanceType, "OFF");
+  base::DictionaryValue desired_caps;
+  desired_caps.Set("loggingPrefs", logging_prefs.DeepCopy());
+  base::DictionaryValue perf_logging_prefs;
+  perf_logging_prefs.SetBoolean("enableNetwork", true);
+  desired_caps.Set("chromeOptions.perfLoggingPrefs",
+                   perf_logging_prefs.DeepCopy());
+  // Should fail because perf log must be enabled if perf log prefs specified.
+  Status status = capabilities.Parse(desired_caps);
+  ASSERT_FALSE(status.IsOk());
+}
+
 TEST(ParseCapabilities, ExcludeSwitches) {
   Capabilities capabilities;
   base::ListValue exclude_switches;
   exclude_switches.AppendString("switch1");
   exclude_switches.AppendString("switch2");
   base::DictionaryValue caps;
   caps.Set("chromeOptions.excludeSwitches", exclude_switches.DeepCopy());
   Status status = capabilities.Parse(caps);
   ASSERT_TRUE(status.IsOk());
   ASSERT_EQ(2u, capabilities.exclude_switches.size());
@@ skipping 100 matching lines @@
   base::DictionaryValue mobile_emulation;
   mobile_emulation.SetString("deviceName", "Google Nexus 5");
   mobile_emulation.SetInteger("deviceMetrics.width", 360);
   mobile_emulation.SetInteger("deviceMetrics.height", 640);
   mobile_emulation.SetDouble("deviceMetrics.pixelRatio", 3.0);
   base::DictionaryValue caps;
   caps.Set("chromeOptions.mobileEmulation", mobile_emulation.DeepCopy());
   Status status = capabilities.Parse(caps);
   ASSERT_FALSE(status.IsOk());
 }