11module . exports = function ( file , api , options ) {
22 const j = api . jscodeshift ;
3+ const root = j ( file . source ) ;
34
4- function moveQUnitImportsFromEmberQUnit ( root ) {
5+ function moveQUnitImportsFromEmberQUnit ( ) {
56 let emberQUnitImports = root . find ( j . ImportDeclaration , { source : { value : 'ember-qunit' } } ) ;
67 // Find `module` and `test` imports
78 let migrateToQUnitImport = [ 'module' , 'test' , 'skip' , 'todo' ] ;
@@ -35,7 +36,7 @@ module.exports = function(file, api, options) {
3536 ) ;
3637 }
3738
38- function updateToNewEmberQUnitImports ( root ) {
39+ function updateToNewEmberQUnitImports ( ) {
3940 let mapping = {
4041 moduleFor : 'setupTest' ,
4142 moduleForComponent : 'setupRenderingTest' ,
@@ -85,8 +86,8 @@ module.exports = function(file, api, options) {
8586 . replace ( Array . from ( specifiers ) . map ( s => j . importSpecifier ( j . identifier ( s ) ) ) ) ;
8687 }
8788
88- function findTestHelperUsageOf ( root , property ) {
89- return root . find ( j . ExpressionStatement , {
89+ function findTestHelperUsageOf ( collection , property ) {
90+ return collection . find ( j . ExpressionStatement , {
9091 expression : {
9192 callee : {
9293 object : {
@@ -123,7 +124,7 @@ module.exports = function(file, api, options) {
123124 return [ moduleName , options , setupIdentifier ] ;
124125 }
125126
126- function updateModuleForToNestedModule ( root ) {
127+ function updateModuleForToNestedModule ( ) {
127128 const POSSIBLE_MODULES = [
128129 { expression : { callee : { name : 'moduleFor' } } } ,
129130 { expression : { callee : { name : 'moduleForComponent' } } } ,
@@ -244,7 +245,7 @@ module.exports = function(file, api, options) {
244245 bodyPath . replace ( bodyReplacement ) ;
245246 }
246247
247- function updateLookupCalls ( root ) {
248+ function updateLookupCalls ( ) {
248249 root
249250 . find ( j . MemberExpression , {
250251 object : {
@@ -259,7 +260,7 @@ module.exports = function(file, api, options) {
259260 } ) ;
260261 }
261262
262- function updateRegisterCalls ( root ) {
263+ function updateRegisterCalls ( ) {
263264 root
264265 . find ( j . MemberExpression , {
265266 object : {
@@ -284,7 +285,7 @@ module.exports = function(file, api, options) {
284285 } ) ;
285286 }
286287
287- function updateInjectCalls ( root ) {
288+ function updateInjectCalls ( ) {
288289 root
289290 . find ( j . CallExpression , {
290291 callee : {
@@ -327,20 +328,19 @@ module.exports = function(file, api, options) {
327328 }
328329
329330 const printOptions = options . printOptions || { quote : 'single' } ;
330- const root = j ( file . source ) ;
331331
332332 // Find `ember-qunit` imports
333333 let emberQUnitImports = root . find ( j . ImportDeclaration , { source : { value : 'ember-qunit' } } ) ;
334334 if ( emberQUnitImports . size ( ) === 0 ) {
335335 return file . source ;
336336 }
337337
338- moveQUnitImportsFromEmberQUnit ( root ) ;
339- updateToNewEmberQUnitImports ( root ) ;
340- updateModuleForToNestedModule ( root ) ;
341- updateLookupCalls ( root ) ;
342- updateRegisterCalls ( root ) ;
343- updateInjectCalls ( root ) ;
338+ moveQUnitImportsFromEmberQUnit ( ) ;
339+ updateToNewEmberQUnitImports ( ) ;
340+ updateModuleForToNestedModule ( ) ;
341+ updateLookupCalls ( ) ;
342+ updateRegisterCalls ( ) ;
343+ updateInjectCalls ( ) ;
344344
345345 return root . toSource ( printOptions ) ;
346346} ;
0 commit comments