update unit test

Author: guMcrey

tests/index.mock.ts

@@ -19,6 +19,9 @@ export const mockWorker = () => {
     postMessage(msg: string) {
       this.onmessage(msg)
     }
+    terminate() {
+      console.log('stop')
+    }
   }
 
   global.window = Object.create(window)
@@ -44,3 +47,12 @@ export const mockFetch = () => {
     })
   }
 }
+
+export const mockDateToLocaleDateString = () => {
+  global.window = Object.create(window)
+  Object.defineProperty(window, 'Date()', {
+    value: {
+      toLocaleDateString: () => '2023/1/18',
+    },
+  })
+}

tests/index.test.ts

@@ -2,12 +2,17 @@
  * @jest-environment jsdom
  */
 
-import {createWorker, createWorkerFunc} from '../utils/index'
+import {
+  createWorker,
+  createWorkerFunc,
+  cancelUpdateFunc,
+} from '../utils/index'
 import {
   mockCreateObjectUrl,
   mockWorker,
   mockSetInterval,
   mockFetch,
+  mockDateToLocaleDateString,
 } from './index.mock'
 
 test('create web worker', () => {
@@ -24,10 +29,71 @@ test('create worker function', () => {
     data: {
       'version-key': '1.1.0',
       'polling-time': 0,
+      immediate: true,
       'origin-version-file-url': 'https://www.example.com',
     },
   })
   temp.postMessage = (obj: {refreshPageVersion: string}) => {
     expect(typeof obj === 'object' && obj.refreshPageVersion).toBeTruthy()
   }
 })
+
+test('cancel update function', () => {
+  mockWorker()
+  const worker = new Worker('https://test.com')
+
+  // ignore-current-version -> true
+  localStorage.setItem('version-rocket:cancelled', '1.0.0')
+  expect(
+    cancelUpdateFunc('ignore-current-version', '1.0.0', true, worker),
+  ).toBeTruthy()
+  expect(
+    cancelUpdateFunc('ignore-current-version', '1.0.0', true, undefined),
+  ).toBeTruthy()
+
+  // ignore-current-version -> false
+  localStorage.setItem('version-rocket:cancelled', '1.1.0')
+  expect(
+    cancelUpdateFunc('ignore-current-version', '1.0.0', false, worker),
+  ).toBeFalsy()
+
+  // ignore-today -> true
+  mockDateToLocaleDateString()
+  localStorage.setItem('version-rocket:cancelled', '2023/1/18')
+  expect(cancelUpdateFunc('ignore-today', '1.0.0', true, worker)).toBeTruthy()
+  expect(
+    cancelUpdateFunc('ignore-today', '1.0.0', true, undefined),
+  ).toBeTruthy()
+
+  // ignore-today -> false
+  localStorage.setItem('version-rocket:cancelled', '2023/1/19')
+  expect(cancelUpdateFunc('ignore-today', '1.0.0', false, worker)).toBeFalsy()
+
+  // ignore-current-version -> true
+  sessionStorage.setItem('version-rocket:cancelled', 'true')
+  expect(
+    cancelUpdateFunc('ignore-current-window', '1.0.0', true, worker),
+  ).toBeTruthy()
+  expect(
+    cancelUpdateFunc('ignore-current-window', '1.0.0', true, undefined),
+  ).toBeTruthy()
+
+  // ignore-current-version -> false
+  sessionStorage.setItem('version-rocket:cancelled', '')
+  expect(
+    cancelUpdateFunc('ignore-current-window', '1.0.0', false, worker),
+  ).toBeFalsy()
+
+  // default
+  expect(cancelUpdateFunc('ignore-test', '1.0.0', false, worker)).toBeFalsy()
+
+  // no args
+  localStorage.setItem('version-rocket:cancelled', '1.1.0')
+  expect(cancelUpdateFunc(undefined, '', undefined, undefined)).toBeFalsy()
+
+  localStorage.setItem('version-rocket:cancelled', '')
+  expect(cancelUpdateFunc('ignore-today', '', undefined, undefined)).toBeFalsy()
+
+  sessionStorage.setItem('version-rocket:cancelled', '')
+  expect(cancelUpdateFunc('ignore-today', '', undefined, undefined)).toBeFalsy()
+})

utils/index.js

@@ -40,7 +40,7 @@ export const createWorkerFunc = () => {
 export const cancelUpdateFunc = (cancelMode, newVersion, cancelUpdateAndStopWorker, worker) => {
     const cancelModeType = cancelMode || 'ignore-current-version';
     const cancelModeTypeValue = localStorage.getItem('version-rocket:cancelled') || '';
-    const todayDate = new Date().toLocaleDateString() || '';
+    const todayDate = new Date().toLocaleDateString();
     const cancelModeTypeValueInSession = sessionStorage.getItem('version-rocket:cancelled') || '';
     const isStopWorker = cancelUpdateAndStopWorker || false;
     switch (cancelModeType) {

utils/index.ts

@@ -45,12 +45,12 @@ export const cancelUpdateFunc = (
   cancelMode: string | undefined,
   newVersion: string,
   cancelUpdateAndStopWorker: boolean | undefined,
-  worker: Worker | undefined
+  worker: Worker | undefined,
 ) => {
   const cancelModeType = cancelMode || 'ignore-current-version'
   const cancelModeTypeValue =
     localStorage.getItem('version-rocket:cancelled') || ''
-  const todayDate = new Date().toLocaleDateString() || ''
+  const todayDate = new Date().toLocaleDateString()
   const cancelModeTypeValueInSession =
     sessionStorage.getItem('version-rocket:cancelled') || ''
   const isStopWorker = cancelUpdateAndStopWorker || false