feat: add assertFunctionName option

src/config.ts

@@ -5,6 +5,7 @@ export interface InternalConfig {
   powerAssert: boolean;
   autoImport: string; // empty => no auto import
   staticTruthCheck: boolean;
+  assertFunctionName: string; // empty => no custom assert function name
 }
 
 /**
@@ -15,6 +16,7 @@ export interface Config {
   powerAssert?: boolean;
   autoImport?: boolean | string; // false => '', true => 'power-assert'
   staticTruthCheck?: boolean;
+  assertFunctionName?: string;
 }
 
 /**
@@ -24,6 +26,7 @@ export const defaultConfig = {
   powerAssert: true,
   autoImport: true,
   staticTruthCheck: false,
+  assertFunctionName: '',
 };
 
 /**
@@ -36,7 +39,7 @@ export const minimalConfig = {
 
 // tslint:disable no-parameter-reassignment
 export const extractConfigFromState = ({
-  opts: { powerAssert, autoImport, staticTruthCheck },
+  opts: { powerAssert, autoImport, staticTruthCheck, assertFunctionName },
 }: {
   opts: Config;
 }): InternalConfig => {
@@ -61,9 +64,15 @@ export const extractConfigFromState = ({
     ({ staticTruthCheck } = defaultConfig);
   }
 
+  // assertFunctionName
+  if (assertFunctionName === undefined) {
+    ({ assertFunctionName } = defaultConfig);
+  }
+
   return {
     powerAssert,
     autoImport,
     staticTruthCheck,
+    assertFunctionName,
   };
 };

src/generate-assert-identifier.ts

@@ -31,11 +31,18 @@ const findExistingImportFromSource = (
   return;
 };
 
-const addImport = (scope: Scope, t: typeof BabelTypes, source: string) => {
+const addImport = (
+  scope: Scope,
+  t: typeof BabelTypes,
+  source: string,
+  name: string,
+) => {
   const program = scope.getProgramParent().path;
 
-  // generate default import from source
-  const id = scope.generateUidIdentifier(ASSERT_IDENTIFIER_NAME);
+  // generate default or specified import from source
+  const id = name
+    ? t.identifier(name)
+    : scope.generateUidIdentifier(ASSERT_IDENTIFIER_NAME);
   (program as any).unshiftContainer(
     'body',
     t.importDeclaration(
@@ -49,16 +56,16 @@ const addImport = (scope: Scope, t: typeof BabelTypes, source: string) => {
 
 export default (
   t: typeof BabelTypes,
-  { autoImport: importSource }: InternalConfig,
+  { autoImport: importSource, assertFunctionName }: InternalConfig,
 ) => (scope: Scope) => {
   if (importSource) {
     const name = findExistingImportFromSource(scope, t, importSource);
     if (name) {
       return t.identifier(name);
     }
 
-    return addImport(scope, t, importSource);
+    return addImport(scope, t, importSource, assertFunctionName);
   }
 
-  return t.identifier(ASSERT_IDENTIFIER_NAME);
+  return t.identifier(assertFunctionName || ASSERT_IDENTIFIER_NAME);
 };

test/__snapshots__/assert-function-name.test.ts.snap

@@ -0,0 +1,9 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`generates import with specified name when autoImport is set 1`] = `
+"import check from \\"power-assert\\";
+
+expect: check(1 === 1);"
+`;
+
+exports[`generates specified assert function 1`] = `"expect: check(1 === 1);"`;

test/assert-function-name.test.ts

@@ -0,0 +1,29 @@
+import { transform } from '@babel/core';
+
+import plugin from '../src';
+import { Config, minimalConfig } from '../src/config';
+
+test('generates specified assert function', () => {
+  const { code } = transform(`expect: 1 === 1;`, {
+    plugins: [
+      [plugin, { ...minimalConfig, assertFunctionName: 'check' } as Config],
+    ],
+  });
+  expect(code).toMatchSnapshot();
+});
+
+test('generates import with specified name when autoImport is set', () => {
+  const { code } = transform(`expect: 1 === 1;`, {
+    plugins: [
+      [
+        plugin,
+        {
+          ...minimalConfig,
+          assertFunctionName: 'check',
+          autoImport: true,
+        } as Config,
+      ],
+    ],
+  });
+  expect(code).toMatchSnapshot();
+});