Add array support to patch

Swizz · Nov 11, 2017 · f28667d · f28667d
1 parent a7e819a
commit f28667d
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -31,13 +31,13 @@ export function diff(from, to) {
   return patch
 }
 
-export function patch(from, diff) {
+export function patch(from, diff, array) {
   var to = {}
   for (var key in from) {
     to[key] = from[key]
   }
   for (var key in diff) {
     to[key] = diff[key]
   }
-  return to
+  return array && "length" in to ? Array.from(to) : to
 }
diff --git a/tests/patch.test.js b/tests/patch.test.js
@@ -111,6 +111,10 @@ describe("array", () => {
     expect(Array.from(patch(from, future.do))).toEqual(to)
 
     expect(Array.from(patch(to, future.undo))).toEqual(from)
+
+    expect(patch(from, future.do, true)).toEqual(to)
+
+    expect(patch(to, future.undo, true)).toEqual(from)
   })
 
   test("patch deleted", () => {
@@ -123,6 +127,10 @@ describe("array", () => {
     expect(Array.from(patch(from, future.do))).toEqual(to)
 
     expect(Array.from(patch(to, future.undo))).toEqual(from)
+
+    expect(patch(from, future.do, true)).toEqual(to)
+
+    expect(patch(to, future.undo, true)).toEqual(from)
   })
 
   test("patch updated", () => {
@@ -135,6 +143,10 @@ describe("array", () => {
     expect(Array.from(patch(from, future.do))).toEqual(to)
 
     expect(Array.from(patch(to, future.undo))).toEqual(from)
+
+    expect(patch(from, future.do, true)).toEqual(to)
+
+    expect(patch(to, future.undo, true)).toEqual(from)
   })
 
   test("patch kept", () => {
@@ -147,5 +159,21 @@ describe("array", () => {
     expect(Array.from(patch(from, future.do))).toEqual(to)
 
     expect(Array.from(patch(to, future.undo))).toEqual(from)
+
+    expect(patch(from, future.do, true)).toEqual(to)
+
+    expect(patch(to, future.undo, true)).toEqual(from)
   })
-})
+
+  test("do not transform object", () => {
+    const from = { a: 1, b: 1 }
+
+    const to = { a: 1, b: 2, c: 3 }
+
+    const future = diff(from, to)
+
+    expect(patch(from, future.do, true)).toEqual(to)
+
+    expect(patch(to, future.undo, true)).toEqual(from)
+  })
+})