ui/views/test/test_widget_observer.h

// Copyright 2012 The Chromium Authors
// 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_WIDGET_OBSERVER_H_
#define UI_VIEWS_TEST_TEST_WIDGET_OBSERVER_H_

#include <stddef.h>

#include "base/memory/raw_ptr.h"
#include "base/scoped_observation.h"
#include "ui/views/widget/widget_observer.h"

namespace views::test {

// A Widget observer class used in the tests below to observe bubbles closing.
class TestWidgetObserver : public WidgetObserver {
 public:
  explicit TestWidgetObserver(Widget* widget);

  TestWidgetObserver(const TestWidgetObserver&) = delete;
  TestWidgetObserver& operator=(const TestWidgetObserver&) = delete;

  ~TestWidgetObserver() override;

  bool widget_closed() const { return widget_ == nullptr; }

 private:
  // WidgetObserver overrides:
  void OnWidgetDestroying(Widget* widget) override;

  raw_ptr<Widget> widget_;
  base::ScopedObservation<Widget, TestWidgetObserver> widget_observation_{this};
};

}  // namespace views::test

#endif  // UI_VIEWS_TEST_TEST_WIDGET_OBSERVER_H_