Add use cases for whitelist-database.ts

Author: sheepzh

package.json

@@ -45,4 +45,4 @@
     "vue": "^3.0.11",
     "vue-router": "^4.0.8"
   }
-}
+}

src/background/context-menus-manager.ts

@@ -1,8 +1,8 @@
-import whitelistDatabase from '../database/whitelist-database'
+import WhitelistDatabase from '../database/whitelist-database'
 import { t2Chrome } from '../util/i18n/chrome/t'
 import { extractHostname, isBrowserUrl } from '../util/pattern'
 
-const db = whitelistDatabase
+const db = new WhitelistDatabase(chrome.storage.local)
 
 const menuId = '_timer_menu_item_' + Date.now()
 

src/database/whitelist-database.ts

@@ -1,13 +1,12 @@
 import { WHITELIST_KEY } from "./constant"
 
-
-const ruleId = '_timer_whitelist_db_change_rule_id'
-
 class WhitelistDatabase {
 
-    private localStorage = chrome.storage.local
+    private localStorage: chrome.storage.StorageArea
 
-    constructor() { }
+    constructor(storage: chrome.storage.StorageArea) {
+        this.localStorage = storage
+    }
 
     private update(selectAll: string[]): Promise<void> {
         const obj = {}
@@ -62,4 +61,4 @@ class WhitelistDatabase {
     }
 }
 
-export default new WhitelistDatabase()
+export default WhitelistDatabase

src/service/timer-service.ts

@@ -1,5 +1,5 @@
 import TimerDatabase, { TimerCondition } from '../database/timer-database'
-import whitelistDatabase from '../database/whitelist-database'
+import WhitelistDatabase from '../database/whitelist-database'
 import ArchivedDatabase from '../database/archived-database'
 import SiteInfo from '../entity/dto/site-info'
 import { log } from '../common/logger'
@@ -13,6 +13,7 @@ const timerDatabase = new TimerDatabase(chrome.storage.local)
 const archivedDatabase = new ArchivedDatabase(chrome.storage.local)
 const iconUrlDatabase = new IconUrlDatabase(chrome.storage.local)
 const mergeRuleDatabase = new MergeRuleDatabase(chrome.storage.local)
+const whitelistDatabase = new WhitelistDatabase(chrome.storage.local)
 
 declare type PageParam = {
     pageNum?: number

src/service/whitelist-service.ts

@@ -1,6 +1,7 @@
 import { log } from "../common/logger"
-import whitelistDatabase from "../database/whitelist-database"
+import WhitelistDatabase from "../database/whitelist-database"
 
+const whitelistDatabase = new WhitelistDatabase(chrome.storage.local)
 /**
  * Service of whitelist
  * 

test/database/whitelist-database.test.ts

@@ -0,0 +1,21 @@
+import WhitelistDatabase from "../../src/database/whitelist-database"
+import storage from "../__mock__/storage"
+
+const db = new WhitelistDatabase(storage.local)
+
+describe('timer-database', () => {
+    beforeEach(async () => storage.local.clear())
+
+    test('1', async () => {
+        await db.add('www.baidu.com')
+        await db.add('google.com')
+        const list = await db.selectAll()
+        expect(list.sort()).toEqual(['www.baidu.com', 'google.com'].sort())
+        expect((await db.includes('www.baidu.com'))).toBeTruthy()
+        await db.remove('www.baidu.com')
+        expect((await db.selectAll())).toEqual(['google.com'])
+        expect((await db.includes('www.baidu.com'))).toBeFalsy()
+        await db.add('google.com')
+        expect((await db.selectAll())).toEqual(['google.com'])
+    })
+})