| // Copyright (c) 2012 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/base64.h" |
| #include "base/file_util.h" |
| #include "base/files/file_path.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/values.h" |
| #include "chrome/common/chrome_switches.h" |
| #include "chrome/test/webdriver/webdriver_capabilities_parser.h" |
| #include "chrome/test/webdriver/webdriver_logging.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "third_party/zlib/google/zip.h" |
| |
| using base::DictionaryValue; |
| using base::ListValue; |
| using base::Value; |
| |
| namespace webdriver { |
| |
| TEST(CapabilitiesParser, NoCaps) { |
| Capabilities caps; |
| DictionaryValue dict; |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, SimpleCaps) { |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("chromeOptions", options); |
| |
| options->SetString("binary", "binary"); |
| options->SetString("channel", "channel"); |
| options->SetBoolean("detach", true); |
| options->SetBoolean("loadAsync", true); |
| |
| Capabilities caps; |
| base::ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_EQ(FILE_PATH_LITERAL("binary"), caps.command.GetProgram().value()); |
| EXPECT_STREQ("channel", caps.channel.c_str()); |
| EXPECT_TRUE(caps.detach); |
| EXPECT_TRUE(caps.load_async); |
| } |
| |
| TEST(CapabilitiesParser, Args) { |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("chromeOptions", options); |
| |
| ListValue* args = new ListValue(); |
| args->Append(new base::StringValue("arg1")); |
| args->Append(new base::StringValue("arg2=val")); |
| args->Append(new base::StringValue("arg3='a space'")); |
| options->Set("args", args); |
| |
| Capabilities caps; |
| base::ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_TRUE(caps.command.HasSwitch("arg1")); |
| EXPECT_STREQ("val", caps.command.GetSwitchValueASCII("arg2").c_str()); |
| EXPECT_STREQ("'a space'", caps.command.GetSwitchValueASCII("arg3").c_str()); |
| } |
| |
| TEST(CapabilitiesParser, Extensions) { |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("chromeOptions", options); |
| |
| ListValue* extensions = new ListValue(); |
| extensions->Append(new base::StringValue("TWFu")); |
| extensions->Append(new base::StringValue("TWFuTWFu")); |
| options->Set("extensions", extensions); |
| |
| Capabilities caps; |
| base::ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| ASSERT_EQ(2u, caps.extensions.size()); |
| std::string contents; |
| ASSERT_TRUE(base::ReadFileToString(caps.extensions[0], &contents)); |
| EXPECT_STREQ("Man", contents.c_str()); |
| contents.clear(); |
| ASSERT_TRUE(base::ReadFileToString(caps.extensions[1], &contents)); |
| EXPECT_STREQ("ManMan", contents.c_str()); |
| } |
| |
| TEST(CapabilitiesParser, Profile) { |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("chromeOptions", options); |
| |
| base::ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| base::FilePath folder = temp_dir.path().AppendASCII("folder"); |
| ASSERT_TRUE(file_util::CreateDirectory(folder)); |
| ASSERT_EQ(4, file_util::WriteFile( |
| folder.AppendASCII("data"), "data", 4)); |
| base::FilePath zip = temp_dir.path().AppendASCII("data.zip"); |
| ASSERT_TRUE(zip::Zip(folder, zip, false /* include_hidden_files */)); |
| std::string contents; |
| ASSERT_TRUE(base::ReadFileToString(zip, &contents)); |
| std::string base64; |
| ASSERT_TRUE(base::Base64Encode(contents, &base64)); |
| options->SetString("profile", base64); |
| |
| Capabilities caps; |
| CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| std::string new_contents; |
| ASSERT_TRUE(base::ReadFileToString( |
| caps.profile.AppendASCII("data"), &new_contents)); |
| EXPECT_STREQ("data", new_contents.c_str()); |
| } |
| |
| TEST(CapabilitiesParser, UnknownCap) { |
| Capabilities caps; |
| DictionaryValue dict; |
| dict.SetString("chromeOptions.nosuchcap", "none"); |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_TRUE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyCap) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| const char kPacUrl[] = "test.wpad"; |
| options->SetString("proxyType", "PAC"); |
| options->SetString("proxyAutoconfigUrl", kPacUrl); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_STREQ(kPacUrl, |
| caps.command.GetSwitchValueASCII(switches::kProxyPacUrl).c_str()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyTypeCapIncompatiblePac) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "pac"); |
| options->SetString("httpProxy", "http://localhost:8001"); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_TRUE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyTypeCapIncompatibleManual) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "manual"); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_TRUE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyTypeCapNullValue) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->Set("proxyType", base::Value::CreateNullValue()); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_TRUE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyTypeManualCap) { |
| const char kProxyServers[] = "ftp=localhost:9001;http=localhost:8001"; |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "manual"); |
| options->SetString("httpProxy", "localhost:8001"); |
| options->SetString("ftpProxy", "localhost:9001"); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_STREQ(kProxyServers, |
| caps.command.GetSwitchValueASCII(switches::kProxyServer).c_str()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyBypassListCap) { |
| const char kBypassList[] = "google.com, youtube.com"; |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "manual"); |
| options->SetString("noProxy", kBypassList); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_STREQ(kBypassList, |
| caps.command.GetSwitchValueASCII(switches::kProxyBypassList).c_str()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyBypassListCapNullValue) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "manual"); |
| options->Set("noProxy", base::Value::CreateNullValue()); |
| options->SetString("httpProxy", "localhost:8001"); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_FALSE(caps.command.HasSwitch(switches::kProxyBypassList)); |
| } |
| |
| TEST(CapabilitiesParser, UnknownProxyCap) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "DIRECT"); |
| options->SetString("badProxyCap", "error"); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, ProxyFtpServerCapNullValue) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("proxy", options); |
| |
| options->SetString("proxyType", "manual"); |
| options->SetString("httpProxy", "localhost:8001"); |
| options->Set("ftpProxy", base::Value::CreateNullValue()); |
| |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| EXPECT_STREQ("http=localhost:8001", |
| caps.command.GetSwitchValueASCII(switches::kProxyServer).c_str()); |
| } |
| |
| TEST(CapabilitiesParser, DriverLoggingCapString) { |
| Capabilities caps; |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("loggingPrefs", options); |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| |
| // A string as the driver logging level works. |
| options->SetString("driver", "INFO"); |
| ASSERT_FALSE(parser.Parse()); |
| |
| // An integer (here, an enum LogLevel value) doesn't work. |
| options->SetInteger("driver", kInfoLogLevel); |
| ASSERT_TRUE(parser.Parse()); |
| } |
| |
| TEST(CapabilitiesParser, ExcludeSwitches) { |
| DictionaryValue dict; |
| DictionaryValue* options = new DictionaryValue(); |
| dict.Set("chromeOptions", options); |
| |
| ListValue* switches = new ListValue(); |
| switches->Append(new base::StringValue(switches::kNoFirstRun)); |
| switches->Append(new base::StringValue(switches::kDisableSync)); |
| switches->Append(new base::StringValue(switches::kDisableTranslate)); |
| options->Set("excludeSwitches", switches); |
| |
| Capabilities caps; |
| CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| ASSERT_FALSE(parser.Parse()); |
| |
| const std::set<std::string>& rm_set = caps.exclude_switches; |
| EXPECT_EQ(static_cast<size_t>(3), rm_set.size()); |
| ASSERT_TRUE(rm_set.find(switches::kNoFirstRun) != rm_set.end()); |
| ASSERT_TRUE(rm_set.find(switches::kDisableSync) != rm_set.end()); |
| ASSERT_TRUE(rm_set.find(switches::kDisableTranslate) != rm_set.end()); |
| } |
| |
| } // namespace webdriver |