chore(tests): add tests for UIRouterConsumer

Author: elboman

src/components/UIRouter.tsx

@@ -94,10 +94,6 @@ export class UIRouter extends Component<UIRouterProps, UIRouterState> {
   }
 
   render() {
-    return (
-      <UIRouterProvider value={this.router}>
-        {this.props.children}
-      </UIRouterProvider>
-    );
+    return <UIRouterProvider value={this.router}>{this.props.children}</UIRouterProvider>;
   }
 }

src/components/__tests__/UIRouter.test.tsx

@@ -4,12 +4,7 @@ import * as React from 'react';
 import * as PropTypes from 'prop-types';
 import { mount } from 'enzyme';
 
-import {
-  UIRouter,
-  UIRouterConsumer,
-  UIRouterReact,
-  memoryLocationPlugin,
-} from '../../index';
+import { UIRouter, UIRouterConsumer, UIRouterReact, memoryLocationPlugin } from '../../index';
 
 class Child extends React.Component<any, any> {
   static propTypes: React.ValidationMap<any> = {
@@ -34,9 +29,7 @@ describe('<UIRouter>', () => {
   it('creates a router instance', () => {
     const wrapper = mount(
       <UIRouter plugins={[memoryLocationPlugin]} states={[]}>
-        <UIRouterConsumer>
-          {router => <Child router={router} />}
-        </UIRouterConsumer>
+        <UIRouterConsumer>{router => <Child router={router} />}</UIRouterConsumer>
       </UIRouter>,
     );
     expect(wrapper.find(Child).props().router).toBeDefined();
@@ -45,14 +38,41 @@ describe('<UIRouter>', () => {
   it('accepts an instance via prop', () => {
     const router = new UIRouterReact();
     router.plugin(memoryLocationPlugin);
-    (router as any).__TEST__ = true;
     const wrapper = mount(
       <UIRouter router={router}>
-        <UIRouterConsumer>
-          {router => <Child router={router} />}
-        </UIRouterConsumer>
+        <UIRouterConsumer>{router => <Child router={router} />}</UIRouterConsumer>
       </UIRouter>,
     );
-    expect(wrapper.find(Child).props().router.__TEST__).toBe(true);
+    expect(wrapper.find(Child).props().router).toBe(router);
+  });
+
+  describe('<UIRouterCosumer>', () => {
+    it('passes down the router instance', () => {
+      const wrapper = mount(
+        <UIRouter plugins={[memoryLocationPlugin]}>
+          <UIRouterConsumer>
+            {router => {
+              expect(router).toBeInstanceOf(UIRouterReact);
+              return null;
+            }}
+          </UIRouterConsumer>
+        </UIRouter>,
+      );
+    });
+
+    it('passes down the correct router instance when passed via props', () => {
+      const router = new UIRouterReact();
+      router.plugin(memoryLocationPlugin);
+      const wrapper = mount(
+        <UIRouter router={router}>
+          <UIRouterConsumer>
+            {_router => {
+              expect(_router).toBe(router);
+              return null;
+            }}
+          </UIRouterConsumer>
+        </UIRouter>,
+      );
+    });
   });
 });