Mozart: Add helpers for building view associates.

mojo/ui/associates/mock_view_inspector.cc

+// 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.
+
+#include "mojo/ui/associates/mock_view_inspector.h"
+
+#include "base/bind.h"
+
+namespace mojo {
+namespace ui {
+
+MockViewInspector::MockViewInspector() {}
+
+MockViewInspector::~MockViewInspector() {
+  for (const auto& pair : hit_tester_callbacks_)
+    pair.second.Run(false);
+}
+
+void MockViewInspector::SetHitTester(
+    uint32_t view_tree_token_value,
+    mojo::gfx::composition::HitTester* hit_tester) {
+  DCHECK(view_tree_token_value);
+
+  hit_testers_.erase(hit_testers_.find(view_tree_token_value),
+                     hit_testers_.end());
+  if (hit_tester)
+    hit_testers_.emplace(view_tree_token_value, hit_tester);
+
+  auto pair = hit_tester_callbacks_.equal_range(view_tree_token_value);
+  for (auto it = pair.first; it != pair.second; ++it)
+    it->second.Run(hit_tester != nullptr);
+  hit_tester_callbacks_.erase(pair.first, pair.second);
+}
+
+void MockViewInspector::CloseHitTesterBindings() {
+  hit_tester_bindings_.CloseAllBindings();
+}
+
+void MockViewInspector::SetSceneMapping(uint32_t scene_token_value,
+                                        mojo::ui::ViewTokenPtr view_token) {
+  DCHECK(scene_token_value);
+
+  if (view_token)
+    scene_mappings_.emplace(scene_token_value, view_token.Pass());
+  else
+    scene_mappings_.erase(scene_mappings_.find(scene_token_value),
+                          scene_mappings_.end());
+}
+
+void MockViewInspector::GetHitTester(
+    mojo::ui::ViewTreeTokenPtr view_tree_token,
+    mojo::InterfaceRequest<mojo::gfx::composition::HitTester>
+        hit_tester_request,
+    const GetHitTesterCallback& callback) {
+  DCHECK(view_tree_token);
+  DCHECK(view_tree_token->value);
+  DCHECK(hit_tester_request.is_pending());
+
+  hit_tester_lookups_++;
+
+  auto it = hit_testers_.find(view_tree_token->value);
+  if (it == hit_testers_.end()) {
+    callback.Run(false);
+    return;
+  }
+
+  hit_tester_bindings_.AddBinding(it->second, hit_tester_request.Pass());
+  hit_tester_callbacks_.emplace(view_tree_token->value, callback);
+}
+
+void MockViewInspector::ResolveScenes(
+    mojo::Array<mojo::gfx::composition::SceneTokenPtr> scene_tokens,
+    const ResolveScenesCallback& callback) {
+  DCHECK(!scene_tokens.is_null());
+
+  scene_lookups_++;
+
+  mojo::Array<mojo::ui::ViewTokenPtr> view_tokens;
+  view_tokens.resize(scene_tokens.size());
+  for (size_t i = 0; i < scene_tokens.size(); i++) {
+    DCHECK(scene_tokens[i]);
+    DCHECK(scene_tokens[i]->value);
+    auto it = scene_mappings_.find(scene_tokens[i]->value);
+    if (it != scene_mappings_.end())
+      view_tokens[i] = it->second.Clone();
+  }
+
+  callback.Run(view_tokens.Pass());
+}
+
+}  // namespace ui
+}  // namespace mojo