| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/base64.h" | 5 #include "base/base64.h" |
| 6 #include "base/file_util.h" | 6 #include "base/file_util.h" |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 8 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/values.h" | 9 #include "base/values.h" |
| 10 #include "chrome/common/chrome_switches.h" | 10 #include "chrome/common/chrome_switches.h" |
| (...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 78 extensions->Append(new base::StringValue("TWFuTWFu")); | 78 extensions->Append(new base::StringValue("TWFuTWFu")); |
| 79 options->Set("extensions", extensions); | 79 options->Set("extensions", extensions); |
| 80 | 80 |
| 81 Capabilities caps; | 81 Capabilities caps; |
| 82 base::ScopedTempDir temp_dir; | 82 base::ScopedTempDir temp_dir; |
| 83 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 83 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 84 CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); | 84 CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); |
| 85 ASSERT_FALSE(parser.Parse()); | 85 ASSERT_FALSE(parser.Parse()); |
| 86 ASSERT_EQ(2u, caps.extensions.size()); | 86 ASSERT_EQ(2u, caps.extensions.size()); |
| 87 std::string contents; | 87 std::string contents; |
| 88 ASSERT_TRUE(file_util::ReadFileToString(caps.extensions[0], &contents)); | 88 ASSERT_TRUE(base::ReadFileToString(caps.extensions[0], &contents)); |
| 89 EXPECT_STREQ("Man", contents.c_str()); | 89 EXPECT_STREQ("Man", contents.c_str()); |
| 90 contents.clear(); | 90 contents.clear(); |
| 91 ASSERT_TRUE(file_util::ReadFileToString(caps.extensions[1], &contents)); | 91 ASSERT_TRUE(base::ReadFileToString(caps.extensions[1], &contents)); |
| 92 EXPECT_STREQ("ManMan", contents.c_str()); | 92 EXPECT_STREQ("ManMan", contents.c_str()); |
| 93 } | 93 } |
| 94 | 94 |
| 95 TEST(CapabilitiesParser, Profile) { | 95 TEST(CapabilitiesParser, Profile) { |
| 96 DictionaryValue dict; | 96 DictionaryValue dict; |
| 97 DictionaryValue* options = new DictionaryValue(); | 97 DictionaryValue* options = new DictionaryValue(); |
| 98 dict.Set("chromeOptions", options); | 98 dict.Set("chromeOptions", options); |
| 99 | 99 |
| 100 base::ScopedTempDir temp_dir; | 100 base::ScopedTempDir temp_dir; |
| 101 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 101 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 102 base::FilePath folder = temp_dir.path().AppendASCII("folder"); | 102 base::FilePath folder = temp_dir.path().AppendASCII("folder"); |
| 103 ASSERT_TRUE(file_util::CreateDirectory(folder)); | 103 ASSERT_TRUE(file_util::CreateDirectory(folder)); |
| 104 ASSERT_EQ(4, file_util::WriteFile( | 104 ASSERT_EQ(4, file_util::WriteFile( |
| 105 folder.AppendASCII("data"), "data", 4)); | 105 folder.AppendASCII("data"), "data", 4)); |
| 106 base::FilePath zip = temp_dir.path().AppendASCII("data.zip"); | 106 base::FilePath zip = temp_dir.path().AppendASCII("data.zip"); |
| 107 ASSERT_TRUE(zip::Zip(folder, zip, false /* include_hidden_files */)); | 107 ASSERT_TRUE(zip::Zip(folder, zip, false /* include_hidden_files */)); |
| 108 std::string contents; | 108 std::string contents; |
| 109 ASSERT_TRUE(file_util::ReadFileToString(zip, &contents)); | 109 ASSERT_TRUE(base::ReadFileToString(zip, &contents)); |
| 110 std::string base64; | 110 std::string base64; |
| 111 ASSERT_TRUE(base::Base64Encode(contents, &base64)); | 111 ASSERT_TRUE(base::Base64Encode(contents, &base64)); |
| 112 options->SetString("profile", base64); | 112 options->SetString("profile", base64); |
| 113 | 113 |
| 114 Capabilities caps; | 114 Capabilities caps; |
| 115 CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); | 115 CapabilitiesParser parser(&dict, temp_dir.path(), Logger(), &caps); |
| 116 ASSERT_FALSE(parser.Parse()); | 116 ASSERT_FALSE(parser.Parse()); |
| 117 std::string new_contents; | 117 std::string new_contents; |
| 118 ASSERT_TRUE(file_util::ReadFileToString( | 118 ASSERT_TRUE(base::ReadFileToString( |
| 119 caps.profile.AppendASCII("data"), &new_contents)); | 119 caps.profile.AppendASCII("data"), &new_contents)); |
| 120 EXPECT_STREQ("data", new_contents.c_str()); | 120 EXPECT_STREQ("data", new_contents.c_str()); |
| 121 } | 121 } |
| 122 | 122 |
| 123 TEST(CapabilitiesParser, UnknownCap) { | 123 TEST(CapabilitiesParser, UnknownCap) { |
| 124 Capabilities caps; | 124 Capabilities caps; |
| 125 DictionaryValue dict; | 125 DictionaryValue dict; |
| 126 dict.SetString("chromeOptions.nosuchcap", "none"); | 126 dict.SetString("chromeOptions.nosuchcap", "none"); |
| 127 CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); | 127 CapabilitiesParser parser(&dict, base::FilePath(), Logger(), &caps); |
| 128 ASSERT_TRUE(parser.Parse()); | 128 ASSERT_TRUE(parser.Parse()); |
| (...skipping 161 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 290 ASSERT_FALSE(parser.Parse()); | 290 ASSERT_FALSE(parser.Parse()); |
| 291 | 291 |
| 292 const std::set<std::string>& rm_set = caps.exclude_switches; | 292 const std::set<std::string>& rm_set = caps.exclude_switches; |
| 293 EXPECT_EQ(static_cast<size_t>(3), rm_set.size()); | 293 EXPECT_EQ(static_cast<size_t>(3), rm_set.size()); |
| 294 ASSERT_TRUE(rm_set.find(switches::kNoFirstRun) != rm_set.end()); | 294 ASSERT_TRUE(rm_set.find(switches::kNoFirstRun) != rm_set.end()); |
| 295 ASSERT_TRUE(rm_set.find(switches::kDisableSync) != rm_set.end()); | 295 ASSERT_TRUE(rm_set.find(switches::kDisableSync) != rm_set.end()); |
| 296 ASSERT_TRUE(rm_set.find(switches::kDisableTranslate) != rm_set.end()); | 296 ASSERT_TRUE(rm_set.find(switches::kDisableTranslate) != rm_set.end()); |
| 297 } | 297 } |
| 298 | 298 |
| 299 } // namespace webdriver | 299 } // namespace webdriver |
| OLD | NEW |