ui/views/test/test_platform_native_widget.h

// 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_