summaryrefslogtreecommitdiffstats
path: root/toolkit/components/normandy/test/browser/browser_CleanupManager.js
blob: f1b49303945ec82ac5e92d11e3ef0d446580f283 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
"use strict";

const { CleanupManager } = ChromeUtils.importESModule(
  "resource://normandy/lib/CleanupManager.sys.mjs"
); /* 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");
});