Allow render props(fkhadra#106)

vasiliydolmatov · Dec 17, 2017 · 801069f · 801069f
1 parent bfd595b
commit 801069f
Show file tree

Hide file tree

Showing 4 changed files with 1,909 additions and 6 deletions.
diff --git a/dist/ReactToastify.min.js b/dist/ReactToastify.min.js
diff --git a/dist/ReactToastify.min.js.map b/dist/ReactToastify.min.js.map
diff --git a/src/ToastContainer.js b/src/ToastContainer.js
@@ -247,7 +247,8 @@ class ToastContainer extends Component {
     return (
       isValidElement(content) ||
       typeof content === 'string' ||
-      typeof content === 'number'
+      typeof content === 'number' ||
+      this.isFunction(content)
     );
   }
 
@@ -310,6 +311,8 @@ class ToastContainer extends Component {
       content = cloneElement(content, {
         closeToast
       });
+    } else if (this.isFunction(content)) {
+      content = content({ closeToast });
     }
 
     this.collection = Object.assign({}, this.collection, {

diff --git a/src/__tests__/ToastContainer.js b/src/__tests__/ToastContainer.js
@@ -75,14 +75,15 @@ describe('ToastContainer', () => {
     expect(component.state().toast).toHaveLength(0);
   });
 
-  it('Should be able to render a react element, a string or a number without crashing', () => {
+  it('Should be able to render a react element, a string, a number, a render props without crashing', () => {
     const component = mount(<ToastContainer />);
     toaster('coucou');
     toaster(123);
     toaster(<div>plop</div>);
+    toaster(() => <div>plop</div>);
     jest.runAllTimers();
 
-    expect(component.state().toast).toHaveLength(3);
+    expect(component.state().toast).toHaveLength(4);
   });
 
   it('Should be able to display new toast on top', () => {