| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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/command_line.h" | 5 #include "base/command_line.h" |
| 6 #include "base/file_path.h" | 6 #include "base/file_path.h" |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/path_service.h" | 8 #include "base/path_service.h" |
| 9 #include "base/string_util.h" | 9 #include "base/string_util.h" |
| 10 #include "chrome/common/chrome_paths.h" | 10 #include "chrome/common/chrome_paths.h" |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 42 browser_directory_ = dir; | 42 browser_directory_ = dir; |
| 43 } | 43 } |
| 44 UITest::SetUp(); | 44 UITest::SetUp(); |
| 45 | 45 |
| 46 // We've shortened the default timeouts, but dom_perf on Vista needs longer | 46 // We've shortened the default timeouts, but dom_perf on Vista needs longer |
| 47 // timeouts. Setting timeouts to the previous values before r52666. | 47 // timeouts. Setting timeouts to the previous values before r52666. |
| 48 set_action_timeout_ms(std::max(60000, action_timeout_ms())); | 48 set_action_timeout_ms(std::max(60000, action_timeout_ms())); |
| 49 } | 49 } |
| 50 | 50 |
| 51 void RunTest(const GURL& url, const char* wait_cookie_name, | 51 void RunTest(const GURL& url, const char* wait_cookie_name, |
| 52 const char* wait_cookie_value, const wchar_t* var_to_fetch, | 52 const char* wait_cookie_value, const char* var_to_fetch, |
| 53 UrlFetchTestResult* result) { | 53 UrlFetchTestResult* result) { |
| 54 scoped_refptr<TabProxy> tab(GetActiveTab()); | 54 scoped_refptr<TabProxy> tab(GetActiveTab()); |
| 55 ASSERT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS, tab->NavigateToURL(url)); | 55 ASSERT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS, tab->NavigateToURL(url)); |
| 56 | 56 |
| 57 if (wait_cookie_name) { | 57 if (wait_cookie_name) { |
| 58 if (wait_cookie_value) { | 58 if (wait_cookie_value) { |
| 59 bool completed = WaitUntilCookieValue(tab.get(), url, wait_cookie_name, | 59 bool completed = WaitUntilCookieValue(tab.get(), url, wait_cookie_name, |
| 60 UITest::test_timeout_ms(), | 60 UITest::test_timeout_ms(), |
| 61 wait_cookie_value); | 61 wait_cookie_value); |
| 62 ASSERT_TRUE(completed); | 62 ASSERT_TRUE(completed); |
| 63 } else { | 63 } else { |
| 64 result->cookie_value = WaitUntilCookieNonEmpty( | 64 result->cookie_value = WaitUntilCookieNonEmpty( |
| 65 tab.get(), url, wait_cookie_name, UITest::test_timeout_ms()); | 65 tab.get(), url, wait_cookie_name, UITest::test_timeout_ms()); |
| 66 ASSERT_TRUE(result->cookie_value.length()); | 66 ASSERT_TRUE(result->cookie_value.length()); |
| 67 } | 67 } |
| 68 } | 68 } |
| 69 if (var_to_fetch) { | 69 if (var_to_fetch) { |
| 70 std::wstring script = StringPrintf( | 70 std::string script = StringPrintf( |
| 71 L"window.domAutomationController.send(%ls);", var_to_fetch); | 71 "window.domAutomationController.send(%s);", var_to_fetch); |
| 72 | 72 |
| 73 std::wstring value; | 73 std::wstring value; |
| 74 bool success = tab->ExecuteAndExtractString(L"", script, &value); | 74 bool success = tab->ExecuteAndExtractString(L"", ASCIIToWide(script), |
| 75 &value); |
| 75 ASSERT_TRUE(success); | 76 ASSERT_TRUE(success); |
| 76 result->javascript_variable = WideToUTF8(value); | 77 result->javascript_variable = WideToUTF8(value); |
| 77 } | 78 } |
| 78 } | 79 } |
| 79 }; | 80 }; |
| 80 | 81 |
| 81 bool WriteValueToFile(std::string value, const FilePath& path) { | 82 bool WriteValueToFile(std::string value, const FilePath& path) { |
| 82 int retval = file_util::WriteFile(path, value.c_str(), value.length()); | 83 int retval = file_util::WriteFile(path, value.c_str(), value.length()); |
| 83 return retval == static_cast<int>(value.length()); | 84 return retval == static_cast<int>(value.length()); |
| 84 } | 85 } |
| (...skipping 29 matching lines...) Expand all Loading... |
| 114 const CommandLine* cmd_line = CommandLine::ForCurrentProcess(); | 115 const CommandLine* cmd_line = CommandLine::ForCurrentProcess(); |
| 115 | 116 |
| 116 if (!cmd_line->HasSwitch("url")) | 117 if (!cmd_line->HasSwitch("url")) |
| 117 return; | 118 return; |
| 118 | 119 |
| 119 std::string cookie_name = | 120 std::string cookie_name = |
| 120 cmd_line->GetSwitchValueASCII("wait_cookie_name"); | 121 cmd_line->GetSwitchValueASCII("wait_cookie_name"); |
| 121 std::string cookie_value = | 122 std::string cookie_value = |
| 122 cmd_line->GetSwitchValueASCII("wait_cookie_value"); | 123 cmd_line->GetSwitchValueASCII("wait_cookie_value"); |
| 123 | 124 |
| 124 std::wstring jsvar = cmd_line->GetSwitchValue("jsvar"); | 125 std::string jsvar = cmd_line->GetSwitchValueASCII("jsvar"); |
| 125 | 126 |
| 126 UrlFetchTestResult result; | 127 UrlFetchTestResult result; |
| 127 RunTest(GURL(WideToASCII(cmd_line->GetSwitchValue("url"))), | 128 RunTest(GURL(cmd_line->GetSwitchValueASCII("url")), |
| 128 cookie_name.length() > 0 ? cookie_name.c_str() : NULL, | 129 cookie_name.length() > 0 ? cookie_name.c_str() : NULL, |
| 129 cookie_value.length() > 0 ? cookie_value.c_str() : NULL, | 130 cookie_value.length() > 0 ? cookie_value.c_str() : NULL, |
| 130 jsvar.length() > 0 ? jsvar.c_str() : NULL, | 131 jsvar.length() > 0 ? jsvar.c_str() : NULL, |
| 131 &result); | 132 &result); |
| 132 | 133 |
| 133 // Write out the cookie if requested | 134 // Write out the cookie if requested |
| 134 FilePath cookie_output_path = | 135 FilePath cookie_output_path = |
| 135 cmd_line->GetSwitchValuePath("wait_cookie_output"); | 136 cmd_line->GetSwitchValuePath("wait_cookie_output"); |
| 136 if (cookie_output_path.value().size() > 0) { | 137 if (cookie_output_path.value().size() > 0) { |
| 137 ASSERT_TRUE(WriteValueToFile(result.cookie_value, cookie_output_path)); | 138 ASSERT_TRUE(WriteValueToFile(result.cookie_value, cookie_output_path)); |
| 138 } | 139 } |
| 139 | 140 |
| 140 // Write out the JS Variable if requested | 141 // Write out the JS Variable if requested |
| 141 FilePath jsvar_output_path = cmd_line->GetSwitchValuePath("jsvar_output"); | 142 FilePath jsvar_output_path = cmd_line->GetSwitchValuePath("jsvar_output"); |
| 142 if (jsvar_output_path.value().size() > 0) { | 143 if (jsvar_output_path.value().size() > 0) { |
| 143 ASSERT_TRUE(WriteValueToFile(result.javascript_variable, | 144 ASSERT_TRUE(WriteValueToFile(result.javascript_variable, |
| 144 jsvar_output_path)); | 145 jsvar_output_path)); |
| 145 } | 146 } |
| 146 } | 147 } |
| 147 | 148 |
| 148 } // namespace | 149 } // namespace |
| OLD | NEW |