Wires up mouse capture code for aura.

ui/aura/window_unittest.cc

 // Copyright (c) 2011 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 "base/basictypes.h"
 #include "base/compiler_specific.h"
 #include "base/message_loop.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/aura/desktop.h"
 #include "ui/aura/event.h"
@@ skipping 24 matching lines @@
                                const gfx::Rect& new_bounds) OVERRIDE {}
   virtual void OnFocus() OVERRIDE {}
   virtual void OnBlur() OVERRIDE {}
   virtual bool OnKeyEvent(KeyEvent* event) OVERRIDE {
     return false;
   }
   virtual int GetNonClientComponent(const gfx::Point& point) const OVERRIDE {
     return HTCLIENT;
   }
   virtual bool OnMouseEvent(MouseEvent* event) OVERRIDE { return false; }
+  virtual void OnCaptureGained() OVERRIDE {}
+  virtual void OnCaptureLost() OVERRIDE {}
   virtual void OnPaint(gfx::Canvas* canvas) OVERRIDE {}
   virtual void OnWindowDestroying() OVERRIDE {}
   virtual void OnWindowDestroyed() OVERRIDE {}
 
  private:
   DISALLOW_COPY_AND_ASSIGN(WindowDelegateImpl);
 };
 
 // Used for verifying destruction methods are invoked.
 class DestroyTrackingDelegateImpl : public WindowDelegateImpl {
@@ skipping 44 matching lines @@
     EXPECT_TRUE(parent_delegate_->in_destroying());
     DestroyTrackingDelegateImpl::OnWindowDestroying();
   }
 
  private:
   DestroyTrackingDelegateImpl* parent_delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(ChildWindowDelegateImpl);
 };
 
+// Used in verifying mouse capture.
+class CaptureWindowDelegateImpl : public WindowDelegateImpl {
+ public:
+  explicit CaptureWindowDelegateImpl()
+      : capture_gained_count_(0),
+        capture_lost_count_(0),
+        mouse_event_count_(0) {
+  }
+
+  int capture_gained_count() const { return capture_gained_count_; }
+  void set_capture_gained_count(int value) { capture_gained_count_ = value; }
+  int capture_lost_count() const { return capture_lost_count_; }
+  void set_capture_lost_count(int value) { capture_lost_count_ = value; }
+  int mouse_event_count() const { return mouse_event_count_; }
+  void set_mouse_event_count(int value) { mouse_event_count_ = value; }
+
+  virtual bool OnMouseEvent(MouseEvent* event) OVERRIDE {
+    mouse_event_count_++;
+    return false;
+  }
+  virtual void OnCaptureGained() OVERRIDE {
+    capture_gained_count_++;
+  }
+  virtual void OnCaptureLost() OVERRIDE {
+    capture_lost_count_++;
+  }
+
+ private:
+  int capture_gained_count_;
+  int capture_lost_count_;
+  int mouse_event_count_;
+
+  DISALLOW_COPY_AND_ASSIGN(CaptureWindowDelegateImpl);
+};
+
 // A simple WindowDelegate implementation for these tests. It owns itself
 // (deletes itself when the Window it is attached to is destroyed).
 class TestWindowDelegate : public WindowDelegateImpl {
  public:
   TestWindowDelegate(SkColor color)
       : color_(color),
         last_key_code_(ui::VKEY_UNKNOWN) {
   }
   virtual ~TestWindowDelegate() {}
 
@@ skipping 179 matching lines @@
 
   parent.MoveChildToFront(&child1);
   ASSERT_EQ(2u, parent.children().size());
   EXPECT_EQ(&child1, parent.children()[1]);
   EXPECT_EQ(&child2, parent.children()[0]);
   ASSERT_EQ(2u, parent.layer()->children().size());
   EXPECT_EQ(child1.layer(), parent.layer()->children()[1]);
   EXPECT_EQ(child2.layer(), parent.layer()->children()[0]);
 }
 
+// Various destruction assertions.
+TEST_F(WindowTest, CaptureTests) {
+  Desktop* desktop = Desktop::GetInstance();
+  CaptureWindowDelegateImpl delegate;
+  scoped_ptr<Window> window(CreateTestWindowWithDelegate(
+      &delegate, 0, gfx::Rect(0, 0, 20, 20), NULL));
+  EXPECT_FALSE(window->HasCapture());
+
+  // Do a capture.
+  window->SetCapture();
+  EXPECT_TRUE(window->HasCapture());
+  EXPECT_EQ(1, delegate.capture_gained_count());
+  EXPECT_EQ(0, delegate.capture_lost_count());
+
+  desktop->OnMouseEvent(MouseEvent(ui::ET_MOUSE_PRESSED, gfx::Point(50, 50),
+                                   ui::EF_LEFT_BUTTON_DOWN));
+  EXPECT_EQ(1, delegate.mouse_event_count());
+  desktop->OnMouseEvent(MouseEvent(ui::ET_MOUSE_RELEASED, gfx::Point(50, 50),
+                                   ui::EF_LEFT_BUTTON_DOWN));
+  EXPECT_EQ(2, delegate.mouse_event_count());
+  delegate.set_mouse_event_count(0);
+
+  window->ReleaseCapture();
+  EXPECT_FALSE(window->HasCapture());
+  EXPECT_EQ(1, delegate.capture_gained_count());
+  EXPECT_EQ(1, delegate.capture_lost_count());
+
+  desktop->OnMouseEvent(MouseEvent(ui::ET_MOUSE_PRESSED, gfx::Point(50, 50),
+                                   ui::EF_LEFT_BUTTON_DOWN));
+  EXPECT_EQ(0, delegate.mouse_event_count());
+}
+
+// Various destruction assertions.
+TEST_F(WindowTest, ReleaseCaptureOnDestroy) {
+  Desktop* desktop = Desktop::GetInstance();
+  RootWindow* root = static_cast<RootWindow*>(desktop->window());
+  CaptureWindowDelegateImpl delegate;
+  scoped_ptr<Window> window(CreateTestWindowWithDelegate(
+      &delegate, 0, gfx::Rect(0, 0, 20, 20), NULL));
+  EXPECT_FALSE(window->HasCapture());
+
+  // Do a capture.
+  window->SetCapture();
+  EXPECT_TRUE(window->HasCapture());
+
+  // Destroy the window.
+  window.reset();
+
+  // Make sure the root doesn't reference the window anymore.
+  EXPECT_EQ(NULL, root->mouse_pressed_handler());
+  EXPECT_EQ(NULL, root->capture_window());
+}
+
 }  // namespace internal
 }  // namespace aura
-