chrome/test/base/dialog_test_browser_window.h

// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_TEST_BASE_DIALOG_TEST_BROWSER_WINDOW_H_
#define CHROME_TEST_BASE_DIALOG_TEST_BROWSER_WINDOW_H_

#include <memory>

#include "chrome/test/base/test_browser_window.h"
#include "components/web_modal/web_contents_modal_dialog_host.h"

class BrowserWindowInterface;

namespace content {
class WebContents;
}

namespace views {
class Widget;
}

// Custom test browser window to provide a parent view to a modal dialog.
class DialogTestBrowserWindow : public TestBrowserWindow,
                                public web_modal::WebContentsModalDialogHost {
 public:
  DialogTestBrowserWindow();
  DialogTestBrowserWindow(const DialogTestBrowserWindow&) = delete;
  DialogTestBrowserWindow& operator=(const DialogTestBrowserWindow&) = delete;
  ~DialogTestBrowserWindow() override;

  // BrowserWindow overrides
  web_modal::WebContentsModalDialogHost* GetWebContentsModalDialogHost()
      override;

  web_modal::WebContentsModalDialogHost* GetWebContentsModalDialogHostFor(
      content::WebContents* web_contents) override;

  // web_modal::WebContentsModalDialogHost overrides
  gfx::NativeView GetHostView() const override;
  gfx::Point GetDialogPosition(const gfx::Size& size) override;
  gfx::Size GetMaximumDialogSize() override;
  void AddObserver(web_modal::ModalDialogHostObserver* observer) override;
  void RemoveObserver(web_modal::ModalDialogHostObserver* observer) override;

 private:
  BrowserWindowInterface* FindBrowser() const;

  // Dummy window for parenting dialogs.
  std::unique_ptr<views::Widget> host_window_;
};

#endif  // CHROME_TEST_BASE_DIALOG_TEST_BROWSER_WINDOW_H_