Add AXTreeDelegate and refactor other AXTree classes slightly.

ui/accessibility/ax_tree_unittest.cc

 // Copyright 2013 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 "base/memory/scoped_ptr.h"
 #include "base/strings/string_number_conversions.h"
+#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/accessibility/ax_node.h"
 #include "ui/accessibility/ax_serializable_tree.h"
 #include "ui/accessibility/ax_tree.h"
 #include "ui/accessibility/ax_tree_serializer.h"
 
 namespace ui {
 
+namespace {
+
+class MockAXTreeDelegate : public AXTreeDelegate {
+ public:
+  MOCK_METHOD1(OnNodeWillBeDeleted, void(AXNode*));

[tfarina, 2014/01/13 22:44:38]

yeah, something like:

class AXTreeDelegateImpl : public AXTreeDelegate {
 public:
  virtual void OnNodeWillBeDeleted(AXNode*) {
    deleted_ = true;
  }

  ...

  bool deleted() const { return deleted_; }

 private:
  bool deleted_;  # Or you could you use an int for a counter, deleted_count_
};

+  MOCK_METHOD1(OnNodeCreated, void(AXNode*));
+  MOCK_METHOD1(OnNodeChanged, void(AXNode*));
+  MOCK_METHOD1(OnRootChanged, void(AXNode*));
+};
+
+MATCHER_P(AXNodeWithId, id, "") { return arg->id() == id; }
+
+}  // anonymous namespace
+
 TEST(AXTreeTest, SerializeSimpleAXTree) {
   AXNodeData root;
   root.id = 1;
   root.role = AX_ROLE_ROOT_WEB_AREA;
   root.state = (1 << AX_STATE_FOCUSABLE) | (1 << AX_STATE_FOCUSED);
   root.location = gfx::Rect(0, 0, 800, 600);
   root.child_ids.push_back(2);
   root.child_ids.push_back(3);
 
   AXNodeData button;
   button.id = 2;
   button.role = AX_ROLE_BUTTON;
   button.state = 0;
   button.location = gfx::Rect(20, 20, 200, 30);
 
   AXNodeData checkbox;
   checkbox.id = 3;
   checkbox.role = AX_ROLE_CHECK_BOX;
   checkbox.state = 0;
   checkbox.location = gfx::Rect(20, 50, 200, 30);
 
   AXTreeUpdate initial_state;
   initial_state.nodes.push_back(root);
   initial_state.nodes.push_back(button);
   initial_state.nodes.push_back(checkbox);
   AXSerializableTree src_tree(initial_state);
 
-  scoped_ptr<AXTreeSource<AXNode> > tree_source(
+  scoped_ptr<AXTreeSource<const AXNode*> > tree_source(
       src_tree.CreateTreeSource());
-  AXTreeSerializer<AXNode> serializer(tree_source.get());
+  AXTreeSerializer<const AXNode*> serializer(tree_source.get());
   AXTreeUpdate update;
   serializer.SerializeChanges(src_tree.GetRoot(), &update);
 
   AXTree dst_tree;
   ASSERT_TRUE(dst_tree.Unserialize(update));
 
-  AXNode* root_node = dst_tree.GetRoot();
+  const AXNode* root_node = dst_tree.GetRoot();
   ASSERT_TRUE(root_node != NULL);
   EXPECT_EQ(root.id, root_node->id());
   EXPECT_EQ(root.role, root_node->data().role);
 
   ASSERT_EQ(2, root_node->child_count());
 
-  AXNode* button_node = root_node->ChildAtIndex(0);
+  const AXNode* button_node = root_node->ChildAtIndex(0);
   EXPECT_EQ(button.id, button_node->id());
   EXPECT_EQ(button.role, button_node->data().role);
 
-  AXNode* checkbox_node = root_node->ChildAtIndex(1);
+  const AXNode* checkbox_node = root_node->ChildAtIndex(1);
   EXPECT_EQ(checkbox.id, checkbox_node->id());
   EXPECT_EQ(checkbox.role, checkbox_node->data().role);
 
   EXPECT_EQ(
       "id=1 ROOT_WEB_AREA FOCUSABLE FOCUSED (0, 0)-(800, 600) child_ids=2,3\n"
       "  id=2 BUTTON (20, 20)-(200, 30)\n"
       "  id=3 CHECKBOX (20, 50)-(200, 30)\n",
       dst_tree.ToString());
 }
 
@@ skipping 94 matching lines @@
   update.nodes[0].id = 1;
   update.nodes[0].role = AX_ROLE_ROOT_WEB_AREA;
   update.nodes[0].child_ids.push_back(3);
   update.nodes[0].child_ids.push_back(2);
   update.nodes[1].id = 2;
   update.nodes[2].id = 3;
   EXPECT_FALSE(tree.Unserialize(update));
   ASSERT_EQ("Node 3 reparented from 2 to 1", tree.error());
 }
 
+TEST(AXTreeTest, TreeDelegateIsCalled) {
+  AXTreeUpdate initial_state;
+  initial_state.nodes.resize(1);
+  initial_state.nodes[0].id = 1;
+  initial_state.nodes[0].role = AX_ROLE_ROOT_WEB_AREA;
+
+  AXTree tree(initial_state);
+  AXTreeUpdate update;
+  update.node_id_to_clear = 1;
+  update.nodes.resize(2);
+  update.nodes[0].id = 2;
+  update.nodes[0].role = AX_ROLE_ROOT_WEB_AREA;
+  update.nodes[0].child_ids.push_back(3);
+  update.nodes[1].id = 3;
+
+  testing::StrictMock<MockAXTreeDelegate> mock_delegate;
+  testing::InSequence s;
+  EXPECT_CALL(mock_delegate, OnNodeWillBeDeleted(AXNodeWithId(1)));
+  EXPECT_CALL(mock_delegate, OnNodeCreated(AXNodeWithId(2)));
+  EXPECT_CALL(mock_delegate, OnNodeCreated(AXNodeWithId(3)));
+  EXPECT_CALL(mock_delegate, OnRootChanged(AXNodeWithId(2)));
+  tree.SetDelegate(&mock_delegate);
+
+  EXPECT_TRUE(tree.Unserialize(update));
+  tree.SetDelegate(NULL);
+}
+
 }  // namespace ui