Update call fixer logic

rules/no-useless-undefined.js

@@ -58,9 +58,20 @@ const create = context => {
 		),
 		[lastArgumentSelector]: listener(
 			(node, fixer) => {
-				const tokenAfter = context.getTokenAfter(node);
-				return (isCommaToken(tokenAfter) ? [node, tokenAfter] : [node])
-					.map(nodeOrToken => remove(nodeOrToken, fixer));
+				const argumentNodes = node.parent.arguments;
+				const lastArgument = argumentNodes[argumentNodes.length - 2];
+				let [start, end] = node.range;
+				if (lastArgument) {
+					start = lastArgument.range[1];
+				} else {
+					// If it's the only argument, and there is trailing comma, we need remove it.
+					const tokenAfter = context.getTokenAfter(node);
+					if (isCommaToken(tokenAfter)) {
+						end = tokenAfter.range[1];
+					}
+				}
+
+				return fixer.removeRange([start, end]);
 			}
 		)
 	};

test/no-useless-undefined.js

@@ -72,14 +72,39 @@ ruleTester.run('better-regex', rule, {
 			output: 'foo();',
 			errors
 		},
+		{
+			code: 'foo(undefined, undefined);',
+			output: 'foo(undefined);',
+			errors
+		},
 		{
 			code: 'foo(undefined,);',
 			output: 'foo();',
 			errors
 		},
+		{
+			code: 'foo(undefined, undefined,);',
+			output: 'foo(undefined,);',
+			errors
+		},
+		{
+			code: 'foo(bar, undefined);',
+			output: 'foo(bar);',
+			errors
+		},
+		{
+			code: 'foo(bar, undefined, undefined);',
+			output: 'foo(bar, undefined);',
+			errors
+		},
 		{
 			code: 'foo(bar, undefined,);',
-			output: 'foo(bar, );',
+			output: 'foo(bar,);',
+			errors
+		},
+		{
+			code: 'foo(bar, undefined, undefined,);',
+			output: 'foo(bar, undefined,);',
 			errors
 		},
 		{