[chromedriver] Load the automation extension as a component extension.

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 "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());
+}
+
+TEST(Switches, NoValue) {
+  Switches switches;
+  switches.SetSwitch("hello");
+
+  CommandLine cmd(CommandLine::NO_PROGRAM);
+  switches.AppendToCommandLine(&cmd);
+  ASSERT_TRUE(cmd.HasSwitch("hello"));
+  ASSERT_EQ(FILE_PATH_LITERAL(""), cmd.GetSwitchValueNative("hello"));
+  ASSERT_EQ("--hello", switches.ToString());
+}
+
+TEST(Switches, Value) {
+  Switches switches;
+  switches.SetSwitch("hello", "there");
+
+  CommandLine cmd(CommandLine::NO_PROGRAM);
+  switches.AppendToCommandLine(&cmd);
+  ASSERT_TRUE(cmd.HasSwitch("hello"));
+  ASSERT_EQ(FILE_PATH_LITERAL("there"), cmd.GetSwitchValueNative("hello"));
+  ASSERT_EQ("--hello=there", switches.ToString());
+}
+
+TEST(Switches, Quoting) {
+  Switches switches;
+  switches.SetSwitch("hello", "a  b");
+  switches.SetSwitch("hello2", "  '\"  ");
+
+  ASSERT_EQ("--hello=\"a  b\" --hello2=\"  '\\\"  \"", switches.ToString());
+}
+
+TEST(Switches, Multiple) {
+  Switches switches;
+  switches.SetSwitch("switch");
+  switches.SetSwitch("hello", "there");
+
+  CommandLine cmd(CommandLine::NO_PROGRAM);
+  switches.AppendToCommandLine(&cmd);
+  ASSERT_TRUE(cmd.HasSwitch("switch"));
+  ASSERT_TRUE(cmd.HasSwitch("hello"));
+  ASSERT_EQ(FILE_PATH_LITERAL("there"), cmd.GetSwitchValueNative("hello"));
+  ASSERT_EQ("--hello=there --switch", switches.ToString());
+}
+
+TEST(Switches, Unparsed) {
+  Switches switches;
+  switches.SetUnparsedSwitch("a");
+  switches.SetUnparsedSwitch("--b");
+  switches.SetUnparsedSwitch("--c=1");
+  switches.SetUnparsedSwitch("d=1");
+  switches.SetUnparsedSwitch("-e=--1=1");
+
+  ASSERT_EQ("---e=--1=1 --a --b --c=1 --d=1", switches.ToString());
+}
+
 TEST(ParseCapabilities, WithAndroidPackage) {
   Capabilities capabilities;
   base::DictionaryValue caps;
   caps.SetString("chromeOptions.androidPackage", "abc");
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
   ASSERT_TRUE(capabilities.IsAndroid());
   ASSERT_EQ("abc", capabilities.android_package);
 }
@@ skipping 19 matching lines @@
 TEST(ParseCapabilities, LogPath) {
   Capabilities capabilities;
   base::DictionaryValue caps;
   caps.SetString("chromeOptions.logPath", "path/to/logfile");
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
   ASSERT_STREQ("path/to/logfile", capabilities.log_path.c_str());
 }
 
-TEST(ParseCapabilities, NoArgs) {
-  Capabilities capabilities;
-  base::ListValue args;
-  ASSERT_TRUE(args.empty());
-  base::DictionaryValue caps;
-  caps.Set("chromeOptions.args", args.DeepCopy());
-  Logger log(Log::kError);
-  Status status = capabilities.Parse(caps, &log);
-  ASSERT_TRUE(status.IsOk());
-  ASSERT_TRUE(capabilities.command.GetSwitches().empty());
-}
-
-TEST(ParseCapabilities, SingleArgWithoutValue) {
-  Capabilities capabilities;
-  base::ListValue args;
-  args.AppendString("enable-nacl");
-  ASSERT_EQ(1u, args.GetSize());
-  base::DictionaryValue caps;
-  caps.Set("chromeOptions.args", args.DeepCopy());
-  Logger log(Log::kError);
-  Status status = capabilities.Parse(caps, &log);
-  ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(1u, capabilities.command.GetSwitches().size());
-  ASSERT_TRUE(capabilities.command.HasSwitch("enable-nacl"));
-}
-
-TEST(ParseCapabilities, SingleArgWithValue) {
-  Capabilities capabilities;
-  base::ListValue args;
-  args.AppendString("load-extension=/test/extension");
-  ASSERT_EQ(1u, args.GetSize());
-  base::DictionaryValue caps;
-  caps.Set("chromeOptions.args", args.DeepCopy());
-  Logger log(Log::kError);
-  Status status = capabilities.Parse(caps, &log);
-  ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(1u, capabilities.command.GetSwitches().size());
-  ASSERT_TRUE(capabilities.command.HasSwitch("load-extension"));
-  ASSERT_STREQ(
-      "/test/extension",
-      capabilities.command.GetSwitchValueASCII("load-extension").c_str());
-}
-
-TEST(ParseCapabilities, MultipleArgs) {
-  Capabilities capabilities;
-  base::ListValue args;
-  args.AppendString("arg1");
-  args.AppendString("arg2=val");
-  args.AppendString("arg3='a space'");
-  ASSERT_EQ(3u, args.GetSize());
-  base::DictionaryValue caps;
-  caps.Set("chromeOptions.args", args.DeepCopy());
-  Logger log(Log::kError);
-  Status status = capabilities.Parse(caps, &log);
-  ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(3u, capabilities.command.GetSwitches().size());
-  ASSERT_TRUE(capabilities.command.HasSwitch("arg1"));
-  ASSERT_TRUE(capabilities.command.HasSwitch("arg2"));
-  ASSERT_STREQ("val", capabilities.command.GetSwitchValueASCII("arg2").c_str());
-  ASSERT_TRUE(capabilities.command.HasSwitch("arg3"));
-  ASSERT_STREQ("'a space'",
-               capabilities.command.GetSwitchValueASCII("arg3").c_str());
-}
-
 TEST(ParseCapabilities, Prefs) {
   Capabilities capabilities;
   base::DictionaryValue prefs;
   prefs.SetString("key1", "value1");
   prefs.SetString("key2.k", "value2");
   base::DictionaryValue caps;
   caps.Set("chromeOptions.prefs", prefs.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
@@ skipping 52 matching lines @@
 
 TEST(ParseCapabilities, DirectProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "DIRECT");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(1u, capabilities.command.GetSwitches().size());
-  ASSERT_TRUE(capabilities.command.HasSwitch("no-proxy-server"));
+  ASSERT_EQ(1u, capabilities.switches.GetSize());
+  ASSERT_TRUE(capabilities.switches.HasSwitch("no-proxy-server"));
 }
 
 TEST(ParseCapabilities, SystemProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "system");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
-  ASSERT_TRUE(capabilities.command.GetSwitches().empty());
+  ASSERT_EQ(0u, capabilities.switches.GetSize());
 }
 
 TEST(ParseCapabilities, PacProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "PAC");
   proxy.SetString("proxyAutoconfigUrl", "test.wpad");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(1u, capabilities.command.GetSwitches().size());
-  ASSERT_STREQ(
-      "test.wpad",
-      capabilities.command.GetSwitchValueASCII("proxy-pac-url").c_str());
+  ASSERT_EQ(1u, capabilities.switches.GetSize());
+  ASSERT_EQ("test.wpad", capabilities.switches.GetSwitchValue("proxy-pac-url"));
 }
 
 TEST(ParseCapabilities, MissingProxyAutoconfigUrl) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "PAC");
   proxy.SetString("httpProxy", "http://localhost:8001");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_FALSE(status.IsOk());
 }
 
 TEST(ParseCapabilities, AutodetectProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "autodetect");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(1u, capabilities.command.GetSwitches().size());
-  ASSERT_TRUE(capabilities.command.HasSwitch("proxy-auto-detect"));
+  ASSERT_EQ(1u, capabilities.switches.GetSize());
+  ASSERT_TRUE(capabilities.switches.HasSwitch("proxy-auto-detect"));
 }
 
 TEST(ParseCapabilities, ManualProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "manual");
   proxy.SetString("ftpProxy", "localhost:9001");
   proxy.SetString("httpProxy", "localhost:8001");
   proxy.SetString("sslProxy", "localhost:10001");
   proxy.SetString("noProxy", "google.com, youtube.com");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(2u, capabilities.command.GetSwitches().size());
-  ASSERT_STREQ(
+  ASSERT_EQ(2u, capabilities.switches.GetSize());
+  ASSERT_EQ(
       "ftp=localhost:9001;http=localhost:8001;https=localhost:10001",
-      capabilities.command.GetSwitchValueASCII("proxy-server").c_str());
-  ASSERT_STREQ(
+      capabilities.switches.GetSwitchValue("proxy-server"));
+  ASSERT_EQ(
       "google.com, youtube.com",
-      capabilities.command.GetSwitchValueASCII("proxy-bypass-list").c_str());
+      capabilities.switches.GetSwitchValue("proxy-bypass-list"));
 }
 
 TEST(ParseCapabilities, MissingSettingForManualProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "manual");
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_FALSE(status.IsOk());
 }
 
 TEST(ParseCapabilities, IgnoreNullValueForManualProxy) {
   Capabilities capabilities;
   base::DictionaryValue proxy;
   proxy.SetString("proxyType", "manual");
   proxy.SetString("ftpProxy", "localhost:9001");
   proxy.Set("sslProxy", base::Value::CreateNullValue());
   proxy.Set("noProxy", base::Value::CreateNullValue());
   base::DictionaryValue caps;
   caps.Set("proxy", proxy.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
-  ASSERT_EQ(1u, capabilities.command.GetSwitches().size());
-  ASSERT_TRUE(capabilities.command.HasSwitch("proxy-server"));
-  ASSERT_STREQ(
+  ASSERT_EQ(1u, capabilities.switches.GetSize());
+  ASSERT_TRUE(capabilities.switches.HasSwitch("proxy-server"));
+  ASSERT_EQ(
       "ftp=localhost:9001",
-      capabilities.command.GetSwitchValueASCII("proxy-server").c_str());
+      capabilities.switches.GetSwitchValue("proxy-server"));
 }
 
 TEST(ParseCapabilities, LoggingPrefsOk) {
   Capabilities capabilities;
   base::DictionaryValue logging_prefs;
   logging_prefs.SetString("Network", "INFO");
   base::DictionaryValue caps;
   caps.Set("loggingPrefs", logging_prefs.DeepCopy());
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
@@ skipping 33 matching lines @@
 TEST(ParseCapabilities, UseExistingBrowser) {
   Capabilities capabilities;
   base::DictionaryValue caps;
   caps.SetString("chromeOptions.useExistingBrowser", "abc:123");
   Logger log(Log::kError);
   Status status = capabilities.Parse(caps, &log);
   ASSERT_TRUE(status.IsOk());
   ASSERT_TRUE(capabilities.IsExistingBrowser());
   ASSERT_EQ(123, capabilities.existing_browser_port);
 }