Mozart: Add helpers for building view associates.

mojo/ui/associates/mock_view_inspector.h

+// 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_