Proxy toString() method of proxied methods

src/createPrototypeProxy.js

@@ -6,6 +6,20 @@ export default function createPrototypeProxy() {
   let current = null;
   let mountedInstances = [];
 
+  /**
+   * Creates a proxied toString() method pointing to the current version's toString().
+   */
+  function proxyToString(name) {
+    // Wrap to always call the current version
+    return function toString() {
+      if (typeof current[name] === 'function') {
+        return current[name].toString();
+      } else {
+        return '<method was deleted>';
+      }
+    };
+  }
+
   /**
    * Creates a proxied method that calls the current version, whenever available.
    */
@@ -19,6 +33,8 @@ export default function createPrototypeProxy() {
 
     // Copy properties of the original function, if any
     assign(proxiedMethod, current[name]);
+    proxiedMethod.toString = proxyToString(name);
+
     return proxiedMethod;
   }
 
@@ -31,6 +47,7 @@ export default function createPrototypeProxy() {
       return current.componentDidMount.apply(this, arguments);
     }
   }
+  proxiedComponentDidMount.toString = proxyToString('componentDidMount');
 
   /**
    * Augments the original componentWillUnmount with instance tracking.
@@ -45,6 +62,7 @@ export default function createPrototypeProxy() {
       return current.componentWillUnmount.apply(this, arguments);
     }
   }
+  proxiedComponentWillUnmount.toString = proxyToString('componentWillUnmount');
 
   /**
    * Defines a property on the proxy.

test/consistency.js

@@ -162,6 +162,26 @@ describe('consistency', () => {
           }
         });
       });
+
+      it('preserves toString() of methods', () => {
+        let proxy = createProxy(Bar);
+
+        const Proxy = proxy.get();
+        ['doNothing', 'render', 'componentWillUnmount'].forEach(name => {
+          const originalMethod = Bar.prototype[name];
+          const proxyMethod = Proxy.prototype[name];
+          expect(originalMethod.toString()).toEqual(proxyMethod.toString());
+        });
+
+        const doNothingBeforeItWasDeleted = Proxy.prototype.doNothing;
+        proxy.update(Baz);
+        ['render', 'componentWillUnmount'].forEach(name => {
+          const originalMethod = Baz.prototype[name];
+          const proxyMethod = Proxy.prototype[name];
+          expect(originalMethod.toString()).toEqual(proxyMethod.toString());
+        });
+        expect(doNothingBeforeItWasDeleted.toString()).toEqual('<method was deleted>');
+      });
     });
   });