fix unit testing

trychenhai · Apr 11, 2022 · b3f9044 · b3f9044
1 parent 564cbd1
commit b3f9044
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/setup/useDragAndDrop.ts b/src/setup/useDragAndDrop.ts
@@ -98,14 +98,14 @@ export default function useDragAndDrop(cmn: IUseCommon, props: INodeProps): {} {
     });
 
     const getDataTransfer = (evt: DragEvent) : string | object | null => {
-        if (!evt.dataTransfer) return null;
+        if (!evt || !evt.dataTransfer) return null;
         const jsonPayload = evt.dataTransfer.getData("application/json");
         if (jsonPayload)  return JSON.parse(jsonPayload);
         return evt.dataTransfer.getData("text/plain");
     };
 
     const isExternalSrc = (evt: DragEvent) : boolean => {
-        return evt.dataTransfer?.items?.length > 0;
+        return evt?.dataTransfer?.items?.length > 0;
     };
 
     const dragstart = (evt: DragEvent): void => {

diff --git a/tests/unit/useDragAndDrop.spec.ts b/tests/unit/useDragAndDrop.spec.ts
@@ -108,8 +108,11 @@ describe("test use Drag and Drop", () => {
             wrapper: wrapper.value
         };
         fakeContext = {
+            dataTransfer: null,
             dragged: fakeDragged,
-            target: fakeTarget
+            target: fakeTarget,
+            evt: undefined,
+            external: false
         };        
         props = {
             parentId: ref(null)