Allow taking screen shot of a non child window which has delegate to paint.

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 6b94940334e4c08ced478b30a3ee137c00e72355..1db807236b70b8275d6c2d9f7dde0abcdcd605da 100644
--- a/ash/test/test_screenshot_delegate.h
+++ b/ash/test/test_screenshot_delegate.h
@@ -33,12 +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_; }
 
+  const aura::Window* GetSelectedWindowAndReset();
+
   void set_can_take_screenshot(bool can_take_screenshot) {
     can_take_screenshot_ = can_take_screenshot;
   }
@@ -46,9 +44,9 @@ 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_;
+  aura::Window* selected_window_;
 
   DISALLOW_COPY_AND_ASSIGN(TestScreenshotDelegate);
 };