Send screenshots back to the client for debugging

chrome/test/webdriver/server.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 <signal.h>
 #include <stdlib.h>
 
 #if defined(OS_WIN)
 #include <windows.h>
 #endif
@@ skipping 19 matching lines @@
 #include "chrome/test/webdriver/error_codes.h"
 #include "chrome/test/webdriver/session_manager.h"
 #include "chrome/test/webdriver/utility_functions.h"
 #include "chrome/test/webdriver/commands/cookie_commands.h"
 #include "chrome/test/webdriver/commands/create_session.h"
 #include "chrome/test/webdriver/commands/execute_command.h"
 #include "chrome/test/webdriver/commands/find_element_commands.h"
 #include "chrome/test/webdriver/commands/implicit_wait_command.h"
 #include "chrome/test/webdriver/commands/navigate_commands.h"
 #include "chrome/test/webdriver/commands/mouse_commands.h"
+#include "chrome/test/webdriver/commands/screenshot_command.h"
 #include "chrome/test/webdriver/commands/session_with_id.h"
 #include "chrome/test/webdriver/commands/source_command.h"
 #include "chrome/test/webdriver/commands/speed_command.h"
 #include "chrome/test/webdriver/commands/target_locator_commands.h"
 #include "chrome/test/webdriver/commands/title_command.h"
 #include "chrome/test/webdriver/commands/url_command.h"
 #include "chrome/test/webdriver/commands/webelement_commands.h"
 #include "third_party/mongoose/mongoose.h"
 
 #if defined(OS_WIN)
@@ skipping 45 matching lines @@
   dispatcher->Add<ElementLocationInViewCommand>(
       "/session/*/element/*/location_in_view");
   dispatcher->Add<ElementNameCommand>(    "/session/*/element/*/name");
   dispatcher->Add<ElementSelectedCommand>("/session/*/element/*/selected");
   dispatcher->Add<ElementSizeCommand>(    "/session/*/element/*/size");
   dispatcher->Add<ElementSubmitCommand>(  "/session/*/element/*/submit");
   dispatcher->Add<ElementTextCommand>(    "/session/*/element/*/text");
   dispatcher->Add<ElementToggleCommand>(  "/session/*/element/*/toggle");
   dispatcher->Add<ElementValueCommand>(   "/session/*/element/*/value");
 
+  dispatcher->Add<ScreenshotCommand>("/session/*/screenshot");
+
   // Mouse Commands
   dispatcher->Add<ClickCommand>("/session/*/element/*/click");
   dispatcher->Add<DragCommand>( "/session/*/element/*/drag");
   dispatcher->Add<HoverCommand>("/session/*/element/*/hover");
 
   // All session based commands should be listed after the element based
   // commands to avoid potential mapping conflicts from an overzealous
   // wildcard match. For example, /session/*/title maps to the handler to
   // fetch the page title. If mapped first, this would overwrite the handler
   // for /session/*/element/*/attribute/title, which should fetch the title
@@ skipping 13 matching lines @@
   dispatcher->Add<ImplicitWaitCommand>( "/session/*/timeouts/implicit_wait");
 
   // Cookie functions.
   dispatcher->Add<CookieCommand>(     "/session/*/cookie");
   dispatcher->Add<NamedCookieCommand>("/session/*/cookie/*");
 
   // Commands that have not been implemented yet. We list these out explicitly
   // so that tests that attempt to use them fail with a meaningful error.
   dispatcher->SetNotImplemented("/session/*/execute_async");
   dispatcher->SetNotImplemented("/session/*/timeouts/async_script");
-  dispatcher->SetNotImplemented("/session/*/screenshot");
 
   // Since the /session/* is a wild card that would match the above URIs, this
   // line MUST be the last registered URI with the server.
   dispatcher->Add<SessionWithID>("/session/*");
 }
 
 }  // namespace webdriver
 
 // Configures mongoose according to the given command line flags.
 // Returns true on success.
 bool SetMongooseOptions(struct mg_context* ctx,
                         const std::string& port,
                         const std::string& root) {
   if (!mg_set_option(ctx, "ports", port.c_str())) {
     std::cout << "ChromeDriver cannot bind to port ("
               << port.c_str() << ")" << std::endl;
     return false;
   }
   if (root.length())
     mg_set_option(ctx, "root", root.c_str());
   // Lower the default idle time to 1 second. Idle time refers to how long a
   // worker thread will wait for new connections before exiting.
   // This is so mongoose quits in a reasonable amount of time.
   mg_set_option(ctx, "idle_time", "1");
   return true;
 }
 
+
 // Sets up and runs the Mongoose HTTP server for the JSON over HTTP
 // protcol of webdriver.  The spec is located at:
 // http://code.google.com/p/selenium/wiki/JsonWireProtocol.
 int main(int argc, char *argv[]) {
   struct mg_context *ctx;
   base::AtExitManager exit;
   base::WaitableEvent shutdown_event(false, false);
   CommandLine::Init(argc, argv);
   CommandLine* cmd_line = CommandLine::ForCurrentProcess();
 
@@ skipping 70 matching lines @@
 
   // Run until we receive command to shutdown.
   shutdown_event.Wait();
 
   // We should not reach here since the service should never quit.
   // TODO(jmikhail): register a listener for SIGTERM and break the
   // message loop gracefully.
   mg_stop(ctx);
   return (EXIT_SUCCESS);
 }