test(Cascader): fix test unit

fty1990 · Nov 23, 2018 · c721069 · c721069
1 parent 1f7f5fe
commit c721069
Showing 1 changed file with 22 additions and 10 deletions.
diff --git a/test/cascader/index-spec.js b/test/cascader/index-spec.js
@@ -41,14 +41,14 @@ describe('Cascader', () => {
     let wrapper;
 
     afterEach(() => {
-      const overlay = document.querySelectorAll('.next-overlay-wrapper');
-      overlay.forEach(dom => {
-        document.body.removeChild(dom);
-      });
-      if (wrapper) {
-          wrapper.unmount();
-          wrapper = null;
-      }
+        const overlay = document.querySelectorAll('.next-overlay-wrapper');
+        overlay.forEach(dom => {
+            document.body.removeChild(dom);
+        });
+        if (wrapper) {
+            wrapper.unmount();
+            wrapper = null;
+        }
     });
 
     it('should render single cascader', () => {
@@ -60,12 +60,16 @@ describe('Cascader', () => {
             changeCalled = true;
             assert(v === '2980');
             delete d.children;
+            delete d._source;
             assert.deepEqual(d, {
                 value: '2980',
                 label: '铜川',
                 pos: '0-0-1'
             });
-            e.selectedPath.forEach(d => delete d.children);
+            e.selectedPath.forEach(d => {
+                delete d.children;
+                delete d._source;
+            });
             assert.deepEqual(e, {
                 selectedPath: [{
                     value: '2973',
@@ -100,12 +104,16 @@ describe('Cascader', () => {
             changeCalled = true;
             assert(v === '2980');
             delete d.children;
+            delete d._source;
             assert.deepEqual(d, {
                 value: '2980',
                 label: '铜川',
                 pos: '0-0-1'
             });
-            e.selectedPath.forEach(d => delete d.children);
+            e.selectedPath.forEach(d => {
+                delete d.children;
+                delete d._source;
+            });
             assert.deepEqual(e, {
                 selectedPath: [{
                     value: '2973',
@@ -208,6 +216,8 @@ describe('Cascader', () => {
         let data;
         let extra;
         const handleChange = (v, d, e) => {
+            d.forEach(d => delete d._source);
+            e.checkedData.forEach(d => delete d._source);
             assert.deepEqual(value, sortByValue(v, true));
             assert.deepEqual(data, sortByValue(d));
             e.checkedData = sortByValue(e.checkedData);
@@ -270,6 +280,8 @@ describe('Cascader', () => {
         let data;
         let extra;
         const handleChange = (v, d, e) => {
+            d.forEach(d => delete d._source);
+            e.checkedData.forEach(d => delete d._source);
             assert.deepEqual(value, sortByValue(v, true));
             assert.deepEqual(data, sortByValue(d));
             e.checkedData = sortByValue(e.checkedData);