Adding upstream version 86.0.1.upstream/86.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 27 insertions, 0 deletions

diff --git a/toolkit/components/normandy/test/browser/browser_CleanupManager.js b/toolkit/components/normandy/test/browser/browser_CleanupManager.js
new file mode 100644
index 0000000000..9ea525ba5d
--- /dev/null
+++ b/toolkit/components/normandy/test/browser/browser_CleanupManager.js
@@ -0,0 +1,27 @@
+"use strict";
+
+ChromeUtils.import(
+  "resource://normandy/lib/CleanupManager.jsm",
+  this
+); /* global CleanupManagerClass */
+
+add_task(async function testCleanupManager() {
+  const spy1 = sinon.spy();
+  const spy2 = sinon.spy();
+  const spy3 = sinon.spy();
+
+  const manager = new CleanupManager.constructor();
+  manager.addCleanupHandler(spy1);
+  manager.addCleanupHandler(spy2);
+  manager.addCleanupHandler(spy3);
+  manager.removeCleanupHandler(spy2); // Test removal
+
+  await manager.cleanup();
+  ok(spy1.called, "cleanup called the spy1 handler");
+  ok(!spy2.called, "cleanup did not call the spy2 handler");
+  ok(spy3.called, "cleanup called the spy3 handler");
+
+  await manager.cleanup();
+  ok(spy1.calledOnce, "cleanup only called the spy1 handler once");
+  ok(spy3.calledOnce, "cleanup only called the spy3 handler once");
+});