test(ui/uploader): fix mock fileReader to class

affects: @varlet/ui
varletjs · Jul 31, 2021 · f8ce7ac · f8ce7ac
1 parent 823942b
commit f8ce7ac
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/packages/varlet-ui/src/uploader/__tests__/index.spec.js b/packages/varlet-ui/src/uploader/__tests__/index.spec.js
@@ -72,6 +72,7 @@ test('test uploader preview', async () => {
   })
 
   await wrapper.vm.handleChange(createEvent('cat.jpg'))
+  await delay(16)
   await wrapper.find('.var-uploader__file').trigger('click')
   await delay(16)
 

diff --git a/packages/varlet-ui/src/utils/jest.ts b/packages/varlet-ui/src/utils/jest.ts
@@ -180,16 +180,19 @@ interface MockFileReaderCtx {
 export function mockFileReader(url: string) {
   const originMethod = window.FileReader
 
-  Object.assign(window, {
-    FileReader(this: MockFileReaderCtx) {
-      this.onload = function () {}
-      this.result = ''
+  class FileReader {
+    result = ''
 
-      this.readAsDataURL = function () {
-        this.result = url
-        this.onload()
-      }
-    },
+    onload() {}
+
+    readAsDataURL() {
+      this.result = url
+      this.onload()
+    }
+  }
+
+  Object.assign(window, {
+    FileReader,
   })
 
   return {