Merge pull request #3 from FrankAst/fix-ci

refactor: fix tests, ci
borodayev · Jan 8, 2019 · 84970b0 · 84970b0
2 parents 61cf2b0 + 85f83fb
commit 84970b0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/__tests__/diff-test.js b/src/__tests__/diff-test.js
@@ -31,7 +31,7 @@ describe('findDiff', () => {
       { k: 'N', p: ['string'], r: 'str1' },
       { k: 'N', p: ['newString'], r: 'str2' },
       { k: 'N', p: ['number'], r: 1 },
-      { k: 'N', p: ['date'], r: new Date('2018-11-29T18:00:00.000Z') },
+      { k: 'N', p: ['date'], r: new Date('2018/11/30') },
     ]);
 
     expect(modifiedDiffs).toEqual([
@@ -43,9 +43,9 @@ describe('findDiff', () => {
       { k: 'E', l: 0, p: ['number'], r: 1 },
       {
         k: 'E',
-        l: new Date('2018-12-29T18:00:00.000Z'),
+        l: new Date('2018/12/30'),
         p: ['date'],
-        r: new Date('2018-11-29T18:00:00.000Z'),
+        r: new Date('2018/11/30'),
       },
       { k: 'N', p: ['newString'], r: 'str2' },
     ]);

diff --git a/src/definitions.js b/src/definitions.js
@@ -65,12 +65,13 @@ export class DiffDoc /* :: extends Mongoose$Document */ {
       .exec();
   }
 
-  static async revertToVersion(doc: Object, v: number): Promise<any> {
+  static async revertToVersion(d: Object, v: number): Promise<any> {
+    const doc = typeof d.toObject === 'function' ? d.toObject() : d;
     const changes: Array<RawChangeT> = [];
     const diffDocs = (await this.findAfterVersion(doc._id, v): any);
 
     if (diffDocs.length === 0) return null;
-    diffDocs.forEach(d => changes.push(...d.c));
+    diffDocs.forEach(diffDoc => changes.push(...diffDoc.c));
     return revertChanges(doc, changes);
   }