Merge pull request cerebral#157 from cerebral/proxyFixes

Proxy fixes

Author: christianalfoni

packages/node_modules/proxy-state-tree/src/index.test.ts

@@ -451,6 +451,16 @@ describe('FUNCTIONS', () => {
 
     expect(tree.get().foo).toBe('bar')
   })
+  test('should be able to set functions', () => {
+    const tree = new ProxyStateTree({})
+    const state = tree.get()
+
+    tree.startMutationTracking()
+    state.foo = () => 'bar'
+    tree.clearMutationTracking()
+
+    expect(state.foo()).toBe('bar')
+  })
   test('should pass proxy-state-tree instance and path', () => {
     const state = {
       foo: (proxyStateTree, path) => {

packages/node_modules/proxy-state-tree/src/index.ts

@@ -102,7 +102,7 @@ export class ProxyStateTree {
     const tree = this.parentTree || this
     const flushId = tree.currentFlushId++
 
-    for (let objectChange of tree.objectChanges) {
+    for (let objectChange of this.objectChanges) {
       if (tree.pathDependencies[objectChange]) {
         paths.add(objectChange)
       }

packages/node_modules/proxy-state-tree/src/proxify.ts

@@ -165,6 +165,10 @@ function createObjectProxy(tree, value, path) {
         })
       }
 
+      if (typeof value === 'function') {
+        return Reflect.set(target, prop, () => value)
+      }
+
       return Reflect.set(target, prop, value)
     },
     deleteProperty(target, prop) {