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 |