Send screenshots back to the client for debugging

chrome/test/webdriver/session_manager.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 "base/logging.h"
 #include "chrome/test/webdriver/utility_functions.h"
 #include "net/base/net_util.h"
 
@@ skipping 14 matching lines @@
   }
 #endif
   if (hostname.empty()) {
     hostname = "localhost";
   }
   return hostname + ":" + port_ + url_base_;
 }
 
 void SessionManager::Add(Session* session) {
   base::AutoLock lock(map_lock_);
-  map_[session->id()] = session;
+  map_[session->Id()] = session;
 }
 
 bool SessionManager::Has(const std::string& id) const {
   base::AutoLock lock(map_lock_);
   return map_.find(id) != map_.end();
 }
 
 bool SessionManager::Remove(const std::string& id) {
   std::map<std::string, Session*>::iterator it;
   Session* session;
@@ skipping 12 matching lines @@
   std::map<std::string, Session*>::const_iterator it;
   base::AutoLock lock(map_lock_);
   it = map_.find(id);
   if (it == map_.end()) {
     VLOG(1) << "No such session with ID " << id;
     return NULL;
   }
   return it->second;
 }
 
-void SessionManager::set_port(const std::string& port) {
+void SessionManager::SetPort(const std::string& port) {
   port_ = port;
 }
 
-void SessionManager::set_url_base(const std::string& url_base) {
+void SessionManager::SetChromeDir(const FilePath& chrome_dir) {
+  chrome_dir_ = chrome_dir;
+}
+
+FilePath SessionManager::ChromeDir() const {
+  return chrome_dir_;
+}
+
+void SessionManager::SetScreenshotOnError(bool screenshot) {
+  screenshot_on_error_ = screenshot;
+}
+
+bool SessionManager::ScreenshotOnError() const {
+  return screenshot_on_error_;
+}
+
+void SessionManager::SetUrlBase(const std::string& url_base) {
   url_base_ = url_base;
 }
 
-std::string SessionManager::url_base() const {
+std::string SessionManager::UrlBase() const {
   return url_base_;
 }
 
-void SessionManager::set_chrome_dir(const FilePath& chrome_dir) {
-  chrome_dir_ = chrome_dir;
-}
-
-FilePath SessionManager::chrome_dir() const {
-  return chrome_dir_;
-}
-
-SessionManager::SessionManager() : port_(""), url_base_("") {}
+SessionManager::SessionManager()
+    : port_(""), url_base_(""), screenshot_on_error_(false) {}
 
 SessionManager::~SessionManager() {}
 
 // static
 SessionManager* SessionManager::GetInstance() {
   return Singleton<SessionManager>::get();
 }
 
 }  // namespace webdriver