| // Copyright 2016 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 UI_VIEWS_TEST_TEST_PLATFORM_NATIVE_WIDGET_H_ |
| #define UI_VIEWS_TEST_TEST_PLATFORM_NATIVE_WIDGET_H_ |
| |
| #include "base/compiler_specific.h" |
| #include "base/macros.h" |
| #include "ui/views/view.h" |
| |
| namespace views { |
| namespace internal { |
| class NativeWidgetDelegate; |
| } |
| namespace test { |
| |
| // NativeWidget implementation that adds the following: |
| // . capture can be mocked. |
| // . a boolean is set when the NativeWidget is destroyed. |
| // Don't create directly, instead go through functions in |
| // native_widget_factory.h that create the appropriate platform specific |
| // implementation. |
| template <typename PlatformNativeWidget> |
| class TestPlatformNativeWidget : public PlatformNativeWidget { |
| public: |
| TestPlatformNativeWidget(internal::NativeWidgetDelegate* delegate, |
| bool mock_capture, |
| bool* destroyed) |
| : PlatformNativeWidget(delegate), |
| mouse_capture_(false), |
| mock_capture_(mock_capture), |
| destroyed_(destroyed) {} |
| |
| ~TestPlatformNativeWidget() override { |
| if (destroyed_) |
| *destroyed_ = true; |
| } |
| |
| // PlatformNativeWidget: |
| void SetCapture() override { |
| if (mock_capture_) |
| mouse_capture_ = true; |
| else |
| PlatformNativeWidget::SetCapture(); |
| } |
| |
| void ReleaseCapture() override { |
| if (mock_capture_) { |
| if (mouse_capture_) |
| PlatformNativeWidget::GetWidget()->OnMouseCaptureLost(); |
| mouse_capture_ = false; |
| } else { |
| PlatformNativeWidget::ReleaseCapture(); |
| } |
| } |
| |
| bool HasCapture() const override { |
| return mock_capture_ ? mouse_capture_ : PlatformNativeWidget::HasCapture(); |
| } |
| |
| private: |
| bool mouse_capture_; |
| const bool mock_capture_; |
| bool* destroyed_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TestPlatformNativeWidget); |
| }; |
| |
| } // namespace test |
| } // namespace views |
| |
| #endif // UI_VIEWS_TEST_TEST_PLATFORM_NATIVE_WIDGET_H_ |