1 files changed, 186 insertions, 0 deletions
diff --git a/toolkit/devtools/performance/test/browser_profiler_tree-abstract-03.js b/toolkit/devtools/performance/test/browser_profiler_tree-abstract-03.js
new file mode 100644
index 000000000..89c5c4311
--- /dev/null
+++ b/toolkit/devtools/performance/test/browser_profiler_tree-abstract-03.js
@@ -0,0 +1,186 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+/**
+ * Tests if the abstract tree base class for the profiler's tree view
+ * is keyboard accessible.
+ */
+
+let { AbstractTreeItem } = Cu.import("resource:///modules/devtools/AbstractTreeItem.jsm", {});
+let { Heritage } = Cu.import("resource:///modules/devtools/ViewHelpers.jsm", {});
+
+function spawnTest () {
+  let container = document.createElement("vbox");
+  gBrowser.selectedBrowser.parentNode.appendChild(container);
+
+  // Populate the tree by pressing RIGHT...
+
+  let treeRoot = new MyCustomTreeItem(gDataSrc, { parent: null });
+  treeRoot.attachTo(container);
+  treeRoot.focus();
+
+  EventUtils.sendKey("RIGHT");
+  ok(treeRoot.expanded,
+    "The root node is now expanded.");
+  is(document.commandDispatcher.focusedElement, treeRoot.target,
+    "The root node is still focused.");
+
+  let fooItem = treeRoot.getChild(0);
+  let barItem = treeRoot.getChild(1);
+
+  EventUtils.sendKey("RIGHT");
+  ok(!fooItem.expanded,
+    "The 'foo' node is not expanded yet.");
+  is(document.commandDispatcher.focusedElement, fooItem.target,
+    "The 'foo' node is now focused.");
+
+  EventUtils.sendKey("RIGHT");
+  ok(fooItem.expanded,
+    "The 'foo' node is now expanded.");
+  is(document.commandDispatcher.focusedElement, fooItem.target,
+    "The 'foo' node is still focused.");
+
+  EventUtils.sendKey("RIGHT");
+  ok(!barItem.expanded,
+    "The 'bar' node is not expanded yet.");
+  is(document.commandDispatcher.focusedElement, barItem.target,
+    "The 'bar' node is now focused.");
+
+  EventUtils.sendKey("RIGHT");
+  ok(barItem.expanded,
+    "The 'bar' node is now expanded.");
+  is(document.commandDispatcher.focusedElement, barItem.target,
+    "The 'bar' node is still focused.");
+
+  let bazItem = barItem.getChild(0);
+
+  EventUtils.sendKey("RIGHT");
+  ok(!bazItem.expanded,
+    "The 'baz' node is not expanded yet.");
+  is(document.commandDispatcher.focusedElement, bazItem.target,
+    "The 'baz' node is now focused.");
+
+  EventUtils.sendKey("RIGHT");
+  ok(bazItem.expanded,
+    "The 'baz' node is now expanded.");
+  is(document.commandDispatcher.focusedElement, bazItem.target,
+    "The 'baz' node is still focused.");
+
+  // Test RIGHT on a leaf node.
+
+  EventUtils.sendKey("RIGHT");
+  is(document.commandDispatcher.focusedElement, bazItem.target,
+    "The 'baz' node is still focused.");
+
+  // Test DOWN on a leaf node.
+
+  EventUtils.sendKey("DOWN");
+  is(document.commandDispatcher.focusedElement, bazItem.target,
+    "The 'baz' node is now refocused.");
+
+  // Test UP.
+
+  EventUtils.sendKey("UP");
+  is(document.commandDispatcher.focusedElement, barItem.target,
+    "The 'bar' node is now refocused.");
+
+  EventUtils.sendKey("UP");
+  is(document.commandDispatcher.focusedElement, fooItem.target,
+    "The 'foo' node is now refocused.");
+
+  EventUtils.sendKey("UP");
+  is(document.commandDispatcher.focusedElement, treeRoot.target,
+    "The root node is now refocused.");
+
+  // Test DOWN.
+
+  EventUtils.sendKey("DOWN");
+  is(document.commandDispatcher.focusedElement, fooItem.target,
+    "The 'foo' node is now refocused.");
+
+  EventUtils.sendKey("DOWN");
+  is(document.commandDispatcher.focusedElement, barItem.target,
+    "The 'bar' node is now refocused.");
+
+  EventUtils.sendKey("DOWN");
+  is(document.commandDispatcher.focusedElement, bazItem.target,
+    "The 'baz' node is now refocused.");
+
+  // Test LEFT.
+
+  EventUtils.sendKey("LEFT");
+  ok(barItem.expanded,
+    "The 'bar' node is still expanded.");
+  is(document.commandDispatcher.focusedElement, barItem.target,
+    "The 'bar' node is now refocused.");
+
+  EventUtils.sendKey("LEFT");
+  ok(!barItem.expanded,
+    "The 'bar' node is not expanded anymore.");
+  is(document.commandDispatcher.focusedElement, barItem.target,
+    "The 'bar' node is still focused.");
+
+  EventUtils.sendKey("LEFT");
+  ok(treeRoot.expanded,
+    "The root node is still expanded.");
+  is(document.commandDispatcher.focusedElement, treeRoot.target,
+    "The root node is now refocused.");
+
+  EventUtils.sendKey("LEFT");
+  ok(!treeRoot.expanded,
+    "The root node is not expanded anymore.");
+  is(document.commandDispatcher.focusedElement, treeRoot.target,
+    "The root node is still focused.");
+
+  // Test LEFT on the root node.
+
+  EventUtils.sendKey("LEFT");
+  is(document.commandDispatcher.focusedElement, treeRoot.target,
+    "The root node is still focused.");
+
+  // Test UP on the root node.
+
+  EventUtils.sendKey("UP");
+  is(document.commandDispatcher.focusedElement, treeRoot.target,
+    "The root node is still focused.");
+
+  container.remove();
+  finish();
+}
+
+function MyCustomTreeItem(dataSrc, properties) {
+  AbstractTreeItem.call(this, properties);
+  this.itemDataSrc = dataSrc;
+}
+
+MyCustomTreeItem.prototype = Heritage.extend(AbstractTreeItem.prototype, {
+  _displaySelf: function(document, arrowNode) {
+    let node = document.createElement("hbox");
+    node.MozMarginStart = (this.level * 10) + "px";
+    node.appendChild(arrowNode);
+    node.appendChild(document.createTextNode(this.itemDataSrc.label));
+    return node;
+  },
+  _populateSelf: function(children) {
+    for (let childDataSrc of this.itemDataSrc.children) {
+      children.push(new MyCustomTreeItem(childDataSrc, {
+        parent: this,
+        level: this.level + 1
+      }));
+    }
+  }
+});
+
+let gDataSrc = {
+  label: "root",
+  children: [{
+    label: "foo",
+    children: []
+  }, {
+    label: "bar",
+    children: [{
+      label: "baz",
+      children: []
+    }]
+  }]
+};