// Copyright 2016 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_UI_TOOLBAR_CAST_MOCK_CAST_TOOLBAR_BUTTON_CONTROLLER_H_
#define CHROME_BROWSER_UI_TOOLBAR_CAST_MOCK_CAST_TOOLBAR_BUTTON_CONTROLLER_H_

#include "chrome/browser/ui/toolbar/cast/cast_toolbar_button_controller.h"
#include "testing/gmock/include/gmock/gmock.h"

class MockCastToolbarButtonController : public CastToolbarButtonController {
 public:
  explicit MockCastToolbarButtonController(Profile* profile);

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

  ~MockCastToolbarButtonController() override;

  MOCK_METHOD(void, OnIssueUpdated, (const media_router::Issue* issue));
  MOCK_METHOD(void,
              OnRoutesUpdated,
              (const std::vector<media_router::MediaRoute>& routes));
  MOCK_METHOD(void, OnDialogShown, ());
  MOCK_METHOD(void, OnDialogHidden, ());
  MOCK_METHOD(void, OnContextMenuShown, ());
  MOCK_METHOD(void, OnContextMenuHidden, ());
  MOCK_METHOD(void, MaybeAddOrRemoveAction, ());
};

#endif  // CHROME_BROWSER_UI_TOOLBAR_CAST_MOCK_CAST_TOOLBAR_BUTTON_CONTROLLER_H_