Fix replacement of renamed/aliased imports (e.g. import { assert as somethingElse } from '@ember/debug')

fixtures/assert-expansion/expectation6.js

@@ -1,2 +1,3 @@
 (true && !((() => true)()) && console.assert((() => true)(), 'This is an assertion'));
-(true && !(false) && console.assert(false, 'This is an assertion 2'));
+(true && !(false) && console.assert(false, 'This is an assertion 2'));
+(true && !(false) && console.assert(false, 'renamed assertion'));

fixtures/assert-expansion/expectation7.js

@@ -1,2 +1,3 @@
 (true && !((() => true)()) && console.assert((() => true)(), 'This is an assertion'));
-(true && !(false) && console.assert(false, 'This is an assertion 2'));
+(true && !(false) && console.assert(false, 'This is an assertion 2'));
+(true && !(false) && console.assert(false, 'renamed assertion'));

fixtures/assert-expansion/sample.js

@@ -1,5 +1,7 @@
 import { DEBUG } from '@ember/env-flags';
 import { assert } from '@ember/debug-tools';
+import { assert as debugAssert } from '@ember/debug-tools';
 
 assert((() => true )(), 'This is an assertion');
-assert(false, 'This is an assertion 2');
+assert(false, 'This is an assertion 2');
+debugAssert(false, 'renamed assertion');

src/utils/builder.js

@@ -27,17 +27,23 @@ module.exports = class Builder {
    *
    * ($DEBUG && $GLOBAL_NS.assert($PREDICATE, $MESSAGE));
    */
-  assert(path) {
+  assert(path, options) {
     let predicate;
     if (this.assertPredicateIndex !== undefined) {
       predicate = (expression, args) => {
         return args[this.assertPredicateIndex];
       };
     }
 
-    this._createMacroExpression(path, {
-      predicate,
-    });
+    this._createMacroExpression(
+      path,
+      Object.assign(
+        {
+          predicate,
+        },
+        options
+      )
+    );
   }
 
   /**
@@ -102,7 +108,12 @@ module.exports = class Builder {
     } else if (options.buildConsoleAPI) {
       callExpression = options.buildConsoleAPI(expression, args);
     } else {
-      callExpression = this._createConsoleAPI(options.consoleAPI || callee, args);
+      callExpression = this._createConsoleAPI(
+        options.consoleAPI ||
+          (options.importedName && t.identifier(options.importedName)) ||
+          callee,
+        args
+      );
     }
 
     let prefixedIdentifiers = [];

src/utils/macros.js

@@ -34,7 +34,7 @@ module.exports = class Macros {
   collectDebugToolsSpecifiers(specifiers) {
     specifiers.forEach(specifier => {
       if (specifier.node.imported && SUPPORTED_MACROS.indexOf(specifier.node.imported.name) > -1) {
-        this.localDebugBindings.push(specifier.get('local'));
+        this.localDebugBindings.push(specifier);
       }
     });
   }
@@ -47,10 +47,15 @@ module.exports = class Macros {
 
     if (
       this.builder.t.isCallExpression(expression) &&
-      this.localDebugBindings.some(b => b.node.name === expression.callee.name)
+      this.localDebugBindings.some(b => b.get('local').node.name === expression.callee.name)
     ) {
-      let imported = path.scope.getBinding(expression.callee.name).path.node.imported.name;
-      this.builder[`${imported}`](path);
+      let specifier = path.scope.getBinding(expression.callee.name).path.node;
+      let imported = specifier.imported.name;
+      // The builder needs to be made aware of the the local name of the ImportSpecifier
+      this.builder[`${imported}`](path, {
+        localName: specifier.local.name,
+        importedName: imported,
+      });
     }
   }
 
@@ -62,12 +67,22 @@ module.exports = class Macros {
           // import declaration in question seems to have already been removed
           return;
         }
+
         let specifiers = importPath.get('specifiers');
 
         if (specifiers.length === this.localDebugBindings.length) {
-          this.localDebugBindings[0].parentPath.parentPath.remove();
+          importPath.remove();
         } else {
-          this.localDebugBindings.forEach(binding => binding.parentPath.remove());
+          this.localDebugBindings.forEach(binding => {
+            let specifier = binding.get('local').parentPath;
+            let importPath = specifier.parentPath;
+
+            if (importPath.get('specifiers').length === 1) {
+              importPath.remove();
+            } else {
+              specifier.remove();
+            }
+          });
         }
       }
     }

tests/create-tests.js

@@ -453,7 +453,7 @@ function createTests(options) {
             `./fixtures/${fixtureName}/expectation${babelVersion}.js`,
             'utf-8'
           );
-          expect(transform(sample, options).code).toEqual(expectation);
+          expect(transform(sample, options).code.trim()).toEqual(expectation.trim());
         });
       },