1 files changed, 151 insertions, 0 deletions
diff --git a/comm/calendar/test/unit/test_data_bags.js b/comm/calendar/test/unit/test_data_bags.js
new file mode 100644
index 0000000000..dd1f2a6abd
--- /dev/null
+++ b/comm/calendar/test/unit/test_data_bags.js
@@ -0,0 +1,151 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+function run_test() {
+  test_listener_set();
+  test_observer_set();
+  test_operation_group();
+}
+
+function test_listener_set() {
+  let set = new cal.data.ListenerSet(Ci.calIOperationListener);
+  let listener1Id = null;
+  let listener2Id = null;
+
+  let listener1 = cal.createAdapter("calIOperationListener", {
+    onOperationComplete(aCalendar, aStatus, aOpType, aId, aDetail) {
+      listener1Id = aId;
+    },
+  });
+  let listener2 = cal.createAdapter("calIOperationListener", {
+    onOperationComplete(aCalendar, aStatus, aOpType, aId, aDetail) {
+      listener2Id = aId;
+    },
+  });
+
+  set.add(listener1);
+  set.add(listener2);
+  set.notify("onOperationComplete", [null, null, null, "test", null]);
+  equal(listener1Id, "test");
+  equal(listener2Id, "test");
+
+  set.delete(listener2);
+  listener1Id = listener2Id = null;
+  set.notify("onOperationComplete", [null, null, null, "test2", null]);
+  equal(listener1Id, "test2");
+  strictEqual(listener2Id, null);
+
+  // Re-adding the listener may lead to an endless loop if the notify
+  // function uses a live list of observers.
+  let called = 0;
+  let listener3 = cal.createAdapter("calIOperationListener", {
+    onOperationComplete(aCalendar, aStatus, aOpType, aId, aDetail) {
+      set.delete(listener3);
+      if (called == 0) {
+        set.add(listener3);
+      }
+      called++;
+    },
+  });
+
+  set.add(listener3);
+  set.notify("onOperationComplete", [null, null, null, "test3", null]);
+  equal(called, 1);
+}
+
+function test_observer_set() {
+  let set = new cal.data.ObserverSet(Ci.calIObserver);
+  let listenerCountBegin1 = 0;
+  let listenerCountBegin2 = 0;
+  let listenerCountEnd1 = 0;
+  let listenerCountEnd2 = 0;
+
+  let listener1 = cal.createAdapter("calIObserver", {
+    onStartBatch() {
+      listenerCountBegin1++;
+    },
+    onEndBatch() {
+      listenerCountEnd1++;
+    },
+  });
+  let listener2 = cal.createAdapter("calIObserver", {
+    onStartBatch() {
+      listenerCountBegin2++;
+    },
+    onEndBatch() {
+      listenerCountEnd2++;
+    },
+  });
+
+  set.add(listener1);
+  equal(listenerCountBegin1, 0);
+  equal(listenerCountEnd1, 0);
+  equal(set.batchCount, 0);
+
+  set.notify("onStartBatch");
+  equal(listenerCountBegin1, 1);
+  equal(listenerCountEnd1, 0);
+  equal(set.batchCount, 1);
+
+  set.add(listener2);
+  equal(listenerCountBegin1, 1);
+  equal(listenerCountEnd1, 0);
+  equal(listenerCountBegin2, 1);
+  equal(listenerCountEnd2, 0);
+  equal(set.batchCount, 1);
+
+  set.add(listener1);
+  equal(listenerCountBegin1, 1);
+  equal(listenerCountEnd1, 0);
+  equal(listenerCountBegin2, 1);
+  equal(listenerCountEnd2, 0);
+  equal(set.batchCount, 1);
+
+  set.notify("onEndBatch");
+  equal(listenerCountBegin1, 1);
+  equal(listenerCountEnd1, 1);
+  equal(listenerCountBegin2, 1);
+  equal(listenerCountEnd2, 1);
+  equal(set.batchCount, 0);
+}
+
+function test_operation_group() {
+  let calledCancel = false;
+  let calledOperationCancel = null;
+  let group = new cal.data.OperationGroup();
+  ok(group.id.endsWith("-0"));
+  ok(group.isPending);
+  equal(group.status, Cr.NS_OK);
+  ok(group.isEmpty);
+
+  let operation = {
+    id: 123,
+    isPending: true,
+    cancel: status => {
+      calledOperationCancel = status;
+    },
+  };
+
+  group.add(operation);
+  ok(!group.isEmpty);
+
+  group.notifyCompleted(Cr.NS_ERROR_FAILURE);
+  ok(!group.isPending);
+  equal(group.status, Cr.NS_ERROR_FAILURE);
+  strictEqual(calledOperationCancel, null);
+
+  group.remove(operation);
+  ok(group.isEmpty);
+
+  group = new cal.data.OperationGroup(() => {
+    calledCancel = true;
+  });
+  ok(group.id.endsWith("-1"));
+  group.add(operation);
+
+  group.cancel();
+  equal(group.status, Ci.calIErrors.OPERATION_CANCELLED);
+  equal(calledOperationCancel, Ci.calIErrors.OPERATION_CANCELLED);
+  ok(calledCancel);
+}