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

#include <unordered_map>

#include "base/macros.h"
#include "mojo/common/binding_set.h"
#include "mojo/services/ui/views/interfaces/view_associates.mojom.h"

namespace mojo {
namespace ui {

class MockViewInspector : public ViewInspector {
 public:
  MockViewInspector();
  ~MockViewInspector() override;

  // Sets the hit tester to use for a particular view tree.
  // Setting to null removes the hit tester.
  // Does not take ownership; the hit tester must outlive this object.
  void SetHitTester(uint32_t view_tree_token_value,
                    mojo::gfx::composition::HitTester* hit_tester);

  // Closes all hit tester bindings without invoking the changed callbacks.
  void CloseHitTesterBindings();

  // Adds a mapping from scene token to view token.
  // Setting to null removes the scene mapping.
  void SetSceneMapping(uint32_t scene_token_value,
                       mojo::ui::ViewTokenPtr view_token);

  // Returns how often |GetHitTester| was called.
  uint32_t hit_tester_lookups() const { return hit_tester_lookups_; }

  // Returns how often |ResolveScenes| was called.
  uint32_t scene_lookups() const { return scene_lookups_; }

  // |ViewInspector|
  void GetHitTester(mojo::ui::ViewTreeTokenPtr view_tree_token,
                    mojo::InterfaceRequest<mojo::gfx::composition::HitTester>
                        hit_tester_request,
                    const GetHitTesterCallback& callback) override;
  void ResolveScenes(
      mojo::Array<mojo::gfx::composition::SceneTokenPtr> scene_tokens,
      const ResolveScenesCallback& callback) override;

 private:
  std::unordered_map<uint32_t, mojo::gfx::composition::HitTester*> hit_testers_;
  mojo::BindingSet<mojo::gfx::composition::HitTester> hit_tester_bindings_;
  std::unordered_multimap<uint32_t, GetHitTesterCallback> hit_tester_callbacks_;

  std::unordered_map<uint32_t, mojo::ui::ViewTokenPtr> scene_mappings_;

  uint32_t hit_tester_lookups_ = 0u;
  uint32_t scene_lookups_ = 0u;

  DISALLOW_COPY_AND_ASSIGN(MockViewInspector);
};

}  // namespace ui
}  // namespace mojo

#endif  // MOJO_UI_ASSOCIATES_MOCK_VIEW_INSPECTOR_H_