Send screenshots back to the client for debugging

chrome/test/webdriver/server.cc

Index: chrome/test/webdriver/server.cc
diff --git a/chrome/test/webdriver/server.cc b/chrome/test/webdriver/server.cc
index fede21277e4574accaffa710f07617fda122a74a..5fdc71bfe6c171749eab457fd7ad3ac692c639c8 100644
--- a/chrome/test/webdriver/server.cc
+++ b/chrome/test/webdriver/server.cc
@@ -194,8 +194,9 @@ int main(int argc, char *argv[]) {
     url_base = cmd_line->GetSwitchValueASCII("url-base");
 
   webdriver::SessionManager* manager = webdriver::SessionManager::GetInstance();
-  manager->set_port(port);
-  manager->set_url_base(url_base);
+  manager->SetPort(port);
+  manager->SetUrlBase(url_base);
+  manager->SetScreenshotOnError(cmd_line->HasSwitch("screenshot-on-error"));
   if (!chrome_dir.empty()) {
     if (!file_util::DirectoryExists(chrome_dir)) {
       std::cout << "Given Chrome directory is inaccessible or does not exist: "
@@ -206,7 +207,7 @@ int main(int argc, char *argv[]) {
       return ENOENT;
 #endif
     }
-    manager->set_chrome_dir(chrome_dir);
+    manager->SetChromeDir(chrome_dir);
   }
 
   // Initialize SHTTPD context.