Fix lint and add more functionality

src/transformers/ava.test.ts

@@ -491,3 +491,43 @@ test(() => {});
 `
   )
 })
+
+test('can handle after.always or afterEach.always', () => {
+  expectTransformation(
+    `
+import test from 'ava';
+
+test.after.always(t => {});
+test.afterEach.always(t => {});
+`,
+    `
+afterAll(() => {});
+afterEach(() => {});
+`
+  )
+})
+
+test('does not mess with the context', () => {
+  expectTransformation(
+    `
+import test from 'ava';
+
+test.beforeEach((test) => {
+  test.context.hello = () => console.log('hello');
+});
+
+test('uses context', test => {
+  test.context.hello();
+});
+`,
+    `
+beforeEach(() => {
+  test.context.hello = () => console.log('hello');
+});
+
+test('uses context', () => {
+  context.hello();
+});
+`
+  )
+})

src/transformers/ava.ts

@@ -2,6 +2,7 @@
  * Codemod for transforming AVA tests into Jest.
  */
 import * as jscodeshift from 'jscodeshift'
+import { Identifier, MemberExpression } from 'jscodeshift'
 
 import { PROP_WITH_SECONDS_ARGS } from '../utils/consts'
 import finale from '../utils/finale'
@@ -186,9 +187,9 @@ const avaToJest: jscodeshift.Transform = (fileInfo, api, options) => {
         let jestMethod = 'test'
         const jestMethodArgs = p.node.arguments
 
-        // List like ['test', 'serial', 'cb']
+        // List like ['test', 'serial', 'cb', 'always']
         const avaMethods = getMemberExpressionElements(p.node.callee).filter(
-          (e) => e !== 'serial' && e !== testFunctionName && e !== 'cb'
+          (e) => e !== 'serial' && e !== testFunctionName && e !== 'cb' && e != 'always'
         )
 
         if (avaMethods.length === 1) {
@@ -199,6 +200,12 @@ const avaToJest: jscodeshift.Transform = (fileInfo, api, options) => {
             jestMethod = avaMethod
             logWarning(`Unknown AVA method "${avaMethod}"`, p)
           }
+        } else if (avaMethods[0] === 'context') {
+          let identifier: Identifier | MemberExpression = j.identifier(avaMethods[0])
+          avaMethods.slice(1).forEach((next) => {
+            identifier = j.memberExpression(identifier, j.identifier(next))
+          })
+          return j.callExpression(identifier, jestMethodArgs)
         } else if (avaMethods.length > 0) {
           logWarning('Skipping setup/teardown hooks is currently not supported', p)
         }

src/utils/tape-ava-helpers.ts

@@ -11,6 +11,9 @@ const { namedTypes } = types
  */
 function renameTestFunctionArgument(j, path, newArgument) {
   const lastArg = path.node.arguments[path.node.arguments.length - 1]
+  if (!lastArg) {
+    return
+  }
   if (lastArg.type === 'ArrowFunctionExpression') {
     const arrowFunction = j.arrowFunctionExpression(
       [j.identifier(newArgument === '' ? '()' : newArgument)],
@@ -251,4 +254,3 @@ export function detectUnsupportedNaming(fileInfo, j, ast, testFunctionName) {
       }
     })
 }
-