blob: 74c57250f1e38ace4f76fdd86e564ce696d6513d [file] [log] [blame]
// Copyright 2013 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 ASH_TEST_TEST_SCREENSHOT_DELEGATE_H_
#define ASH_TEST_TEST_SCREENSHOT_DELEGATE_H_
#include "ash/screenshot_delegate.h"
#include "base/basictypes.h"
#include "base/compiler_specific.h"
#include "ui/gfx/rect.h"
namespace ash {
namespace test {
class TestScreenshotDelegate : public ScreenshotDelegate {
public:
TestScreenshotDelegate();
virtual ~TestScreenshotDelegate();
// Overridden from ScreenshotDelegate:
virtual void HandleTakeScreenshotForAllRootWindows() OVERRIDE;
virtual void HandleTakePartialScreenshot(
aura::Window* window, const gfx::Rect& rect) OVERRIDE;
virtual bool CanTakeScreenshot() OVERRIDE;
int handle_take_screenshot_count() const {
return handle_take_screenshot_count_;
}
int handle_take_partial_screenshot_count() const {
return handle_take_partial_screenshot_count_;
}
const gfx::Rect& last_rect() const { return last_rect_; }
void set_can_take_screenshot(bool can_take_screenshot) {
can_take_screenshot_ = can_take_screenshot;
}
private:
int handle_take_screenshot_count_;
int handle_take_partial_screenshot_count_;
gfx::Rect last_rect_;
bool can_take_screenshot_;
DISALLOW_COPY_AND_ASSIGN(TestScreenshotDelegate);
};
} // namespace test
} // namespace ash
#endif // ASH_TEST_TEST_SCREENSHOT_DELEGATE_H_