1 files changed, 107 insertions, 0 deletions
diff --git a/browser/components/places/tests/unit/test_PUIU_batchUpdatesForNode.js b/browser/components/places/tests/unit/test_PUIU_batchUpdatesForNode.js
new file mode 100644
index 0000000000..58b68d7574
--- /dev/null
+++ b/browser/components/places/tests/unit/test_PUIU_batchUpdatesForNode.js
@@ -0,0 +1,107 @@
+const { sinon } = ChromeUtils.importESModule(
+  "resource://testing-common/Sinon.sys.mjs"
+);
+
+/* eslint-disable mozilla/use-chromeutils-generateqi */
+
+add_task(async function test_no_result_node() {
+  let functionSpy = sinon.stub().returns(Promise.resolve());
+
+  await PlacesUIUtils.batchUpdatesForNode(null, 1, functionSpy);
+
+  Assert.ok(
+    functionSpy.calledOnce,
+    "Passing a null result node should still call the wrapped function"
+  );
+});
+
+add_task(async function test_under_batch_threshold() {
+  let functionSpy = sinon.stub().returns(Promise.resolve());
+  let resultNode = {
+    QueryInterface() {
+      return this;
+    },
+    onBeginUpdateBatch: sinon.spy(),
+    onEndUpdateBatch: sinon.spy(),
+  };
+
+  await PlacesUIUtils.batchUpdatesForNode(resultNode, 1, functionSpy);
+
+  Assert.ok(functionSpy.calledOnce, "Wrapped function should be called once");
+  Assert.ok(
+    resultNode.onBeginUpdateBatch.notCalled,
+    "onBeginUpdateBatch should not have been called"
+  );
+  Assert.ok(
+    resultNode.onEndUpdateBatch.notCalled,
+    "onEndUpdateBatch should not have been called"
+  );
+});
+
+add_task(async function test_over_batch_threshold() {
+  let functionSpy = sinon.stub().callsFake(() => {
+    Assert.ok(
+      resultNode.onBeginUpdateBatch.calledOnce,
+      "onBeginUpdateBatch should have been called before the function"
+    );
+    Assert.ok(
+      resultNode.onEndUpdateBatch.notCalled,
+      "onEndUpdateBatch should not have been called before the function"
+    );
+
+    return Promise.resolve();
+  });
+  let resultNode = {
+    QueryInterface() {
+      return this;
+    },
+    onBeginUpdateBatch: sinon.spy(),
+    onEndUpdateBatch: sinon.spy(),
+  };
+
+  await PlacesUIUtils.batchUpdatesForNode(resultNode, 100, functionSpy);
+
+  Assert.ok(functionSpy.calledOnce, "Wrapped function should be called once");
+  Assert.ok(
+    resultNode.onBeginUpdateBatch.calledOnce,
+    "onBeginUpdateBatch should have been called"
+  );
+  Assert.ok(
+    resultNode.onEndUpdateBatch.calledOnce,
+    "onEndUpdateBatch should have been called"
+  );
+});
+
+add_task(async function test_wrapped_function_throws() {
+  let error = new Error("Failed!");
+  let functionSpy = sinon.stub().throws(error);
+  let resultNode = {
+    QueryInterface() {
+      return this;
+    },
+    onBeginUpdateBatch: sinon.spy(),
+    onEndUpdateBatch: sinon.spy(),
+  };
+
+  let raisedError;
+  try {
+    await PlacesUIUtils.batchUpdatesForNode(resultNode, 100, functionSpy);
+  } catch (ex) {
+    raisedError = ex;
+  }
+
+  Assert.ok(functionSpy.calledOnce, "Wrapped function should be called once");
+  Assert.ok(
+    resultNode.onBeginUpdateBatch.calledOnce,
+    "onBeginUpdateBatch should have been called"
+  );
+  Assert.ok(
+    resultNode.onEndUpdateBatch.calledOnce,
+    "onEndUpdateBatch should have been called"
+  );
+  Assert.equal(
+    raisedError,
+    error,
+    "batchUpdatesForNode should have raised the error from the wrapped function"
+  );
+});