Complete unit test

Author: sheepzh

test/database/archived-database.test.ts

@@ -26,4 +26,34 @@ describe('archived-database', () => {
         expect(baiduInfo.total).toEqual(2)
         expect(baiduInfo.time).toEqual(4)
     })
+
+    test('2', async () => {
+        const key = `_a_${baidu}`
+        const toImport = {}
+        toImport[key] = {
+            focus: 1,
+            time: 2,
+            total: 3
+        }
+        await archivedDatabase.importData(toImport)
+        const result = await archivedDatabase.selectArchived(new Set([baidu]))
+        expect(result.length).toEqual(1)
+        const baiduInfo = result[0]
+        expect(baiduInfo.focus).toEqual(1)
+        expect(baiduInfo.time).toEqual(2)
+        expect(baiduInfo.total).toEqual(3)
+    })
+
+    test('3', async () => {
+        const key = `_a_${baidu}`
+        const toImport = {}
+        toImport[key] = {}
+        await archivedDatabase.importData(toImport)
+        const result = await archivedDatabase.selectArchived(new Set([baidu]))
+        expect(result.length).toEqual(1)
+        const baiduInfo = result[0]
+        expect(baiduInfo.focus).toEqual(0)
+        expect(baiduInfo.time).toEqual(0)
+        expect(baiduInfo.total).toEqual(0)
+    })
 })

test/database/icon-url-database.test.ts

@@ -6,7 +6,11 @@ const db = new IconUrlDatabase(storage.local)
 const baidu = 'baidu.com'
 
 describe('icon-url-database', () => {
-    beforeEach(async () => storage.local.clear())
+    beforeEach(async () => {
+        await storage.local.clear()
+        const mockUserAgent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.141 Safari/537.36"
+        Object.defineProperty(global.navigator, 'userAgent', { value: mockUserAgent, configurable: true })
+    })
 
     test('1', async () => {
         await db.put(baidu, 'test1')