// Copyright 2017 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_MEDIA_ROUTER_TEST_MOCK_MOJO_MEDIA_ROUTER_H_
#define CHROME_BROWSER_MEDIA_ROUTER_TEST_MOCK_MOJO_MEDIA_ROUTER_H_

#include "components/media_router/browser/logger_impl.h"
#include "components/media_router/browser/test/mock_media_router.h"
#include "components/media_router/common/media_route_provider_helper.h"
#include "components/media_router/common/mojom/media_router.mojom.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace media_router {

class MockMojoMediaRouter : public MockMediaRouter, public mojom::MediaRouter {
 public:
  MockMojoMediaRouter();
  ~MockMojoMediaRouter() override;

  // mojom::MediaRouter overrides:
  void RegisterMediaRouteProvider(
      mojom::MediaRouteProviderId provider_id,
      mojo::PendingRemote<mojom::MediaRouteProvider> provider_remote) override {
    RegisterMediaRouteProviderInternal(provider_id, provider_remote);
  }
  MOCK_METHOD2(
      RegisterMediaRouteProviderInternal,
      void(mojom::MediaRouteProviderId provider_id,
           mojo::PendingRemote<mojom::MediaRouteProvider>& provider_remote));
  MOCK_METHOD1(OnIssue, void(const IssueInfo& issue));
  MOCK_METHOD1(ClearTopIssueForSink, void(const std::string& sink_id));
  MOCK_METHOD4(OnSinksReceived,
               void(mojom::MediaRouteProviderId provider_id,
                    const std::string& media_source,
                    const std::vector<MediaSinkInternal>& internal_sinks,
                    const std::vector<url::Origin>& origins));
  MOCK_METHOD2(OnRoutesUpdated,
               void(mojom::MediaRouteProviderId provider_id,
                    const std::vector<MediaRoute>& routes));
  MOCK_METHOD2(OnPresentationConnectionStateChanged,
               void(const std::string& route_id,
                    blink::mojom::PresentationConnectionState state));
  MOCK_METHOD3(OnPresentationConnectionClosed,
               void(const std::string& route_id,
                    blink::mojom::PresentationConnectionCloseReason reason,
                    const std::string& message));
  MOCK_METHOD2(OnRouteMessagesReceived,
               void(const std::string& route_id,
                    std::vector<mojom::RouteMessagePtr> messages));
  MOCK_METHOD1(GetLogger, void(mojo::PendingReceiver<mojom::Logger> receiver));
  MOCK_METHOD0(GetLogger, LoggerImpl*());
  MOCK_METHOD1(GetLogsAsString, void(GetLogsAsStringCallback callback));
  void GetMediaSinkServiceStatus(
      mojom::MediaRouter::GetMediaSinkServiceStatusCallback callback) override {
    GetMediaSinkServiceStatusInternal(callback);
  }
  MOCK_METHOD1(
      GetMediaSinkServiceStatusInternal,
      void(mojom::MediaRouter::GetMediaSinkServiceStatusCallback& callback));
  MOCK_METHOD0(GetMediaSinkServiceStatus, std::string());
  MOCK_METHOD(void,
              GetDebugger,
              (mojo::PendingReceiver<mojom::Debugger> receiver),
              (override));
  MOCK_METHOD(MediaRouterDebugger&, GetDebugger, (), (override));
};

}  // namespace media_router

#endif  // CHROME_BROWSER_MEDIA_ROUTER_TEST_MOCK_MOJO_MEDIA_ROUTER_H_