components/blocked_content/test/test_popup_navigation_delegate.h

// Copyright 2020 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 COMPONENTS_BLOCKED_CONTENT_TEST_TEST_POPUP_NAVIGATION_DELEGATE_H_
#define COMPONENTS_BLOCKED_CONTENT_TEST_TEST_POPUP_NAVIGATION_DELEGATE_H_

#include "components/blocked_content/popup_navigation_delegate.h"
#include "third_party/blink/public/mojom/window_features/window_features.mojom.h"
#include "url/gurl.h"

namespace blocked_content {

// Test delegate which stores results of calls in a ResultHolder.
class TestPopupNavigationDelegate : public PopupNavigationDelegate {
 public:
  // Struct used to hold results from calls on TestPopupNavigationDelegate since
  // the delegate will be destroyed on calls to ShowBlockedPopup().
  struct ResultHolder {
    ResultHolder();
    ~ResultHolder();

    bool did_navigate = false;
    blink::mojom::WindowFeatures navigation_window_features;
    base::Optional<WindowOpenDisposition> navigation_disposition;
    int total_popups_blocked_on_page = 0;
  };

  TestPopupNavigationDelegate(const GURL& url, ResultHolder* result_holder);

  // PopupNavigationDelegate:
  content::RenderFrameHost* GetOpener() override;
  bool GetOriginalUserGesture() override;
  const GURL& GetURL() override;
  NavigateResult NavigateWithGesture(
      const blink::mojom::WindowFeatures& window_features,
      base::Optional<WindowOpenDisposition> updated_disposition) override;
  void OnPopupBlocked(content::WebContents* web_contents,
                      int total_popups_blocked_on_page) override;

 private:
  const GURL url_;
  ResultHolder* result_holder_;
};

}  // namespace blocked_content

#endif  // COMPONENTS_BLOCKED_CONTENT_TEST_TEST_POPUP_NAVIGATION_DELEGATE_H_