Send screenshots back to the client for debugging

chrome/test/webdriver/commands/response.h

 // 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.
 
 #ifndef CHROME_TEST_WEBDRIVER_COMMANDS_RESPONSE_H_
 #define CHROME_TEST_WEBDRIVER_COMMANDS_RESPONSE_H_
 
 #include <sstream>
 #include <string>
 
 #include "base/values.h"
 #include "chrome/test/webdriver/error_codes.h"
 
 namespace webdriver {
 
 // All errors in webdriver must use this macro in order to send back
 // a proper stack trace to the client
 #define SET_WEBDRIVER_ERROR(response, msg, err) \
   response->SetError(err, msg, __FILE__, __LINE__); \
   LOG(ERROR) << msg
 
+// Use this macro for errors in which a screenshot can be taken and
+// sent back to the client.  A proper stack trace will also be included.
+#define SET_WEBDRIVER_SCREENSHOT_ERROR(response, session, msg, err) { \
+  LOG(ERROR) << msg; \
+  std::string __screenshot__; \
+  session->ScreenshotAsBase64(&__screenshot__); \
+  response->SetError(err, msg, __FILE__, __LINE__); \
+  response->SetScreenshot(__screenshot__); \
+}
+
 // A simple class that encapsulates the information describing the response to
 // a |Command|. In Webdriver all responses must be sent back as a JSON value,
 // conforming to the spec found at:
 // http://code.google.com/p/selenium/wiki/JsonWireProtocol#Messages
 class Response {
  public:
   // Creates a new |Response| with a default status of |kSuccess| and a
   // |NullValue|.
   Response();
   ~Response();
@@ skipping 13 matching lines @@
   // set using the |SET_WEBDRIVER_ERROR| macro above.
   void SetError(ErrorCode error, const std::string& message,
                 const std::string& file, int line);
 
   // Sets a JSON field in this response. The |key| may be a "." delimitted
   // string to indicate the value should be set in a nested object. Any
   // previously set value for the |key| will be deleted.
   // This object assumes ownership of |value|.
   void SetField(const std::string& key, Value* value);
 
+  // Sets the screenshot to return to the client for debugging.  The file must
+  // be a base64 encoded PNG file.
+  void SetScreenshot(const std::string& screenshot);
+
   // Returns this response as a JSON string.
   std::string ToJSON() const;
 
  private:
   DictionaryValue data_;
+  std::string screenshot_;
 
   DISALLOW_COPY_AND_ASSIGN(Response);
 };
 
 }  // namespace webdriver
 
 #endif  // CHROME_TEST_WEBDRIVER_COMMANDS_RESPONSE_H_