summaryrefslogtreecommitdiffstats
path: root/ash/wm/partial_screenshot_view_unittest.cc
blob: 26e27e3ca8bc192d76b3d83963e22bc5b76e6b8a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
// Copyright (c) 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.

#include "ash/wm/partial_screenshot_view.h"

#include "ash/screenshot_delegate.h"
#include "ash/shell.h"
#include "ash/shell_window_ids.h"
#include "ash/test/ash_test_base.h"
#include "ash/test/test_overlay_delegate.h"
#include "ash/test/test_screenshot_delegate.h"
#include "ui/aura/window_event_dispatcher.h"
#include "ui/events/test/event_generator.h"
#include "ui/views/widget/widget.h"
#include "ui/views/widget/widget_observer.h"

namespace ash {

class PartialScreenshotViewTest : public test::AshTestBase,
                                  public views::WidgetObserver {
 public:
  PartialScreenshotViewTest() : view_(NULL) {}
  virtual ~PartialScreenshotViewTest() {
    if (view_)
      view_->GetWidget()->RemoveObserver(this);
  }

  void StartPartialScreenshot() {
    std::vector<PartialScreenshotView*> views =
        PartialScreenshotView::StartPartialScreenshot(GetScreenshotDelegate());
    if (!views.empty()) {
      view_ = views[0];
      view_->GetWidget()->AddObserver(this);
    }
  }

 protected:
  PartialScreenshotView* view_;

 private:
  // views::WidgetObserver:
  virtual void OnWidgetDestroying(views::Widget* widget) OVERRIDE {
    if (view_ && view_->GetWidget() == widget)
      view_ = NULL;
    widget->RemoveObserver(this);
  }

  DISALLOW_COPY_AND_ASSIGN(PartialScreenshotViewTest);
};

TEST_F(PartialScreenshotViewTest, BasicMouse) {
  StartPartialScreenshot();
  ASSERT_TRUE(view_);

  ui::test::EventGenerator generator(Shell::GetPrimaryRootWindow());

  generator.MoveMouseTo(100, 100);
  generator.PressLeftButton();
  EXPECT_FALSE(view_->is_dragging_);
  EXPECT_EQ("100,100", view_->start_position_.ToString());

  generator.MoveMouseTo(200, 200);
  EXPECT_TRUE(view_->is_dragging_);
  EXPECT_EQ("200,200", view_->current_position_.ToString());

  generator.ReleaseLeftButton();
  EXPECT_FALSE(view_->is_dragging_);
  EXPECT_EQ("100,100 100x100", GetScreenshotDelegate()->last_rect().ToString());
  EXPECT_EQ(1, GetScreenshotDelegate()->handle_take_partial_screenshot_count());
}

TEST_F(PartialScreenshotViewTest, BasicTouch) {
  StartPartialScreenshot();
  ASSERT_TRUE(view_);

  ui::test::EventGenerator generator(Shell::GetPrimaryRootWindow());

  generator.set_current_location(gfx::Point(100,100));
  generator.GestureTapDownAndUp(gfx::Point(100,100));
  EXPECT_FALSE(view_->is_dragging_);
  EXPECT_EQ(0, GetScreenshotDelegate()->handle_take_partial_screenshot_count());

  generator.PressTouch();
  EXPECT_FALSE(view_->is_dragging_);
  EXPECT_EQ("100,100", view_->start_position_.ToString());

  generator.MoveTouch(gfx::Point(200, 200));
  EXPECT_TRUE(view_->is_dragging_);
  EXPECT_EQ("200,200", view_->current_position_.ToString());

  generator.ReleaseTouch();
  EXPECT_FALSE(view_->is_dragging_);
  EXPECT_EQ(1, GetScreenshotDelegate()->handle_take_partial_screenshot_count());
  EXPECT_EQ("100,100 100x100", GetScreenshotDelegate()->last_rect().ToString());
}

// Partial screenshot session should not start when there is already another
// overlay. See: http://crbug.com/341958
TEST_F(PartialScreenshotViewTest, DontStartOverOverlay) {
  OverlayEventFilter* overlay_filter = Shell::GetInstance()->overlay_filter();
  test::TestOverlayDelegate delegate;
  overlay_filter->Activate(&delegate);
  EXPECT_EQ(&delegate, overlay_filter->delegate_);

  StartPartialScreenshot();
  EXPECT_TRUE(view_ == NULL);
  EXPECT_EQ(&delegate, overlay_filter->delegate_);
  overlay_filter->Deactivate(&delegate);

  StartPartialScreenshot();
  EXPECT_TRUE(view_ != NULL);

  // Someone else attempts to activate the overlay session, which should cancel
  // the current partial screenshot session.
  overlay_filter->Activate(&delegate);
  RunAllPendingInMessageLoop();
  EXPECT_EQ(&delegate, overlay_filter->delegate_);
  EXPECT_TRUE(view_ == NULL);
}

}  // namespace ash