// Copyright 2026 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_BROWSER_CONTEXTUAL_TASKS_MOCK_CONTEXTUAL_TASKS_PANEL_CONTROLLER_H_
#define CHROME_BROWSER_CONTEXTUAL_TASKS_MOCK_CONTEXTUAL_TASKS_PANEL_CONTROLLER_H_

#include <memory>
#include <optional>
#include <vector>

#include "base/uuid.h"
#include "chrome/browser/contextual_tasks/contextual_tasks_panel_controller.h"
#include "components/contextual_search/contextual_search_session_handle.h"
#include "components/contextual_tasks/public/contextual_task.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "third_party/omnibox_proto/chrome_aim_entry_point.pb.h"

namespace content {
class WebContents;
}

namespace contextual_tasks {

class MockContextualTasksPanelController
    : public ContextualTasksPanelController {
 public:
  MockContextualTasksPanelController();
  ~MockContextualTasksPanelController() override;

  MOCK_METHOD(void,
              Show,
              (bool,
               omnibox::ChromeAimEntryPoint,
               bool,
               std::optional<base::TimeTicks>),
              (override));
  MOCK_METHOD(void, Close, (), (override));
  MOCK_METHOD(void, OpenInZeroState, (), (override));
  MOCK_METHOD(bool, IsPanelOpenForContextualTask, (), (const, override));
  MOCK_METHOD(ContextualTasksPanelController::EntrySource,
              GetActiveEntrySource,
              (),
              (const, override));
  MOCK_METHOD(std::optional<tabs::TabHandle>,
              GetAutoSuggestedTabHandle,
              (),
              (override));
  MOCK_METHOD(void,
              OnTaskChanged,
              (content::WebContents*, base::Uuid),
              (override));
  MOCK_METHOD(void, OnAiInteraction, (), (override));
  MOCK_METHOD(void,
              SetPendingTaskForTab,
              (tabs::TabInterface*, const base::Uuid&),
              (override));
  MOCK_METHOD(content::WebContents*,
              GetActiveWebContents,
              (),
              (const override));
  MOCK_METHOD(content::WebContents*,
              GetToolbarWebContents,
              (),
              (const override));
  MOCK_METHOD(std::vector<content::WebContents*>,
              GetPanelWebContentsList,
              (),
              (const, override));
  MOCK_METHOD(std::unique_ptr<content::WebContents>,
              DetachWebContentsForTask,
              (const base::Uuid&),
              (override));
  MOCK_METHOD(contextual_search::ContextualSearchSessionHandle*,
              GetContextualSearchSessionHandleForPanel,
              (),
              (override));
  MOCK_METHOD(void,
              TransferWebContentsFromTab,
              (const base::Uuid&, std::unique_ptr<content::WebContents>),
              (override));
  MOCK_METHOD(std::optional<ContextualTask>, GetCurrentTask, (), (override));
  MOCK_METHOD((std::pair<std::optional<base::Uuid>,
                         contextual_search::ContextualSearchSessionHandle*>),
              GetSessionHandleForActiveTabOrPanel,
              (),
              (override));
  MOCK_METHOD(size_t, GetNumberOfActiveTasks, (), (const, override));
  MOCK_METHOD(void, MoveTaskUiToNewTab, (), (override));
  MOCK_METHOD(void, NotifyExpandToFullTabStateChanged, (), (override));
  MOCK_METHOD(bool, CanExpandToFullTab, (), (const, override));
  MOCK_METHOD(void,
              ShowPageInfoBubble,
              (bool is_pointer_interaction),
              (override));
  MOCK_METHOD(void, OnLogoPointerDown, (), (override));
  MOCK_METHOD(void, AddObserver, (Observer*), (override));
  MOCK_METHOD(void, RemoveObserver, (Observer*), (override));
};

}  // namespace contextual_tasks

#endif  // CHROME_BROWSER_CONTEXTUAL_TASKS_MOCK_CONTEXTUAL_TASKS_PANEL_CONTROLLER_H_