feat(tests): render React tree returned by render function (#40)

Author: GHEMID-Mohamed

src/new.spec.js

@@ -4,7 +4,7 @@
 
 const assert = require("assert");
 const { createElement } = require("react");
-const { configure, mount } = require("enzyme");
+const { configure, mount, shallow } = require("enzyme");
 
 const CircularComputedError = require("./_CircularComputedError");
 const { withStore } = require("./");
@@ -92,6 +92,13 @@ describe("withStore", () => {
       assert(isReadOnly(props));
       expect(ownProps(props)).toEqual(["bar"]);
     });
+
+    it("returns the React tree to render", () => {
+      const wrapper = shallow(
+        createElement(withStore({}, () => createElement("h1")))
+      );
+      expect(wrapper.getElement()).toEqual(createElement("h1"));
+    });
   });
 
   describe("computed", () => {