Send screenshots back to the client for debugging

chrome/test/webdriver/session.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/session_manager.h"
 
 #include <sstream>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/message_loop_proxy.h"
 #include "base/process.h"
 #include "base/process_util.h"
 #include "base/scoped_ptr.h"
+#include "base/scoped_temp_dir.h"
 #include "base/stringprintf.h"
 #include "base/string_number_conversions.h"
 #include "base/string_split.h"
 #include "base/string_util.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/platform_thread.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
@@ skipping 22 matching lines @@
 FrameId& FrameId::operator=(const FrameId& other) {
   window_id = other.window_id;
   frame_path = other.frame_path;
   return *this;
 }
 
 Session::Session()
     : id_(GenerateRandomID()),
       thread_(id_.c_str()),
       implicit_wait_(0),
+      screenshot_on_error_(false),
       current_target_(FrameId(0, FramePath())) {
   SessionManager::GetInstance()->Add(this);
 }
 
 Session::~Session() {
   SessionManager::GetInstance()->Remove(id_);
 }
 
 bool Session::Init(const FilePath& browser_dir) {
   bool success = false;
@@ skipping 789 matching lines @@
   int x = 0, y = 0;
   if (!loc_dict->GetInteger("x", &x) ||
       !loc_dict->GetInteger("y", &y)) {
     LOG(ERROR) << "Location atom returned bad coordinate dictionary";
     code = kUnknownError;
   }
   *location = gfx::Point(x, y);
   return kSuccess;
 }
 
+bool Session::GetScreenShot(std::string* png) {
+  bool success = false;
+  ScopedTempDir screenshots_dir;
+
+  // Create a temp directory for screenshots.
+  if (!screenshots_dir.CreateUniqueTempDir()) {
+    return false;
+  }
+
+  FilePath path = screenshots_dir.path().AppendASCII("screen");
+
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::CaptureEntirePageAsPNG,
+      current_target_.window_id,
+      path,
+      &success));
+
+  if (success) {
+    success = file_util::ReadFileToString(path, png);
+  }
+  return success;
+}
+
+void Session::set_screenshot_on_error(bool error) {
+  screenshot_on_error_ = error;
+}
+
+bool Session::get_screenshot_on_error() const {
+  return screenshot_on_error_;
+}
+
+const std::string& Session::id() const {
+  return id_;
+}
+
+int Session::implicit_wait() const {
+  return implicit_wait_;
+}
+
+void Session::set_implicit_wait(const int& timeout) {
+  implicit_wait_ = timeout > 0 ? timeout : 0;
+}
+
+Session::Speed Session::speed() {
+  return speed_;
+}
+
+void Session::set_speed(Speed speed) {
+  speed_ = speed;
+}
+
 }  // namespace webdriver