Merge pull request #853 from naholyr/feat/expose-wrapped-component

HOC: expose wrapped component as WrappedComponent property
i18next · May 21, 2019 · 7481688 · 7481688
2 parents 58d508b + 64a5ded
commit 7481688
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 0 deletions.
diff --git a/src/withSSR.js b/src/withSSR.js
@@ -15,6 +15,7 @@ export function withSSR() {
 
     I18nextWithSSR.getInitialProps = composeInitialProps(WrappedComponent);
     I18nextWithSSR.displayName = `withI18nextSSR(${getDisplayName(WrappedComponent)})`;
+    I18nextWithSSR.WrappedComponent = WrappedComponent;
 
     return I18nextWithSSR;
   };

diff --git a/src/withTranslation.js b/src/withTranslation.js
@@ -22,6 +22,9 @@ export function withTranslation(ns, options = {}) {
     I18nextWithTranslation.displayName = `withI18nextTranslation(${getDisplayName(
       WrappedComponent,
     )})`;
+
+    I18nextWithTranslation.WrappedComponent = WrappedComponent;
+
     return options.withRef ? React.forwardRef(I18nextWithTranslation) : I18nextWithTranslation;
   };
 }
diff --git a/test/withSSR.spec.js b/test/withSSR.spec.js
@@ -47,6 +47,11 @@ describe('withSSR', () => {
     setI18n(i18n);
   });
 
+  it('should export wrapped component', () => {
+    const HocElement = withSSR()(TestComponent);
+    expect(HocElement.WrappedComponent).toBe(TestComponent);
+  });
+
   it('should set values', () => {
     const HocElement = withSSR()(TestComponent);
     const wrapper = mount(<HocElement initialI18nStore={{ foo: 'bar' }} initialLanguage="de" />);

diff --git a/test/withTranslation.spec.js b/test/withTranslation.spec.js
@@ -13,6 +13,11 @@ describe('withTranslation', () => {
     return <div>{t('key1')}</div>;
   }
 
+  it('should export wrapped component', () => {
+    const HocElement = withTranslation()(TestComponent);
+    expect(HocElement.WrappedComponent).toBe(TestComponent);
+  });
+
   it('should render correct content', () => {
     const HocElement = withTranslation()(TestComponent);
     const wrapper = mount(<HocElement />);