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

#include "components/viz/service/input/mock_input_manager.h"

namespace viz {

MockInputManager::MockInputManager(FrameSinkManagerImpl* frame_sink_manager)
    : InputManager(frame_sink_manager) {}

MockInputManager::~MockInputManager() = default;

bool MockInputManager::RIRExistsForFrameSinkId(
    const FrameSinkId& frame_sink_id) {
  return rir_map_.contains(frame_sink_id);
}

RenderInputRouterSupportBase* MockInputManager::GetSupportForFrameSink(
    const FrameSinkId& id) {
  return frame_sink_metadata_map_.find(id)->second.rir_support.get();
}

input::RenderInputRouter* MockInputManager::GetRenderInputRouterForId(
    const FrameSinkId& id) {
  auto itr = rir_map_.find(id);
  if (itr == rir_map_.end()) {
    return nullptr;
  }
  return itr->second.get();
}

}  // namespace viz