Send screenshots back to the client for debugging

chrome/test/webdriver/server.cc

 // Copyright (c) 2010 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 62 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");
 
   // 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.
@@ skipping 97 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);
 }