Take screenshot of active window

ash/test/test_screenshot_delegate.h

Index: ash/test/test_screenshot_delegate.h
diff --git a/ash/test/test_screenshot_delegate.h b/ash/test/test_screenshot_delegate.h
index 2ada0ab3c06c62713f66d417a0ca1088b42a91bd..6b94940334e4c08ced478b30a3ee137c00e72355 100644
--- a/ash/test/test_screenshot_delegate.h
+++ b/ash/test/test_screenshot_delegate.h
@@ -22,6 +22,7 @@ class TestScreenshotDelegate : public ScreenshotDelegate {
   void HandleTakeScreenshotForAllRootWindows() override;
   void HandleTakePartialScreenshot(aura::Window* window,
                                    const gfx::Rect& rect) override;
+  void HandleTakeWindowScreenshot(aura::Window* window) override;
   bool CanTakeScreenshot() override;
 
   int handle_take_screenshot_count() const {
@@ -32,6 +33,10 @@ class TestScreenshotDelegate : public ScreenshotDelegate {
     return handle_take_partial_screenshot_count_;
   }
 
+  int handle_take_window_screenshot_count() const {
+    return handle_take_window_screenshot_count_;
+  }
+
   const gfx::Rect& last_rect() const { return last_rect_; }
 
   void set_can_take_screenshot(bool can_take_screenshot) {
@@ -41,6 +46,7 @@ class TestScreenshotDelegate : public ScreenshotDelegate {
  private:
   int handle_take_screenshot_count_;
   int handle_take_partial_screenshot_count_;
+  int handle_take_window_screenshot_count_;
   gfx::Rect last_rect_;
   bool can_take_screenshot_;