object name could be underscore or lodash

you-dont-need · Jul 2, 2016 · 4652fdd · 4652fdd
1 parent 2933a1e
commit 4652fdd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/rules/all.js b/lib/rules/all.js
@@ -39,7 +39,7 @@ const errorMessages = {
 module.exports = {
   create (context) {
     function checkForUnneededLibraryFunctions (node) {
-      const functionName = getUnderscoreFunctionName(node);
+      const functionName = get_FunctionName(node);
       const errorMessage = errorMessages[functionName];
 
       if (errorMessage) {
@@ -56,7 +56,7 @@ module.exports = {
   }
 };
 
-function getUnderscoreFunctionName (node) {
+function get_FunctionName (node) {
   const callee = node.callee;
-  return callee && callee.object && callee.object.name === '_' && callee.property && callee.property.name;
+  return callee && callee.object && (callee.object.name === '_' || callee.object.name === 'lodash' || callee.object.name === 'underscore') && callee.property && callee.property.name;
 }