Send screenshots back to the client for debugging

chrome/test/webdriver/automation.cc

 // Copyright (c) 2011 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/webdriver/automation.h"
 
 #if defined(OS_WIN)
 #include <windows.h>
 #endif
 
 #include "base/base_paths.h"
 #include "base/command_line.h"
 #include "base/environment.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/path_service.h"
 #include "base/utf_string_conversions.h"
+#include "base/scoped_temp_dir.h"

[kkania, 2011/03/16 18:19:10]

i don't think you need this here anymore

[Joe, 2011/03/17 00:15:25]

Done.

 #include "base/values.h"
 #include "chrome/common/automation_constants.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/automation/automation_json_requests.h"
 #include "chrome/test/automation/automation_proxy.h"
 #include "chrome/test/automation/proxy_launcher.h"
 #include "chrome/test/webdriver/frame_path.h"
 #include "googleurl/src/gurl.h"
@@ skipping 195 matching lines @@
   int windex = 0, tab_index = 0;
   if (!GetIndicesForTab(tab_id, &windex, &tab_index)) {
     *success = false;
     return;
   }
 
   *success = SendWebKeyEventJSONRequest(
       automation(), windex, tab_index, key_event);
 }
 
+void Automation::GetScreenShot(int tab_id,
+                               const FilePath& path,
+                               bool* success) {
+  int windex = 0, tab_index = 0;
+
+  if (!GetIndicesForTab(tab_id, &windex, &tab_index)) {
+    *success = false;
+    return;
+  }
+
+  *success = SendGetScreenShotJSONRequest(
+      automation(), windex, tab_index, path);
+}
+
 void Automation::NavigateToURL(int tab_id,
                                const std::string& url,
                                bool* success) {
   int browser_index = 0, tab_index = 0;
   if (!GetIndicesForTab(tab_id, &browser_index, &tab_index)) {
     *success = false;
     return;
   }
 
   AutomationMsg_NavigationResponseValues navigate_response;
@@ skipping 152 matching lines @@
     int tab_id, int* browser_index, int* tab_index) {
   if (!SendGetIndicesFromTabIdJSONRequest(automation(), tab_id,
                                           browser_index, tab_index)) {
     LOG(ERROR) << "Could not get browser and tab indices for WebDriver tab id";
     return false;
   }
   return true;
 }
 
 }  // namespace webdriver