Merge pull request #12 from plusnew/hostNodes

[commonWrapper] add hosteNodes function
plusnew · May 20, 2019 · 49974c0 · 49974c0
2 parents 7818111 + 4a4870f
commit 49974c0
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "enzyme-adapter-plusnew",
-  "version": "0.6.44",
+  "version": "0.6.45",
   "description": "adapter for enzyme testsuite to the plusnew framework",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

diff --git a/src/wrapper/PlusnewCommonWrapper.ts b/src/wrapper/PlusnewCommonWrapper.ts
@@ -92,6 +92,10 @@ abstract class PlusnewCommonWrapper {
     return this.wrapper.hasClass.apply(this.wrapper, arguments);
   }
 
+  public hostNodes(): this {
+    return new this.WrapperClass(this.wrapper.hostNodes.apply(this.wrapper, arguments));
+  }
+
   public html(): string {
     return this.wrapper.html.apply(this.wrapper, arguments);
   }

diff --git a/test/mount_shallow.spec.tsx b/test/mount_shallow.spec.tsx
@@ -3,11 +3,11 @@ import PlusnewCommonWrapper from 'wrapper/PlusnewCommonWrapper';
 import { configure } from 'enzyme';
 import plusnew, { component, store, Props } from 'plusnew';
 
-interface common {
+interface Common {
   (element: plusnew.JSX.Element): PlusnewCommonWrapper;
 }
 
-function getMountFunction(callback: (mountWrapper: common) => void) {
+function getMountFunction(callback: (mountWrapper: Common) => void) {
   callback(mount as any);
   callback(shallow as any);
 }
@@ -428,6 +428,31 @@ describe('testing both renderers with:', () => {
     });
   });
 
+  describe('hostNodes()', () => {
+    it('basic test', () => {
+      getMountFunction((mount) => {
+        const MyComponent = component(
+          'MyComponent',
+          (_Props: Props<{ className: string }>) => false,
+        );
+
+        const MainComponent = component(
+          'Component',
+          () =>
+            <div>
+              <MyComponent className="foo" />
+              <span className="foo" />
+            </div>,
+        );
+
+        const wrapper = mount(<MainComponent />);
+
+        const twoNodes = wrapper.find('.foo');
+        expect(twoNodes.hostNodes().length).toBe(1);
+      });
+    });
+  });
+
   describe('html()', () => {
     it('basic test', () => {
       getMountFunction((mount) => {