use return statement with declared class

Author: alexeyraspopov

src/Coroutine.js

@@ -5,7 +5,7 @@ import invariant from 'invariant';
 function create(asyncFn, defaultVariables = () => ({})) {
   const componentName = asyncFn.name || asyncFn.displayName;
 
-  class AsyncComponent extends Component {
+  return class AsyncComponent extends Component {
     static get displayName() {
       return `Coroutine(${componentName})`;
     }
@@ -41,8 +41,6 @@ function create(asyncFn, defaultVariables = () => ({})) {
       return this.state.body;
     }
   }
-
-  return AsyncComponent;
 }
 
 export default { create };

tests/Coroutine.test.js

@@ -7,6 +7,7 @@ describe('Coroutine', async () => {
     async function render() {
       return <p>test</p>;
     }
+
     const TestComponent = Coroutine.create(render);
     const tree = Renderer.create(<TestComponent />);
 
@@ -20,7 +21,7 @@ describe('Coroutine', async () => {
   it('should pass initial information', async () => {
     function getVariables() {
       return { number: 13 };
-    };
+    }
 
     async function render({ number }) {
       return <p>{ number }</p>;