fixes #3103

docs/api/ComponentRegistry.md

@@ -2,9 +2,9 @@
 
 ## registerComponent
 
-`registerComponent(componentName: string, getComponentClassFunc: ComponentProvider): void`
+`registerComponent(componentName: string, getComponentClassFunc: ComponentProvider): React.ComponentType<any>`
 
-[source](https://github.com/wix/react-native-navigation/blob/v2/lib/src/components/ComponentRegistry.ts#L11)
+[source](https://github.com/wix/react-native-navigation/blob/v2/lib/src/components/ComponentRegistry.ts#L12)
 
 ---
 

docs/api/Navigation.md

@@ -8,7 +8,7 @@
 
 ## registerComponent
 
-`registerComponent(componentName: string, getComponentClassFunc: ComponentProvider): void`
+`registerComponent(componentName: string, getComponentClassFunc: ComponentProvider): React.ComponentType<any>`
 
 [source](https://github.com/wix/react-native-navigation/blob/v2/lib/src/Navigation.ts#L52)
 

lib/src/Navigation.ts

@@ -49,8 +49,8 @@ export class Navigation {
    * Every navigation component in your app must be registered with a unique name.
    * The component itself is a traditional React component extending React.Component.
    */
-  public registerComponent(componentName: string, getComponentClassFunc: ComponentProvider) {
-    this.componentRegistry.registerComponent(componentName, getComponentClassFunc);
+  public registerComponent(componentName: string, getComponentClassFunc: ComponentProvider): React.ComponentType<any> {
+    return this.componentRegistry.registerComponent(componentName, getComponentClassFunc);
   }
 
   /**

lib/src/components/ComponentRegistry.test.tsx

@@ -48,4 +48,10 @@ describe('ComponentRegistry', () => {
     const tree = renderer.create(<Component componentId='123' />);
     expect(tree.toJSON()!.children).toEqual(['Hello, World!']);
   });
+
+  it('returns the wrapped registered component', () => {
+    const result = uut.registerComponent('example.MyComponent.name', () => MyComponent);
+    expect(result).toBeDefined();
+    expect(result).toBe(mockRegistry.mock.calls[0][1]());
+  });
 });

lib/src/components/ComponentRegistry.ts

@@ -1,3 +1,4 @@
+import * as React from 'react';
 import { AppRegistry, ComponentProvider } from 'react-native';
 import { ComponentWrapper } from './ComponentWrapper';
 
@@ -8,10 +9,11 @@ export class ComponentRegistry {
     this.store = store;
   }
 
-  registerComponent(componentName: string, getComponentClassFunc: ComponentProvider) {
+  registerComponent(componentName: string, getComponentClassFunc: ComponentProvider): React.ComponentType<any> {
     const OriginalComponentClass = getComponentClassFunc();
     const NavigationComponent = ComponentWrapper.wrap(componentName, OriginalComponentClass, this.store);
     this.store.setOriginalComponentClassForName(componentName, OriginalComponentClass);
     AppRegistry.registerComponent(componentName, () => NavigationComponent);
+    return NavigationComponent;
   }
 }