Removing an empty statement will no longer leave trailing whitespace

Author: vweijsters

StyleCop.Analyzers/StyleCop.Analyzers.Test/ReadabilityRules/SA1106UnitTests.cs

@@ -340,6 +340,52 @@ public async Task TestMemberAsync(string declaration)
             await this.VerifyCSharpFixAsync(testCode, fixedCode).ConfigureAwait(false);
         }
 
+        /// <summary>
+        /// Verifies that the code fix will remove all unnecessary whitespace.
+        /// This is a regression for #1556
+        /// </summary>
+        /// <returns>A <see cref="Task"/> representing the asynchronous unit test.</returns>
+        [Fact]
+        public async Task VerifyCodeFixWillRemoveUnnecessaryWhitespaceAsync()
+        {
+            var testCode = @"
+class TestClass
+{
+    public void TestMethod1()
+    {
+        throw new System.NotImplementedException(); ;
+    }
+
+    public void TestMethod2()
+    {
+        throw new System.NotImplementedException(); /* c1 */ ;
+    }
+}";
+            var fixedCode = @"
+class TestClass
+{
+    public void TestMethod1()
+    {
+        throw new System.NotImplementedException();
+    }
+
+    public void TestMethod2()
+    {
+        throw new System.NotImplementedException(); /* c1 */
+    }
+}";
+
+            DiagnosticResult[] expected =
+            {
+                this.CSharpDiagnostic().WithLocation(6, 53),
+                this.CSharpDiagnostic().WithLocation(11, 62)
+            };
+
+            await this.VerifyCSharpDiagnosticAsync(testCode, expected, CancellationToken.None).ConfigureAwait(false);
+            await this.VerifyCSharpDiagnosticAsync(fixedCode, EmptyDiagnosticResults, CancellationToken.None).ConfigureAwait(false);
+            await this.VerifyCSharpFixAsync(testCode, fixedCode).ConfigureAwait(false);
+        }
+
         /// <inheritdoc/>
         protected override IEnumerable<DiagnosticAnalyzer> GetCSharpDiagnosticAnalyzers()
         {

StyleCop.Analyzers/StyleCop.Analyzers/ReadabilityRules/SA1106CodeFixProvider.cs

@@ -106,20 +106,39 @@ private static async Task<Document> RemoveEmptyStatementAsync(Document document,
 
         private static async Task<Document> RemoveSemicolonTextAsync(Document document, SyntaxToken token, CancellationToken cancellationToken)
         {
+            TextChange textChange;
+
             SourceText sourceText = await document.GetTextAsync(cancellationToken).ConfigureAwait(false);
             TextLine line = sourceText.Lines.GetLineFromPosition(token.SpanStart);
             if (sourceText.ToString(line.Span).Trim() == token.Text)
             {
                 // remove the line containing the semicolon token
-                TextChange textChange = new TextChange(line.SpanIncludingLineBreak, string.Empty);
+                textChange = new TextChange(line.SpanIncludingLineBreak, string.Empty);
                 return document.WithText(sourceText.WithChanges(textChange));
             }
+
+            TextSpan spanToRemove;
+            var whitespaceIndex = TriviaHelper.IndexOfTrailingWhitespace(token.LeadingTrivia);
+            if (whitespaceIndex >= 0)
+            {
+                spanToRemove = TextSpan.FromBounds(token.LeadingTrivia[whitespaceIndex].Span.Start, token.Span.End);
+            }
             else
             {
-                // remove just the semicolon
-                TextChange textChange = new TextChange(token.Span, string.Empty);
-                return document.WithText(sourceText.WithChanges(textChange));
+                var previousToken = token.GetPreviousToken();
+                whitespaceIndex = TriviaHelper.IndexOfTrailingWhitespace(previousToken.TrailingTrivia);
+                if (whitespaceIndex >= 0)
+                {
+                    spanToRemove = TextSpan.FromBounds(previousToken.TrailingTrivia[whitespaceIndex].Span.Start, token.Span.End);
+                }
+                else
+                {
+                    spanToRemove = token.Span;
+                }
             }
+
+            textChange = new TextChange(spanToRemove, string.Empty);
+            return document.WithText(sourceText.WithChanges(textChange));
         }
     }
 }