Move newtab-tracker-stats tests to esbuild

duckduckgo · Jul 19, 2023 · c42ca7a · c42ca7a
1 parent 96e7d38
commit c42ca7a
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 8 deletions.
diff --git a/unit-test/legacy/newtab-tracker-stats.js → unit-test/background/newtab-tracker-stats.js b/unit-test/legacy/newtab-tracker-stats.js → unit-test/background/newtab-tracker-stats.js
@@ -1,6 +1,5 @@
 const { NewTabTrackerStats } = require('../../shared/js/background/newtab-tracker-stats')
 const { TrackerStats } = require('../../shared/js/background/classes/tracker-stats')
-const browserWrapper = require('../../shared/js/background/wrapper')
 const constants = require('../../shared/data/constants')
 const testTDS = require('../data/tds.json')
 const { dataFormatSchema } = require('../../shared/js/newtab/schema')
@@ -116,6 +115,11 @@ describe('NewTabTrackerStats', () => {
 describe('sending data', () => {
     beforeEach(() => {
         jasmine.clock().install()
+        // @ts-ignore
+        while (chrome.storage.local._setCalls.length > 0) {
+            // @ts-ignore
+            chrome.storage.local._setCalls.pop()
+        }
     })
     afterEach(() => {
         jasmine.clock().uninstall()
@@ -127,7 +131,8 @@ describe('sending data', () => {
         // @ts-ignore
         newtab.assignTopCompanies(testTDS.entities)
         const sendSpy = spyOn(newtab, '_publish')
-        const syncSpy = spyOn(browserWrapper, 'syncToStorage')
+        // @ts-ignore
+        const syncSpy = chrome.storage.local._setCalls
 
         const now = 1673473220560
 
@@ -145,8 +150,8 @@ describe('sending data', () => {
         expect(sendSpy).toHaveBeenCalledTimes(1)
 
         // assert that values were synced to storage
-        expect(syncSpy).toHaveBeenCalledTimes(1)
-        expect(syncSpy).toHaveBeenCalledWith({
+        expect(syncSpy.length).toBe(1)
+        expect(syncSpy[0]).toEqual({
             [NewTabTrackerStats.storageKey]: {
                 stats: {
                     current: {

diff --git a/unit-test/helpers/mock-browser-api.js b/unit-test/helpers/mock-browser-api.js
@@ -1,10 +1,16 @@
 globalThis.browser = {
     storage: {
         local: {
-            set: () => {},
+            set: (value) => {
+                browser.storage.local._setCalls.push(value)
+            },
             get: () => {
                 return {}
-            }
+            },
+            _setCalls: []
+        },
+        managed: {
+            get: () => {}
         }
     },
     browserAction: {

diff --git a/unit-test/inject-chrome-shim.js b/unit-test/inject-chrome-shim.js
@@ -1,10 +1,16 @@
 const chrome = {
     storage: {
         local: {
-            set: () => {},
+            set: (value) => {
+                chrome.storage.local._setCalls.push(value)
+            },
             get: () => {
                 return {}
-            }
+            },
+            _setCalls: []
+        },
+        managed: {
+            get: () => {}
         }
     },
     browserAction: {