1 files changed, 123 insertions, 0 deletions
diff --git a/remote/shared/webdriver/test/xpcshell/test_NodeCache.js b/remote/shared/webdriver/test/xpcshell/test_NodeCache.js
new file mode 100644
index 0000000000..8111cd0bc7
--- /dev/null
+++ b/remote/shared/webdriver/test/xpcshell/test_NodeCache.js
@@ -0,0 +1,123 @@
+const { NodeCache } = ChromeUtils.importESModule(
+  "chrome://remote/content/shared/webdriver/NodeCache.sys.mjs"
+);
+
+const nodeCache = new NodeCache();
+
+const SVG_NS = "http://www.w3.org/2000/svg";
+
+const browser = Services.appShell.createWindowlessBrowser(false);
+
+const domEl = browser.document.createElement("div");
+browser.document.body.appendChild(domEl);
+
+const svgEl = browser.document.createElementNS(SVG_NS, "rect");
+browser.document.body.appendChild(svgEl);
+
+registerCleanupFunction(() => {
+  nodeCache.clear({ all: true });
+});
+
+add_test(function addElement() {
+  const domElRef = nodeCache.add(domEl);
+  equal(nodeCache.size, 1);
+
+  const domElRefOther = nodeCache.add(domEl);
+  equal(nodeCache.size, 1);
+  equal(domElRefOther, domElRef);
+
+  nodeCache.add(svgEl);
+  equal(nodeCache.size, 2);
+
+  run_next_test();
+});
+
+add_test(function addInvalidElement() {
+  Assert.throws(() => nodeCache.add("foo"), /UnknownError/);
+
+  run_next_test();
+});
+
+add_test(function clear() {
+  nodeCache.add(domEl);
+  nodeCache.add(svgEl);
+  equal(nodeCache.size, 2);
+
+  // Clear requires explicit arguments.
+  Assert.throws(() => nodeCache.clear(), /Error/);
+
+  // Clear references for a different browsing context
+  const browser2 = Services.appShell.createWindowlessBrowser(false);
+  let imgEl = browser2.document.createElement("img");
+  browser2.document.body.appendChild(imgEl);
+
+  nodeCache.add(imgEl);
+  nodeCache.clear({ browsingContext: browser.browsingContext });
+  equal(nodeCache.size, 1);
+
+  // Clear all references
+  nodeCache.add(domEl);
+  equal(nodeCache.size, 2);
+
+  nodeCache.clear({ all: true });
+  equal(nodeCache.size, 0);
+
+  run_next_test();
+});
+
+add_test(function resolveElement() {
+  const domElSharedId = nodeCache.add(domEl);
+  deepEqual(nodeCache.resolve(domElSharedId), domEl);
+
+  const svgElSharedId = nodeCache.add(svgEl);
+  deepEqual(nodeCache.resolve(svgElSharedId), svgEl);
+  deepEqual(nodeCache.resolve(domElSharedId), domEl);
+
+  run_next_test();
+});
+
+add_test(function resolveUnknownElement() {
+  Assert.throws(() => nodeCache.resolve("foo"), /NoSuchElementError/);
+
+  run_next_test();
+});
+
+add_test(function resolveElementNotAttachedToDOM() {
+  const imgEl = browser.document.createElement("img");
+
+  const imgElSharedId = nodeCache.add(imgEl);
+  deepEqual(nodeCache.resolve(imgElSharedId), imgEl);
+
+  run_next_test();
+});
+
+add_test(async function resolveElementRemoved() {
+  let imgEl = browser.document.createElement("img");
+  const imgElSharedId = nodeCache.add(imgEl);
+
+  // Delete element and force a garbage collection
+  imgEl = null;
+
+  await doGC();
+
+  const el = nodeCache.resolve(imgElSharedId);
+  deepEqual(el, null);
+
+  run_next_test();
+});
+
+add_test(function elementReferencesDifferentPerNodeCache() {
+  const sharedId = nodeCache.add(domEl);
+
+  const nodeCache2 = new NodeCache();
+  const sharedId2 = nodeCache2.add(domEl);
+
+  notEqual(sharedId, sharedId2);
+  equal(nodeCache.resolve(sharedId), nodeCache2.resolve(sharedId2));
+
+  Assert.throws(() => nodeCache.resolve(sharedId2), /NoSuchElementError/);
+
+  nodeCache2.clear({ all: true });
+
+  run_next_test();
+});