@@ -28,9 +28,9 @@ module.exports = {
2828      unexpected : 'Make an assertion on the page state before taking a screenshot' , 
2929    } , 
3030  } , 
31-   create   ( context )  { 
31+   create ( context )  { 
3232    return  { 
33-       CallExpression   ( node )  { 
33+       CallExpression ( node )  { 
3434        if  ( isCallingCyScreenshot ( node )  &&  ! isPreviousAnAssertion ( node ) )  { 
3535          context . report ( {  node,  messageId : 'unexpected'  } ) 
3636        } 
@@ -39,12 +39,12 @@ module.exports = {
3939  } , 
4040} 
4141
42- function  isRootCypress   ( node )  { 
42+ function  isRootCypress ( node )  { 
4343  while  ( node . type  ===  'CallExpression' )  { 
4444    if  ( node . callee . type  !==  'MemberExpression' )  return  false 
4545
46-     if  ( node . callee . object . type  ===  'Identifier'   && 
47-          node . callee . object . name  ===  'cy' )  { 
46+     if  ( node . callee . object . type  ===  'Identifier' 
47+       &&  node . callee . object . name  ===  'cy' )  { 
4848      return  true 
4949    } 
5050
@@ -54,26 +54,26 @@ function isRootCypress (node) {
5454  return  false 
5555} 
5656
57- function  getPreviousInChain   ( node )  { 
58-   return  node . type  ===  'CallExpression'   && 
59-           node . callee . type  ===  'MemberExpression'   && 
60-           node . callee . object . type  ===  'CallExpression'   && 
61-           node . callee . object . callee . type  ===  'MemberExpression'   && 
62-           node . callee . object . callee . property . type  ===  'Identifier'   && 
63-           node . callee . object . callee . property . name 
57+ function  getPreviousInChain ( node )  { 
58+   return  node . type  ===  'CallExpression' 
59+     &&   node . callee . type  ===  'MemberExpression' 
60+     &&   node . callee . object . type  ===  'CallExpression' 
61+     &&   node . callee . object . callee . type  ===  'MemberExpression' 
62+     &&   node . callee . object . callee . property . type  ===  'Identifier' 
63+     &&  node . callee . object . callee . property . name 
6464} 
6565
66- function  getCallExpressionCypressCommand   ( node )  { 
67-   return  isRootCypress ( node )   && 
68-           node . callee . property . type  ===  'Identifier'   && 
69-           node . callee . property . name 
66+ function  getCallExpressionCypressCommand ( node )  { 
67+   return  isRootCypress ( node ) 
68+     &&   node . callee . property . type  ===  'Identifier' 
69+     &&  node . callee . property . name 
7070} 
7171
72- function  isCallingCyScreenshot   ( node )  { 
72+ function  isCallingCyScreenshot ( node )  { 
7373  return  getCallExpressionCypressCommand ( node )  ===  'screenshot' 
7474} 
7575
76- function  getPreviousCypressCommand   ( node )  { 
76+ function  getPreviousCypressCommand ( node )  { 
7777  const  previousInChain  =  getPreviousInChain ( node ) 
7878
7979  if  ( previousInChain )  { 
@@ -97,15 +97,15 @@ function getPreviousCypressCommand (node) {
9797
9898  const  previousStatement  =  body [ index  -  1 ] 
9999
100-   if  ( previousStatement . type  !==  'ExpressionStatement'   || 
101-        previousStatement . expression . type  !==  'CallExpression' )  { 
100+   if  ( previousStatement . type  !==  'ExpressionStatement' 
101+     ||  previousStatement . expression . type  !==  'CallExpression' )  { 
102102    return  null 
103103  } 
104104
105105  return  getCallExpressionCypressCommand ( previousStatement . expression ) 
106106} 
107107
108- function  isPreviousAnAssertion   ( node )  { 
108+ function  isPreviousAnAssertion ( node )  { 
109109  const  previousCypressCommand  =  getPreviousCypressCommand ( node ) 
110110
111111  return  assertionCommands . indexOf ( previousCypressCommand )  >=  0 
0 commit comments