Support removing wrapper element by passing a falsy component.

Author: STRML

lib/Router.js

@@ -30,16 +30,20 @@ function createRouter(name, component) {
     },
 
     render: function() {
-      // Pass all props except this component to the Router (containing div/body) and the children,
-      // which are swapped out by the route handler.
-      var props = assign({}, this.props);
-      delete props.component;
-      delete props.children;
-      delete props.childProps;
-
       // Render the Route's handler.
       var handler = this.renderRouteHandler(this.props.childProps);
-      return React.createElement(this.props.component, props, handler);
+
+      if (!this.props.component) {
+        return handler;
+      } else {
+        // Pass all props except this component to the Router (containing div/body) and the children,
+        // which are swapped out by the route handler.
+        var props = assign({}, this.props);
+        delete props.component;
+        delete props.children;
+        delete props.childProps;
+        return React.createElement(this.props.component, props, handler);
+      }
     }
   });
 }

tests/server/server.js

@@ -87,6 +87,34 @@ describe('react-router-component (on server)', function() {
     });
   });
 
+  describe('Custom Component', function() {
+    class AppSection extends React.Component {
+      render() {
+        return <Locations className="App" path={this.props.path} component="section">
+                 <Location path="/" handler={<div>mainpage</div>} />
+               </Locations>;
+      }
+    }
+
+    class AppNoWrapper extends React.Component {
+      render() {
+        return <Locations className="App" path={this.props.path} component={null}>
+                 <Location path="/" handler={<div>mainpage</div>} />
+               </Locations>;
+      }
+    }
+
+    it('renders to <section>', function() {
+      var markup = ReactDOMServer.renderToStaticMarkup(<AppSection path="/" />);
+      assert.equal(markup, '<section class="App"><div>mainpage</div></section>');
+    });
+
+    it('removes wrapper with falsy value', function() {
+      var markup = ReactDOMServer.renderToStaticMarkup(<AppNoWrapper path="/" />);
+      assert.equal(markup, '<div>mainpage</div>');
+    });
+  });
+
   describe('contextual router', function() {
     class rendersSubSlug extends React.Component {
       render() {