@@ -7,25 +7,6 @@ import { LoginPage, mapStateToProps } from '../../components/login';
77const { shell, ipcRenderer, remote } = require ( 'electron' ) ;
88const BrowserWindow = remote . BrowserWindow ;
99
10- function setup ( props ) {
11- const options = {
12- context : {
13- router : {
14- push : jest . fn ( ) ,
15- replace : jest . fn ( )
16- }
17- }
18- } ;
19-
20- const wrapper = shallow ( < LoginPage { ...props } /> , options ) ;
21-
22- return {
23- context : options . context ,
24- props : props ,
25- wrapper : wrapper ,
26- } ;
27- } ;
28-
2910describe ( 'components/login.js' , ( ) => {
3011 beforeEach ( function ( ) {
3112 BrowserWindow ( ) . loadURL . mockReset ( ) ;
@@ -58,7 +39,7 @@ describe('components/login.js', () => {
5839 isFetching : false
5940 } ;
6041
61- const { wrapper } = setup ( props ) ;
42+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
6243
6344 expect ( wrapper ) . toBeDefined ( ) ;
6445 expect ( wrapper . find ( '.desc' ) . text ( ) ) . toContain ( 'in your menu bar.' ) ;
@@ -84,7 +65,7 @@ describe('components/login.js', () => {
8465 isFetching : false
8566 } ;
8667
87- const { wrapper } = setup ( props ) ;
68+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
8869
8970 expect ( wrapper ) . toBeDefined ( ) ;
9071
@@ -116,7 +97,7 @@ describe('components/login.js', () => {
11697 isFetching : false
11798 } ;
11899
119- const { wrapper } = setup ( props ) ;
100+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
120101
121102 expect ( wrapper ) . toBeDefined ( ) ;
122103
@@ -148,7 +129,7 @@ describe('components/login.js', () => {
148129 isFetching : false
149130 } ;
150131
151- const { wrapper } = setup ( props ) ;
132+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
152133
153134 expect ( wrapper ) . toBeDefined ( ) ;
154135
@@ -179,7 +160,7 @@ describe('components/login.js', () => {
179160 isFetching : false
180161 } ;
181162
182- const { wrapper } = setup ( props ) ;
163+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
183164
184165 expect ( wrapper ) . toBeDefined ( ) ;
185166
@@ -197,7 +178,7 @@ describe('components/login.js', () => {
197178 isFetching : false
198179 } ;
199180
200- const { wrapper, context } = setup ( props ) ;
181+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
201182
202183 expect ( wrapper ) . toBeDefined ( ) ;
203184
@@ -207,8 +188,8 @@ describe('components/login.js', () => {
207188
208189 expect ( ipcRenderer . send ) . toHaveBeenCalledTimes ( 1 ) ;
209190 expect ( ipcRenderer . send ) . toHaveBeenCalledWith ( 'reopen-window' ) ;
210- expect ( context . router . push ) . toHaveBeenCalledTimes ( 1 ) ;
211- expect ( context . router . push ) . toHaveBeenCalledWith ( '/notifications ' ) ;
191+
192+ expect ( wrapper . props ( ) . to ) . toEqual ( '/' ) ;
212193 } ) ;
213194
214195 it ( 'should request the github token if the oauth code is received' , ( ) => {
@@ -222,7 +203,7 @@ describe('components/login.js', () => {
222203 isFetching : false
223204 } ;
224205
225- const { wrapper } = setup ( props ) ;
206+ const wrapper = shallow ( < LoginPage { ... props } /> ) ;
226207
227208 expect ( wrapper ) . toBeDefined ( ) ;
228209
0 commit comments