Send screenshots back to the client for debugging

chrome/test/webdriver/commands/create_session.cc

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// 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/commands/create_session.h"
 
 #include <sstream>
 #include <string>
 
 #include "base/file_path.h"
 #include "base/values.h"
@@ skipping 18 matching lines @@
 
   // Session manages its own liftime, so do not call delete.
   Session* session = new Session();
   if (!session->Init(session_manager->chrome_dir())) {
     SET_WEBDRIVER_ERROR(response,
                         "Failed to initialize session",
                         kInternalServerError);
     return;
   }
 
+  bool screenshot_on_error = false;
+  DictionaryValue* capabilities = NULL;
+  if (GetDictionaryParameter("desiredCapabilities", &capabilities)) {
+    capabilities->GetBoolean("takeScreenshotOnError", &screenshot_on_error);
+    session->set_screenshot_on_error(screenshot_on_error);
+  }
+
   VLOG(1) << "Created session " << session->id();
   std::ostringstream stream;
   stream << "http://" << session_manager->GetAddress() << "/session/"
          << session->id();
   response->SetStatus(kSeeOther);
   response->SetValue(Value::CreateStringValue(stream.str()));
 }
 
 }  // namespace webdriver