feat: add tabs.onRemoved and tabs.onCreated mocks

clarkbw · Nov 23, 2023 · 6d79ad5 · 6d79ad5
1 parent c0b1bbf
commit 6d79ad5
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/__tests__/tabs.test.js b/__tests__/tabs.test.js
@@ -102,6 +102,22 @@ describe('browser.tabs', () => {
       expect(browser.tabs.onUpdated[method]).toHaveBeenCalledTimes(1);
       expect(callback).toHaveBeenCalledTimes(0);
     });
+
+    test(`onRemoved.${method}`, () => {
+      const callback = jest.fn();
+      expect(jest.isMockFunction(browser.tabs.onRemoved[method])).toBe(true);
+      browser.tabs.onRemoved[method](callback);
+      expect(browser.tabs.onRemoved[method]).toHaveBeenCalledTimes(1);
+      expect(callback).toHaveBeenCalledTimes(0);
+    });
+
+    test(`onCreated.${method}`, () => {
+      const callback = jest.fn();
+      expect(jest.isMockFunction(browser.tabs.onCreated[method])).toBe(true);
+      browser.tabs.onCreated[method](callback);
+      expect(browser.tabs.onCreated[method]).toHaveBeenCalledTimes(1);
+      expect(callback).toHaveBeenCalledTimes(0);
+    });
   });
   test('reload', (done) => {
     const callback = jest.fn(() => done());

diff --git a/dist/setup.js b/dist/setup.js
@@ -226,6 +226,16 @@ var tabs = {
     removeListener: jest.fn(),
     hasListener: jest.fn()
   },
+  onRemoved: {
+    addListener: jest.fn(),
+    removeListener: jest.fn(),
+    hasListener: jest.fn()
+  },
+  onCreated: {
+    addListener: jest.fn(),
+    removeListener: jest.fn(),
+    hasListener: jest.fn()
+  },
   sendMessage: jest.fn(function (tabId, message, cb) {
     onMessageListeners.forEach(function (listener) {
       return listener(tabId, message);

diff --git a/src/tabs.js b/src/tabs.js
@@ -45,6 +45,16 @@ export const tabs = {
     removeListener: jest.fn(),
     hasListener: jest.fn(),
   },
+  onRemoved: {
+    addListener: jest.fn(),
+    removeListener: jest.fn(),
+    hasListener: jest.fn(),
+  },
+  onCreated: {
+    addListener: jest.fn(),
+    removeListener: jest.fn(),
+    hasListener: jest.fn(),
+  },
   sendMessage: jest.fn((tabId, message, cb) => {
     onMessageListeners.forEach((listener) => listener(tabId, message));
     if (cb !== undefined) {